Update sync.py

sync.py

@@ -1,16 +1,5 @@
-import os, json, requests
-from datetime import datetime
-from pymongo import MongoClient
-
-MONGO_URI = os.environ.get("MONGODB_URI")
-TMDB_KEY = os.environ.get("TMDB_API_KEY")
-
-client = MongoClient(MONGO_URI)
-db = client['myanmarswe_cinema']
-metas_col, streams_col, users_col = db['metas'], db['streams'], db['users']
-
+# sync.py ထဲက get_tmdb_data function ကို အခုလို အစားထိုးပါ
 def get_tmdb_data(tmdb_id, m_type):
-    # m_type က sheet ထဲမှာ series ဖြစ်နေရင် tmdb အတွက် tv လို့ ပြောင်းပေးရမယ်
     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,images,recommendations,belongs_to_collection"
     
@@ -18,26 +7,23 @@ def get_tmdb_data(tmdb_id, m_type):
         res = requests.get(url).json()
         if "id" not in res: return None
 
-        # --- Smart Logo Finder ---
+        # --- Logo Search ---
         logo_url = None
-        images = res.get("images", {})
-        logos = images.get("logos", [])
+        logos = res.get("images", {}).get("logos", [])
         if logos:
-            # အင်္ဂလိပ်စာတန်းထိုးပါတဲ့ logo ကို အရင်ရှာမယ်
             en_logos = [l for l in logos if l.get("iso_639_1") == "en"]
             logo_path = en_logos[0]["file_path"] if en_logos else logos[0]["file_path"]
             logo_url = f"https://image.tmdb.org/t/p/w500{logo_path}"
 
-        # --- Cast & Director (For Details Screen) ---
+        # --- Cast & Crew (Stremio UI အတွက် အရေးကြီးသည်) ---
         credits = res.get("credits", {})
-        cast = [c["name"] for c in credits.get("cast", [])[:10]]
-        # Stremio UI မှာ Cast ပေါ်ဖို့ director ပါရှိတာ ပိုစိတ်ချရပါတယ်
+        cast = [c["name"] for c in credits.get("cast", [])[:15]] # ပထမ ၁၅ ယောက်
         directors = [c["name"] for c in credits.get("crew", []) if c["job"] == "Director"]
 
         meta = {
             "_id": f"tmdb:{tmdb_id}",
             "id": f"tmdb:{tmdb_id}",
-            "type": m_type, # 'movie' or 'series'
+            "type": m_type,
             "name": res.get("title") or res.get("name"),
             "poster": f"https://image.tmdb.org/t/p/w500{res.get('poster_path')}",
             "background": f"https://image.tmdb.org/t/p/original{res.get('backdrop_path')}",
@@ -55,11 +41,12 @@ def get_tmdb_data(tmdb_id, m_type):
             "updated_at": datetime.utcnow()
         }
 
+        # --- TV Shows Episodes Logic (Fix for Detail Screen) ---
         if m_type == "series":
             videos = []
             for s in res.get("seasons", []):
                 s_num = s.get("season_number")
-                if s_num == 0: continue
+                if s_num == 0: continue # Specials ကို ကျော်မယ်
                 for e_num in range(1, s.get("episode_count", 0) + 1):
                     videos.append({
                         "id": f"tmdb:{tmdb_id}:{s_num}:{e_num}",
@@ -69,26 +56,24 @@ def get_tmdb_data(tmdb_id, m_type):
                     })
             meta["videos"] = videos
 
-        return meta
-    except Exception as e:
-        print(f"Sync Error: {e}")
-        return None
-
-def process_sync(data):
-    if data.get("users"):
-        users_col.delete_many({})
-        users_col.insert_many(data["users"])
+        # --- Smart Collection Processing (New) ---
+        if meta.get("collection_info"):
+            col_id = meta["collection_info"]["id"]
+            col_url = f"https://api.themoviedb.org/3/collection/{col_id}?api_key={TMDB_KEY}"
+            col_res = requests.get(col_url).json()
+            if "id" in col_res:
+                col_meta = {
+                    "_id": f"tmdb_col:{col_id}",
+                    "id": f"tmdb_col:{col_id}",
+                    "type": "movie",
+                    "name": col_res.get("name"),
+                    "poster": f"https://image.tmdb.org/t/p/w500{col_res.get('poster_path')}",
+                    "background": f"https://image.tmdb.org/t/p/original{col_res.get('backdrop_path')}",
+                    "description": col_res.get("overview"),
+                    "videos": [{"id": f"tmdb:{p['id']}", "title": p.get("title")} for p in col_res.get("parts", [])],
+                    "updated_at": datetime.utcnow()
+                }
+                metas_col.update_one({"id": col_meta["id"]}, {"$set": col_meta}, upsert=True)
 
-    items = data.get("movies_series", [])
-    for item in items:
-        tid, m_type = str(item['tmdb_id']), item['type']
-        full_meta = get_tmdb_data(tid, m_type)
-        if full_meta:
-            if item.get('custom_overview'): full_meta['description'] = item['custom_overview']
-            metas_col.update_one({"id": full_meta["id"]}, {"$set": full_meta}, upsert=True)
-            
-            # Stream URL Update
-            stream_query = {"tmdb_id": tid, "season": item.get('season'), "episode": item.get('episode')}
-            stream_data = {**stream_query, "type": m_type, "url": item['stream_url'], "updated_at": datetime.utcnow()}
-            streams_col.update_one(stream_query, {"$set": stream_data}, upsert=True)
-    return {"status": "success"}
+        return meta
+    except: return None