Update sync.py

sync.py

@@ -14,7 +14,17 @@ def get_tmdb_meta(tmdb_id, m_type):
     try:
         res = requests.get(url).json()
         
-        # Logo Logic
+        # Runtime Logic
+        runtime_str = ""
+        if m_type == "movie":
+            rt = res.get("runtime")
+            runtime_str = f"{rt} min" if rt else ""
+        else:
+            # TV Show အတွက် ပျမ်းမျှ မိနစ်ကို ယူသည်
+            runtimes = res.get("episode_run_time", [])
+            runtime_str = f"{runtimes[0]} min" if runtimes else ""
+
+        # Logo
         logo = None
         logos = res.get("images", {}).get("logos", [])
         if logos:
@@ -22,21 +32,17 @@ def get_tmdb_meta(tmdb_id, m_type):
             target_logo = en_logos[0] if en_logos else logos[0]
             logo = f"https://image.tmdb.org/t/p/w500{target_logo['file_path']}"
 
-        # Cast & Director (Strictly String Arrays)
+        # Cast & Director (Strict String Lists)
         credits = res.get("credits", {})
-        cast_list = [str(c["name"]) for c in credits.get("cast", [])[:15]]
+        cast = [str(c["name"]) for c in credits.get("cast", [])[:15]]
         
-        dir_list = []
+        director = []
         if m_type == "movie":
-            dir_list = [str(c["name"]) for c in credits.get("crew", []) if c.get("job") == "Director"]
+            director = [str(c["name"]) for c in credits.get("crew", []) if c.get("job") == "Director"]
         else:
-            dir_list = [str(c["name"]) for c in res.get("created_by", [])]
-            if not dir_list:
-                dir_list = [str(c["name"]) for c in credits.get("crew", []) if c.get("job") in ["Director", "Executive Producer"]][:2]
-
-        # Year (releaseInfo)
-        date_str = res.get("release_date") or res.get("first_air_date") or ""
-        year = date_str[:4] if date_str else "N/A"
+            director = [str(c["name"]) for c in res.get("created_by", [])]
+            if not director:
+                director = [str(c["name"]) for c in credits.get("crew", []) if c.get("job") == "Executive Producer"][:2]
 
         meta = {
             "id": f"tmdb:{tmdb_id}",
@@ -46,10 +52,11 @@ def get_tmdb_meta(tmdb_id, m_type):
             "background": f"https://image.tmdb.org/t/p/original{res.get('backdrop_path')}",
             "logo": logo,
             "description": res.get("overview", ""),
-            "cast": cast_list,
-            "director": dir_list,
+            "cast": cast,
+            "director": director,
+            "runtime": runtime_str, # Runtime field အသစ်
             "genres": [g["name"] for g in res.get("genres", [])],
-            "releaseInfo": year,
+            "releaseInfo": str(res.get("release_date") or res.get("first_air_date") or "0000")[:4],
             "imdbRating": str(round(res.get("vote_average", 0), 1)) if res.get("vote_average") else None,
             "updated_at": datetime.utcnow()
         }
@@ -57,15 +64,14 @@ def get_tmdb_meta(tmdb_id, m_type):
         if m_type == "series":
             videos = []
             for s in res.get("seasons", []):
-                s_num = s.get("season_number")
-                if s_num == 0: continue
-                s_url = f"https://api.themoviedb.org/3/tv/{tmdb_id}/season/{s_num}?api_key={TMDB_KEY}"
+                if s.get("season_number") == 0: continue
+                s_url = f"https://api.themoviedb.org/3/tv/{tmdb_id}/season/{s.get('season_number')}?api_key={TMDB_KEY}"
                 s_res = requests.get(s_url).json()
                 for ep in s_res.get("episodes", []):
                     videos.append({
-                        "id": f"tmdb:{tmdb_id}:{s_num}:{ep['episode_number']}",
-                        "title": ep.get("name") or f"Episode {ep['episode_number']}",
-                        "season": s_num, "episode": ep['episode_number'],
+                        "id": f"tmdb:{tmdb_id}:{s.get('season_number')}:{ep['episode_number']}",
+                        "title": ep.get("name") or f"S{s.get('season_number')} E{ep['episode_number']}",
+                        "season": s.get("season_number"), "episode": ep['episode_number'],
                         "thumbnail": f"https://image.tmdb.org/t/p/w500{ep['still_path']}" if ep.get('still_path') else meta["poster"]
                     })
             meta["videos"] = videos
@@ -77,22 +83,26 @@ def process_sync(data):
         users_col.delete_many({})
         users_col.insert_many(data["users"])
 
+    processed_metas = {}
     for item in data.get("movies_series", []):
         t_id, m_type = str(item['tmdb_id']), item['type']
         meta_id = f"tmdb:{t_id}"
         
-        new_meta = get_tmdb_meta(t_id, m_type)
-        if new_meta:
-            # Custom Overview Logic
-            if item.get('custom_overview') and str(item['custom_overview']).strip():
-                new_meta['description'] = str(item['custom_overview'])
-            
-            metas_col.update_one({"id": meta_id}, {"$set": new_meta}, upsert=True)
+        if meta_id not in processed_metas:
+            new_meta = get_tmdb_meta(t_id, m_type)
+            if new_meta:
+                # Custom overview maintain logic
+                existing = metas_col.find_one({"id": meta_id})
+                if item.get('custom_overview'):
+                    new_meta['description'] = str(item['custom_overview'])
+                elif existing and existing.get('is_custom_overview'):
+                    new_meta['description'] = existing['description']
+                
+                processed_metas[meta_id] = new_meta
         
-        # Stream sync
-        s_query = {"tmdb_id": t_id}
-        if m_type == "series":
-            s_query.update({"season": int(item.get('season', 1)), "episode": int(item.get('episode', 1))})
-        streams_col.update_one(s_query, {"$set": {**s_query, "url": item['stream_url'], "updated_at": datetime.utcnow()}}, upsert=True)
+        s_query = {"tmdb_id": t_id, "season": item.get('season'), "episode": item.get('episode')}
+        streams_col.update_one(s_query, {"$set": {**s_query, "url": item['stream_url']}}, upsert=True)
     
+    for m_id, m_data in processed_metas.items():
+        metas_col.update_one({"id": m_id}, {"$set": m_data}, upsert=True)
     return {"status": "success"}