Merge pull request #11 from HunyyDev/dev/NMPhap

app/routers/video.py

@@ -15,21 +15,22 @@ from fastapi import (
     BackgroundTasks,
     status,
 )
-import requests
+from firebase_admin import messaging
+from app import db
 from app import supabase
 from app.dependencies import get_current_user
 from app.routers.image import inferenceImage
+from google.cloud.firestore_v1.base_query import FieldFilter
 
 router = APIRouter(prefix="/video", tags=["Video"])
 
 
-@router.post("/{artifactId}")
+@router.post("")
 async def handleVideoRequest(
-    artifactId: str,
     file: UploadFile,
     background_tasks: BackgroundTasks,
     threshold: float = 0.3,
-    _=Depends(get_current_user),
+    user=Depends(get_current_user),
 ):
     if re.search("^video\/", file.content_type) is None:
         raise HTTPException(
@@ -38,12 +39,15 @@ async def handleVideoRequest(
         )
 
     try:
+        if user["sub"] is None:
+            return HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail="User not found")
         id = str(now())
+        _, artifact_ref = db.collection("artifacts").add({"name": id + ".mp4", "status": "pending"})
         os.mkdir(id)
         async with aiofiles.open(os.path.join(id, "input.mp4"), "wb") as out_file:
             while content := await file.read(1024):
                 await out_file.write(content)
-        background_tasks.add_task(inferenceVideo, artifactId, id, threshold)
+        background_tasks.add_task(inferenceVideo, artifact_ref.id, id, threshold)
         return id + ".mp4"
     except ValueError as err:
         print(err)
@@ -145,7 +149,42 @@ async def inferenceVideo(artifactId: str, inputDir: str, threshold: float):
 
 
 def updateArtifact(artifactId: str, body):
-    url = "https://firebasetot.onrender.com/artifacts/" + artifactId
-    payload = json.dumps(body)
-    headers = {"Content-Type": "application/json"}
-    requests.request("PATCH", url, headers=headers, data=payload)
+    artifact_snapshot = db.collection("artifacts").document(artifactId).get()
+    if not artifact_snapshot.exists:
+        artifact_snapshot.update(body)
+    sendMessage(artifactId)
+# This function cannot be automation test because the requirement of another device to receive notification
+def sendMessage(artifactId: str, message: str = None):
+    token = []
+    artifact = db.collection("artifacts").document(artifactId).get()
+    if not artifact.exists:
+        return
+    user_ref = db.collection("user").where(filter=FieldFilter("artifacts", "array-contains", "artifacts/" + artifactId))
+    for user in user_ref:
+        token.append(user.get().to_dict()['deviceId'])
+    if message is not None:
+        messaging.MulticastMessage(data={"notification": {
+            "title": message,
+            "body":
+                "Video " +
+                artifact.name +
+                " has done inference. Click here to see the video",
+            },}, android=messaging.AndroidConfig(
+                notification=messaging.AndroidNotification(
+                    icon='stock_ticker_update',
+                    color='#f45342'
+            ),))
+    else:
+        messaging.MulticastMessage(data={"notification": {
+            "title": "Video " + artifact.name + " has done inference.",
+            "body":
+                "Video " +
+                artifact.name +
+                " has done inference. Click here to see the video",
+            },}, android=messaging.AndroidConfig(
+                notification=messaging.AndroidNotification(
+                    icon='stock_ticker_update',
+                    color='#f45342'
+            ),))
+    response = messaging.send_multicast(message)
+    return response.success_count

tests/test_friend_request.py

@@ -11,6 +11,7 @@ from app.main import app
 from app.constants import deviceId
 from fastapi.routing import APIRoute
 from firebase_admin import firestore
+from app import db
 from google.cloud.firestore_v1.base_query import FieldFilter
 def endpoints():
     endpoints = []
@@ -54,16 +55,6 @@ def inviter():
     data = response.json()
     inviter = {"id": data['localId'], "token": data["idToken"]}
     yield inviter
-@pytest.fixture()
-def firebase_app():
-    if "test" in firebase_admin._apps:
-        firebase_app = firebase_admin.get_app('test')
-        yield firebase_app
-    else:
-        firebase_app = firebase_admin.initialize_app(        credential= credentials.Certificate(
-            json.loads(os.environ.get("FIREBASE_CREDENTIALS"))
-        ),name='test')
-        yield firebase_app
     
 @pytest.fixture()
 def invitee():
@@ -83,9 +74,8 @@ def invitee():
     yield invitee
 class TestFriendRequest():
     @pytest.mark.skipif("/friend_request" not in endpoints(),reason="Route not defined")
-    def test_post_friend(self, firebase_app, client, inviter, invitee):
+    def test_post_friend(self, client, inviter, invitee):
         # Call the firebase database
-        db = firestore.client(app=firebase_app)
         friend_request_ref = db.collection('friend_request')
         # Remove all the friend_request use for testing in the past
         query = friend_request_ref.where(filter=FieldFilter("inviter", "==", inviter['id']))
@@ -160,4 +150,4 @@ class TestFriendRequest():
         # Delete entity for next time test
         user_ref.document(inviter['id']).delete()
         user_ref.document(invitee['id']).delete()
-        
+