"""Neo4J DB connector."""

from lambdacommon.common_config import logger
from neo4j import GraphDatabase

import config


def get_driver():
    """Connect to Neo4J and return Neo4J driver."""
    try:
        driver = GraphDatabase.driver(
            config.NEO4J_URL,
            auth=(config.NEO4J_USERNAME, config.NEO4J_PASSWORD),
            max_retry_time=config.NEO4J_MAX_RETRY_TIME)
        return driver
    except Exception as e:
        logger.exception('Failed to connect to Neo4J.')
        raise e