#!/usr/bin/python3
import os

from neo4j import GraphDatabase, basic_auth
from settings import settings


# Neo4j 数据库操作类
class Neo4jProvider:

    """创建 Neo4j 数据库连接"""

    def __init__(self) -> None:
        # 获取环境变量值，如果没有就返回默认值
        url = settings.NEO4J_URI
        username = settings.NEO4J_USER
        password = settings.NEO4J_PASSWORD
        neo4j_version = settings.NEO4J_VERSION
        database = settings.NEO4J_DATABASE
        port = int(settings.NEO4J_PORT)
        self.driver = GraphDatabase.driver(url, auth=basic_auth(username, password))