Update sync.py

sync.py

@@ -5,15 +5,15 @@ from pymongo import MongoClient
 TMDB_KEY = os.environ.get("TMDB_API_KEY")
 client = MongoClient(os.environ.get("MONGODB_URI"))
 db = client['myanmarswe_cinema']
-metas_col, streams_col = db['metas'], db['streams']
+metas_col, streams_col, users_col = db['metas'], db['streams'], db['users']
 
 def get_tmdb_meta(tmdb_id, m_type):
     tmdb_type = "movie" if m_type == "movie" else "tv"
-    url = f"https://api.themoviedb.org/3/{tmdb_type}/{tmdb_id}?api_key={TMDB_KEY}&append_to_response=credits,videos"
+    url = f"https://api.themoviedb.org/3/{tmdb_type}/{tmdb_id}?api_key={TMDB_KEY}&append_to_response=credits"
     
     try:
         res = requests.get(url).json()
-        # Cast ကို String list အဖြစ် တိုက်ရိုက်ယူခြင်း
+        # Cast ကို String list အဖြစ်ယူခြင်း (Detail Screen အတွက်)
         cast = [c["name"] for c in res.get("credits", {}).get("cast", [])[:15]]
         
         meta = {
@@ -24,18 +24,16 @@ def get_tmdb_meta(tmdb_id, m_type):
             "background": f"https://image.tmdb.org/t/p/original{res.get('backdrop_path')}",
             "description": res.get("overview"),
             "cast": cast,
-            "director": [c["name"] for c in res.get("credits", {}).get("crew", []) if c["job"] == "Director"],
             "year": (res.get("release_date") or res.get("first_air_date") or "0000")[:4],
             "updated_at": datetime.utcnow()
         }
 
-        # TV Show Episode ပုံများ ထည့်သွင်းခြင်း
+        # TV Show Episode Poster Fix
         if m_type == "series":
             videos = []
             for s in res.get("seasons", []):
                 s_num = s.get("season_number")
                 if s_num == 0: continue
-                # Episode တစ်ခုချင်းစီအတွက် Poster ယူရန်
                 for e_num in range(1, s.get("episode_count", 0) + 1):
                     videos.append({
                         "id": f"tmdb:{tmdb_id}:{s_num}:{e_num}",
@@ -45,13 +43,11 @@ def get_tmdb_meta(tmdb_id, m_type):
                     })
             meta["videos"] = videos
 
-        # Collection Data Processing
+        # Collection Logic (Backdrop and Sorting)
         if res.get("belongs_to_collection"):
             col_id = res["belongs_to_collection"]["id"]
             col_res = requests.get(f"https://api.themoviedb.org/3/collection/{col_id}?api_key={TMDB_KEY}").json()
-            
-            # Release Date အလိုက် အစီအစဉ်တကျဖြစ်အောင် စီခြင်း
-            sorted_parts = sorted(col_res.get("parts", []), key=lambda x: x.get("release_date", "9999-99-99"))
+            sorted_parts = sorted(col_res.get("parts", []), key=lambda x: x.get("release_date", "9999"))
             
             col_videos = []
             for i, p in enumerate(sorted_parts):
@@ -72,4 +68,22 @@ def get_tmdb_meta(tmdb_id, m_type):
             metas_col.update_one({"id": col_meta["id"]}, {"$set": col_meta}, upsert=True)
 
         return meta
-    except: return None
+    except: return None
+
+def process_sync(data):
+    # User data update
+    if "users" in data:
+        users_col.delete_many({})
+        users_col.insert_many(data["users"])
+
+    # Movies/Series sync
+    for item in data.get("movies_series", []):
+        meta = get_tmdb_meta(item['tmdb_id'], item['type'])
+        if meta:
+            if item.get('custom_overview'): meta['description'] = item['custom_overview']
+            metas_col.update_one({"id": meta["id"]}, {"$set": meta}, upsert=True)
+            
+            s_query = {"tmdb_id": str(item['tmdb_id']), "season": item.get('season'), "episode": item.get('episode')}
+            streams_col.update_one(s_query, {"$set": {**s_query, "url": item['stream_url']}}, upsert=True)
+            
+    return {"status": "success"}