neo4j:
    image: neo4j:latest
    container_name: storage
    environment:
      - NEO4J_AUTH=neo4j/password
    ports:
      - '7888:7474'
      - '7999:7687'
    volumes:
      - './src/core/neo_db/data:/usr/src/app/neo_db/data'
      - './src/core/neo_db/conf:/usr/src/app/neo_db/conf'
      - './src/core/neo_db/plugins:/usr/src/app/neo_db/plugins'
      - './src/core/neo_db/logs:/usr/src/app/neo_db/logs'
    restart: unless-stopped


async def get_neo_session() -> AsyncGenerator:
    async with AsyncGraphDatabase.driver(
        settings.neo4j.NEO4J_URL,
        auth=settings.neo4j.NEO4J_AUTH,
    ) as driver:
        async with driver.session() as session:
            yield session


        query = (
            "CREATE (u1:User { user_id: user_1_id })"
            "CREATE (u2:User { user_id: user_2_id })"
            "CREATE (u1)-[:KNOWS]->(u2)"
            "RETURN u1.user_id, u2.user_id"
        )
        try:
            await neo_session.run(
                query,
                user_1_id=user_id_1,
                user_id_2=user_id_2,
                result_transformer_=lambda r: r.single(strict=True),
            )
        except (DriverError, Neo4jError) as e:
            print("\n\n\n\n\n\n", e, "\n\n\n\n\n\n")
            raise HTTPException(
                detail="Something went wrong",
                status_code=status.HTTP_400_BAD_REQUEST,
            )