Update sync.py

sync.py

@@ -2,37 +2,38 @@ import os, json, requests
 from datetime import datetime
 from pymongo import MongoClient
 
-# Environment Variables
 MONGO_URI = os.environ.get("MONGODB_URI")
 TMDB_KEY = os.environ.get("TMDB_API_KEY")
 
-# MongoDB Setup
 client = MongoClient(MONGO_URI)
 db = client['myanmarswe_cinema']
 metas_col = db['metas']
 streams_col = db['streams']
 users_col = db['users']
 
-def get_tmdb_data(tmdb_id, is_movie=True):
-    m_type = "movie" if is_movie else "tv"
-    # Append to response ဖြင့် Data အစုံကို တစ်ခါတည်းယူခြင်း
+def get_tmdb_data(tmdb_id, m_type):
     url = f"https://api.themoviedb.org/3/{m_type}/{tmdb_id}?api_key={TMDB_KEY}&append_to_response=credits,videos,images,recommendations,belongs_to_collection"
     
     try:
         res = requests.get(url).json()
         if "id" not in res: return None
 
-        # Trailer & Media
         trailers = [{"source": v["key"], "type": "Trailer"} for v in res.get("videos", {}).get("results", []) if v["type"] == "Trailer" and v["site"] == "YouTube"][:1]
         
-        # Runtime Error Handling
-        if is_movie:
+        # Runtime Formatting
+        if m_type == "movie":
             run_time = f"{res.get('runtime')} min"
         else:
             ert = res.get('episode_run_time', [])
             run_time = f"{ert[0]} min" if ert else "N/A"
 
-        # Ultra-Detail Meta Object
+        # Cast formatting for Stremio (Array of strings)
+        cast_list = [c["name"] for c in res.get("credits", {}).get("cast", [])[:15]]
+        
+        # Logo handling (Logo မရှိရင် Poster ကို သုံးမယ်)
+        logo_path = res.get("poster_path") 
+        # TMDB images endpoint ကနေ logo ရှာရင် ပိုကောင်းပေမယ့် လောလောဆယ် poster ကို logo fallback လုပ်ထားပါတယ်
+
         meta = {
             "_id": f"tmdb:{tmdb_id}",
             "id": f"tmdb:{tmdb_id}",
@@ -40,21 +41,20 @@ def get_tmdb_data(tmdb_id, is_movie=True):
             "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')}",
-            "logo": f"https://image.tmdb.org/t/p/original{res.get('poster_path')}", # Logo path can be refined
+            "logo": f"https://image.tmdb.org/t/p/w500{logo_path}",
             "description": res.get("overview"),
             "releaseInfo": (res.get("release_date") or res.get("first_air_date", "0000"))[:4],
-            "imdbRating": str(res.get("vote_average", "N/A")),
+            "imdbRating": "{:.1f}".format(res.get("vote_average", 0)) if res.get("vote_average") else "N/A",
             "runtime": run_time,
-            "cast": [c["name"] for c in res.get("credits", {}).get("cast", [])[:10]],
+            "cast": cast_list,
             "genres": [g["name"] for g in res.get("genres", [])],
             "trailers": trailers,
             "recommendations": [str(r["id"]) for r in res.get("recommendations", {}).get("results", [])[:10]],
             "collection_info": res.get("belongs_to_collection"),
-            "updated_at": datetime.utcnow() # နောက်ဆုံးထည့်တာ ရှေ့ဆုံးပြဖို့
+            "updated_at": datetime.utcnow()
         }
 
-        # TV Show Seasons Detail
-        if not is_movie:
+        if m_type == "tv":
             videos = []
             for s in res.get("seasons", []):
                 s_num = s.get("season_number")
@@ -63,58 +63,38 @@ def get_tmdb_data(tmdb_id, is_movie=True):
                     videos.append({
                         "id": f"tmdb:{tmdb_id}:{s_num}:{e_num}",
                         "title": f"S{s_num} E{e_num}",
-                        "season": s_num, "episode": e_num
+                        "season": s_num, "episode": e_num,
+                        "released": res.get("first_air_date") # Released date field is important for series UI
                     })
             meta["videos"] = videos
 
         return meta
-    except Exception as e:
-        print(f"TMDB Error: {e}")
-        return None
+    except: return None
 
 def process_sync(data):
-    # 1. Sync Users
-    users_list = data.get("users", [])
-    if users_list:
-        users_col.delete_many({}) # Refresh user list
-        users_col.insert_many(users_list)
+    if data.get("users"):
+        users_col.delete_many({})
+        users_col.insert_many(data["users"])
 
-    # 2. Sync Movies/Series
     items = data.get("movies_series", [])
     for item in items:
         tid = str(item['tmdb_id'])
-        is_movie = (item['type'] == 'movie')
+        m_type = item['type'] # 'movie' or 'series'
         
-        # Get Metadata from TMDB
-        full_meta = get_tmdb_data(tid, is_movie)
+        full_meta = get_tmdb_data(tid, m_type)
         if full_meta:
-            # Custom Overview Override
             if item.get('custom_overview'):
                 full_meta['description'] = item['custom_overview']
             
-            # Save Meta to MongoDB (Update if exists)
-            metas_col.update_one({"_id": full_meta["_id"]}, {"$set": full_meta}, upsert=True)
+            metas_col.update_one({"id": full_meta["id"]}, {"$set": full_meta}, upsert=True)
             
-            # Save Stream URL to Streams Collection
             stream_data = {
-                "tmdb_id": tid,
-                "type": item['type'],
-                "url": item['stream_url'],
-                "season": item.get('season'),
-                "episode": item.get('episode'),
+                "tmdb_id": tid, "type": m_type, "url": item['stream_url'],
+                "season": item.get('season'), "episode": item.get('episode'),
                 "updated_at": datetime.utcnow()
             }
-            # Stream တွေကို update လုပ်မယ် (သို့) အသစ်ထည့်မယ်
             streams_col.update_one(
                 {"tmdb_id": tid, "season": item.get('season'), "episode": item.get('episode')},
-                {"$set": stream_data},
-                upsert=True
+                {"$set": stream_data}, upsert=True
             )
-            
-            # Smart Collection Logic (If movie belongs to a set)
-            if is_movie and full_meta.get("collection_info"):
-                col_id = str(full_meta["collection_info"]["id"])
-                # Collection Meta ကိုလည်း တစ်ခါတည်း သိမ်းမယ်
-                # (ဒီနေရာမှာ Collection details တွေကို ထပ်ယူတဲ့ logic ထည့်နိုင်ပါတယ်)
-
     return {"status": "success"}