Update sync.py

sync.py

@@ -9,13 +9,12 @@ 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"
-    # credits နှင့် images ကို append_to_response ဖြင့် တစ်ခါတည်း ယူသည်
     url = f"https://api.themoviedb.org/3/{tmdb_type}/{tmdb_id}?api_key={TMDB_KEY}&append_to_response=credits,images"
     
     try:
         res = requests.get(url).json()
         
-        # Logo ရယူခြင်း
+        # Logo
         logo = None
         logos = res.get("images", {}).get("logos", [])
         if logos:
@@ -23,21 +22,22 @@ 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 ရယူခြင်း
+        # Cast & Director (Array of Strings)
         credits = res.get("credits", {})
-        cast = [c["name"] for c in credits.get("cast", [])[:15]]
+        cast = [str(c["name"]) for c in credits.get("cast", [])[:10]] # Top 10 cast
         
-        director = []
+        directors = []
         if m_type == "movie":
-            director = [c["name"] for c in credits.get("crew", []) if c.get("job") == "Director"]
+            directors = [str(c["name"]) for c in credits.get("crew", []) if c.get("job") == "Director"]
         else:
-            director = [c["name"] for c in res.get("created_by", [])]
-            if not director: 
-                director = [c["name"] for c in credits.get("crew", []) if c.get("job") == "Executive Producer"][:2]
+            directors = [str(c["name"]) for c in res.get("created_by", [])]
+            if not directors:
+                directors = [str(c["name"]) for c in credits.get("crew", []) if c.get("job") in ["Executive Producer", "Director"]][:2]
+
+        # Year (releaseInfo)
+        raw_date = res.get("release_date") or res.get("first_air_date") or ""
+        year = str(raw_date)[:4] if raw_date else ""
 
-        genres = [g["name"] for g in res.get("genres", [])]
-        vote_avg = res.get("vote_average")
-        
         meta = {
             "id": f"tmdb:{tmdb_id}",
             "type": m_type,
@@ -47,11 +47,10 @@ def get_tmdb_meta(tmdb_id, m_type):
             "logo": logo,
             "description": res.get("overview"),
             "cast": cast,
-            "director": director,
-            "genres": genres,
-            "releaseInfo": str(res.get("release_date") or res.get("first_air_date") or "0000")[:4],
-            "imdbRating": str(round(vote_avg, 1)) if vote_avg else None,
-            "runtime": f"{res.get('runtime')} min" if res.get('runtime') else None,
+            "director": directors,
+            "genres": [g["name"] for g in res.get("genres", [])],
+            "releaseInfo": year, # Stremio အတွက် Year field
+            "imdbRating": str(round(res.get("vote_average", 0), 1)) if res.get("vote_average") else None,
             "updated_at": datetime.utcnow()
         }
 
@@ -60,24 +59,20 @@ def get_tmdb_meta(tmdb_id, m_type):
             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}"
                 s_res = requests.get(s_url).json()
-                
                 for ep in s_res.get("episodes", []):
-                    e_num = ep.get("episode_number")
                     videos.append({
-                        "id": f"tmdb:{tmdb_id}:{s_num}:{e_num}",
-                        "title": ep.get("name") or f"S{s_num} E{e_num}",
-                        "season": s_num, "episode": e_num,
-                        "thumbnail": f"https://image.tmdb.org/t/p/w500{ep.get('still_path')}" if ep.get('still_path') else meta["poster"],
+                        "id": f"tmdb:{tmdb_id}:{s_num}:{ep['episode_number']}",
+                        "title": ep.get("name") or f"S{s_num} E{ep['episode_number']}",
+                        "season": s_num, "episode": ep['episode_number'],
+                        "thumbnail": f"https://image.tmdb.org/t/p/w500{ep['still_path']}" if ep.get('still_path') else meta["poster"],
                         "overview": ep.get("overview", "")
                     })
             meta["videos"] = videos
-
         return meta
-    except Exception as e: 
-        print(f"TMDB Fetch Error: {e}")
+    except Exception as e:
+        print(f"Error fetching TMDB {tmdb_id}: {e}")
         return None
 
 def process_sync(data):
@@ -85,42 +80,24 @@ 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 = str(item['tmdb_id'])
-        m_type = item['type']
+        t_id, m_type = str(item['tmdb_id']), item['type']
         meta_id = f"tmdb:{t_id}"
         
-        # Meta အချက���အလက်ကို TMDB မှ ဆွဲယူသည်
-        if meta_id not in processed_metas:
-            new_meta = get_tmdb_meta(t_id, m_type)
-            if new_meta:
-                # Custom Overview စစ်ဆေးခြင်း
-                # Spreadsheet မှ custom_overview ပါလာလျှင် TMDB description အစား ၎င်းကို သုံးမည်
-                custom_ov = item.get('custom_overview')
-                if custom_ov and str(custom_ov).strip():
-                    new_meta['description'] = str(custom_ov).strip()
-                    new_meta['is_custom_overview'] = True
-                else:
-                    new_meta['is_custom_overview'] = False
-                
-                processed_metas[meta_id] = new_meta
+        # Meta ကို အမြဲ Update လုပ်ရန် (Cast/Director အသစ်ဝင်စေရန်)
+        new_meta = get_tmdb_meta(t_id, m_type)
+        if new_meta:
+            # Custom Overview ရှိမရှိ စစ်ဆေးခြင်း
+            if item.get('custom_overview'):
+                new_meta['description'] = str(item['custom_overview'])
+            
+            metas_col.update_one({"id": meta_id}, {"$set": new_meta}, upsert=True)
         
-        # Stream link ကို အမြဲ update လုပ်သည်
+        # Stream update
         s_query = {"tmdb_id": t_id}
         if m_type == "series":
-            s_query["season"] = int(item.get('season', 1))
-            s_query["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.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)
     
-    # Meta data အားလုံးကို MongoDB တွင် သိမ်းဆည်းသည်
-    for m_id, m_data in processed_metas.items():
-        metas_col.update_one({"id": m_id}, {"$set": m_data}, upsert=True)
-
     return {"status": "success"}