Add slack notifications

app.py

@@ -5,6 +5,7 @@ import pandas as pd
 from bson import ObjectId
 from dotenv import load_dotenv
 import requests
+from datetime import timedelta
 from collections import defaultdict
 
 load_dotenv()
@@ -49,6 +50,45 @@ language_maps = {
 }
 
 
+def print_slack(platform, video_id, translator_name, assignment="translator"):
+    if platform == "dev":
+        return
+    else:
+        client = pymongo.MongoClient(os.environ.get("MONGO_URI"))
+        source_db = client.deepsync_prod
+    
+    headers = {
+        'Content-type': 'application/json',
+    }
+
+    video = source_db.dubbedvideos.find_one({"_id": ObjectId(video_id)})
+    video_name = video['title']
+    source_language = language_maps[video['sourceLanguage'][:2]]
+    target_language = language_maps[video['targetLanguage'][:2]]
+    video_duration = str(timedelta(seconds=video['duration']))
+    translator_details = "/".join(translator_name.split("/")[:3])
+
+    message = f"""```Platform : {platform}
+{'Translator' if assignment == 'translator' else 'QA'} Assigned
+Video Name : {video_name}
+Source Language : {source_language}
+Target Language : {target_language}
+Video Duration : {video_duration} hours
+Details : {translator_details}
+```
+"""
+
+    json_data = {
+        'text': message,
+    }
+
+    response = requests.post(
+        os.environ.get('SLACK_WEBHOOK'),
+        headers=headers,
+        json=json_data,
+    )
+
+
 def get_translators(platform):
     if platform == "dev":
         api_url = "https://api.dev-env.deepsync.co/api/v1"
@@ -339,6 +379,7 @@ def assign_translator(platform, translator, video):
     }
     response = requests.post(f"{api_url}/dashboard/dubbed/video/assign/{role}", json=data, headers=HEADERS)
     if str(response.status_code).startswith('2'):
+        print_slack(platform, video_id, translator, "translator")
         return "Successfully assigned."
     else:
         return str(response.status_code)
@@ -358,6 +399,7 @@ def assign_qa(platform, qa, video):
     }
     response = requests.post(f"{api_url}/dashboard/dubbed/video/assign/{role}", json=data, headers=HEADERS)
     if str(response.status_code).startswith('2'):
+        print_slack(platform, video_id, qa, "qa")
         return "Successfully assigned."
     else:
         return str(response.status_code)