"""Neo4J DB connector."""

from neo4j import GraphDatabase
from tests.integration.utils import config

URI = config.NEO4J_URL
AUTH = (config.NEO4J_USER, config.NEO4J_PASSWORD)
MAX_RETRY_TIME = config.NEO4J_MAX_RETRY_TIME


def execute_query(query, params):
    """Execute cypher query."""
    with GraphDatabase.driver(URI, auth=AUTH, max_transaction_retry_time=MAX_RETRY_TIME) as driver:
        return driver.execute_query(query, params)