Add Neo4j graph database support and update environment variables

app/__init__.py

@@ -30,12 +30,3 @@ except ValueError:
     pass
 
 db = firestore.client()
-
-# LOAD NEO4J DB
-URI = os.environ.get("NEO4J_URI")
-AUTH = (os.environ.get("NEO4J_USERNAME"), os.environ.get("NEO4J_PASSWORD"))
-driver = GraphDatabase.driver(URI, auth=AUTH)
-driver.verify_connectivity()
-driver.execute_query(
-    "CREATE CONSTRAINT uid IF NOT EXISTS FOR (p:Person) REQUIRE p.uid IS UNIQUE"
-)

app/graphdb/__init__.py

@@ -0,0 +1,12 @@
+import os
+
+from neo4j import GraphDatabase
+
+
+URI = os.environ.get("NEO4J_URI")
+AUTH = (os.environ.get("NEO4J_USERNAME"), os.environ.get("NEO4J_PASSWORD"))
+driver = GraphDatabase.driver(URI, auth=AUTH)
+driver.verify_connectivity()
+driver.execute_query(
+    "CREATE CONSTRAINT uid IF NOT EXISTS FOR (p:Person) REQUIRE p.uid IS UNIQUE"
+)

app/main.py

@@ -2,8 +2,7 @@ from fastapi import FastAPI
 from starlette.responses import RedirectResponse
 
 
-from app.graphdb.main import insert2PersonAndSetFriend, deleteFriend
-from .routers import image, video, friend_request, me, auth
+from .routers import image, video, me, auth, friend_request
 
 app = FastAPI()
 

app/routers/friend_request.py

@@ -7,7 +7,7 @@ from fastapi import APIRouter, Depends, HTTPException, Response
 from app.dependencies import get_current_user
 from app import db, logger
 from enum import Enum
-from app.graphdb.main import insert2PersonAndSetFriend
+from ..graphdb.main import insert2PersonAndSetFriend
 from fastapi import status
 
 

compose.yaml

@@ -5,10 +5,12 @@ services:
         ports:
             - "3000:3000"
         environment:
-            - SUPABASE_URL=your-supabase-url
-            - SUPABASE_KEY=your-supabase-key
-            - FIREBASE_CREDENTIALS=your-firebase-credentials
-            - FIREBASE_API_KEY=your-firebase-api-key
-            - NEO4J_URI=your-neo4j-uri
-            - NEO4J_USERNAME=your-neo4j-username
-            - NEO4J_PASSWORD=your-neo4j-password
+            - SUPABASE_URL=your_supabase_url
+            - SUPABASE_KEY=your_supabase_sevice_role_key
+            - FIREBASE_CREDENTIALS=your_firebase_credentials
+            - FIREBASE_API_KEY=your_firebase_api_key
+            - NEO4J_URI=your_neo4j_uri
+            - NEO4J_USERNAME=your_neo4j_username
+            - NEO4J_PASSWORD=your_neo4j_password
+            - AURA_INSTANCEID=your_neo4j_aura_instance_id
+            - AURA_INSTANCENAME=your_neo4j_aura_instance_name