from neo4j import READ_ACCESS, GraphDatabase

from src import config


__all__ = ["get_session"]

assert config.NEO4J_URL, "Neo4j URL is required"
assert config.NEO4J_USERNAME, "Neo4j username is required"
assert config.NEO4J_PASSWORD, "Neo4j password is required"

neo4j_driver = GraphDatabase.driver(
    config.NEO4J_URL,
    auth=(config.NEO4J_USERNAME, config.NEO4J_PASSWORD),
    max_transaction_retry_time=config.NEO4J_MAX_RETRY_TIME,
)


def get_session():
    return neo4j_driver.session(default_access_mode=READ_ACCESS)