Update sync.py

sync.py

@@ -1,4 +1,4 @@
-import os, json, requests
+import os, requests
 from datetime import datetime
 from pymongo import MongoClient
 
@@ -7,131 +7,94 @@ 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']
+metas_col = db['metas']
 
 def get_tmdb_data(tmdb_id, m_type):
     tmdb_type = "movie" if m_type == "movie" else "tv"
+    # Episode Poster အတွက် images ပါ ထပ်တောင်းထားပါတယ်
     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"
     
     try:
         res = requests.get(url).json()
         if "id" not in res: return None
 
-        logo_url = None
-        images = res.get("images", {})
-        logos = images.get("logos", [])
-        if logos:
-            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 & Crew Explicit Data Casting Fix ---
+        # Cast List - String သီးသန့်ဖြစ်ရပါမယ်
         credits = res.get("credits", {})
-        cast_list = credits.get("cast", [])
-        cast = [str(c.get("name")) for c in cast_list[:15] if c.get("name")]
-        
-        crew_list = credits.get("crew", [])
-        directors = [str(c.get("name")) for c in crew_list if c.get("job") == "Director" and c.get("name")]
-        
-        if m_type == "series" and not directors:
-            creator_list = res.get("created_by", [])
-            directors = [str(c.get("name")) for c in creator_list if c.get("name")]
-
-        run_t = res.get("runtime") if m_type == "movie" else (res.get("episode_run_time", [None])[0] if res.get("episode_run_time") else None)
-        runtime_str = f"{run_t} min" if run_t else "N/A"
+        cast = [str(c.get("name")) for c in credits.get("cast", [])[:15]]
+        directors = [str(c.get("name")) for c in credits.get("crew", []) if c.get("job") == "Director"]
 
         meta = {
             "_id": f"tmdb:{tmdb_id}",
             "id": f"tmdb:{tmdb_id}",
             "type": m_type,
-            "name": res.get("title") or res.get("name") or "Unknown",
-            "poster": f"https://image.tmdb.org/t/p/w500{res.get('poster_path')}" if res.get('poster_path') else None,
-            "background": f"https://image.tmdb.org/t/p/original{res.get('backdrop_path')}" if res.get('backdrop_path') else None,
-            "logo": logo_url,
+            "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')}",
             "description": res.get("overview") or "",
             "releaseInfo": str((res.get("release_date") or res.get("first_air_date", "0000"))[:4]),
-            "imdbRating": "{:.1f}".format(res.get("vote_average", 0)) if res.get("vote_average") else "N/A",
-            "runtime": runtime_str,
-            "cast": cast if cast else ["Unknown Actor"],  # Error ကာကွယ်ရန် Fallback ထည့်ထားခြင်း
-            "director": directors if directors else ["Unknown Director"],
+            "imdbRating": str(res.get("vote_average", "N/A")),
+            "cast": cast if cast else ["N/A"],
+            "director": directors if directors else ["N/A"],
             "genres": [g["name"] for g in res.get("genres", [])],
-            "trailers": [{"source": v["key"], "type": "Trailer"} for v in res.get("videos", {}).get("results", []) if v["type"] == "Trailer" and v["site"] == "YouTube"][:1],
-            "recommendations": [str(r["id"]) for r in res.get("recommendations", {}).get("results", [])[:10]],
-            "collection_info": res.get("belongs_to_collection"),
             "updated_at": datetime.utcnow()
         }
 
+        # Series Episode logic with Posters
         if m_type == "series":
             videos = []
-            first_air = res.get("first_air_date")
-            rel_date = f"{first_air}T00:00:00.000Z" if first_air else "2000-01-01T00:00:00.000Z"
-
             for s in res.get("seasons", []):
                 s_num = s.get("season_number")
                 if s_num == 0: continue
-                for e_num in range(1, s.get("episode_count", 0) + 1):
+                # Episode list fetching for thumbnails
+                ep_url = f"https://api.themoviedb.org/3/tv/{tmdb_id}/season/{s_num}?api_key={TMDB_KEY}"
+                ep_res = requests.get(ep_url).json()
+                for ep in ep_res.get("episodes", []):
                     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,
-                        "released": rel_date
+                        "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'],
+                        "released": f"{ep.get('air_date')}T00:00:00.000Z",
+                        "thumbnail": f"https://image.tmdb.org/t/p/w500{ep.get('still_path')}" if ep.get("still_path") else meta["background"]
                     })
             meta["videos"] = videos
 
-        # --- Smart Collection Processing with Strict Series ID Format ---
-        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_videos = []
-                for i, p in enumerate(col_res.get("parts", [])):
-                    p_date = p.get("release_date")
-                    c_rel_date = f"{p_date}T00:00:00.000Z" if p_date else "2000-01-01T00:00:00.000Z"
-                    col_videos.append({
-                        "id": f"tmdb_col:{col_id}:1:{i+1}", # Stremio Series Standard အတိုင်း ID Format ပြောင်းလဲခြင်း
-                        "movie_id": str(p['id']), # Stream ရှာရန် မူရင်း Movie ID ကို ခေတ္တသိမ်းထားခြင်း
-                        "title": p.get('title') or f"Part {i+1}", 
-                        "season": 1, 
-                        "episode": i+1,
-                        "released": c_rel_date
-                    })
+        # Collection Logic - နှစ်အလိုက် Sort စီခြင်း
+        if res.get("belongs_to_collection"):
+            col_id = res["belongs_to_collection"]["id"]
+            c_url = f"https://api.themoviedb.org/3/collection/{col_id}?api_key={TMDB_KEY}"
+            c_res = requests.get(c_url).json()
+            
+            # Release Date အလိုက် Sort စီပါမယ်
+            sorted_parts = sorted(c_res.get("parts", []), key=lambda x: x.get("release_date", "9999-99-99"))
+            
+            col_videos = []
+            for i, p in enumerate(sorted_parts):
+                col_videos.append({
+                    "id": f"tmdb_col:{col_id}:1:{i+1}",
+                    "movie_id": str(p['id']), # Link ချိတ်ရန် ID သိမ်းထားခြင်း
+                    "title": f"({p.get('release_date', '')[:4]}) {p.get('title')}",
+                    "season": 1,
+                    "episode": i+1,
+                    "released": f"{p.get('release_date')}T00:00:00.000Z",
+                    "thumbnail": f"https://image.tmdb.org/t/p/w500{p.get('backdrop_path')}" if p.get('backdrop_path') else None
+                })
 
-                col_meta = {
-                    "_id": f"tmdb_col:{col_id}",
-                    "id": f"tmdb_col:{col_id}",
-                    "type": "series", 
-                    "name": col_res.get("name") or "Collection",
-                    "poster": f"https://image.tmdb.org/t/p/w500{col_res.get('poster_path')}" if col_res.get('poster_path') else "https://via.placeholder.com/500x750?text=No+Poster",
-                    "background": f"https://image.tmdb.org/t/p/original{col_res.get('backdrop_path')}" if col_res.get('backdrop_path') else "https://via.placeholder.com/1280x720?text=No+Background",
-                    "description": col_res.get("overview") or "No description available.",
-                    "cast": [], 
-                    "director": [],
-                    "genres": ["Collection"],
-                    "videos": col_videos,
-                    "updated_at": datetime.utcnow()
-                }
-                metas_col.update_one({"id": col_meta["id"]}, {"$set": col_meta}, upsert=True)
+            col_meta = {
+                "_id": f"tmdb_col:{col_id}",
+                "id": f"tmdb_col:{col_id}",
+                "type": "series",
+                "name": c_res.get("name"),
+                "poster": f"https://image.tmdb.org/t/p/w500{c_res.get('poster_path')}",
+                "background": f"https://image.tmdb.org/t/p/original{c_res.get('backdrop_path')}",
+                "description": c_res.get("overview") or f"All movies in {c_res.get('name')}",
+                "genres": ["Collection"],
+                "videos": col_videos,
+                "updated_at": datetime.utcnow()
+            }
+            metas_col.update_one({"id": col_meta["id"]}, {"$set": col_meta}, upsert=True)
 
         return meta
-    except Exception as e: 
-        print(f"Error fetching TMDB: {e}")
-        return None
-
-def process_sync(data):
-    if data.get("users"):
-        users_col.delete_many({})
-        users_col.insert_many(data["users"])
-
-    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)
-            
-            s_query = {"tmdb_id": tid, "season": item.get('season'), "episode": item.get('episode')}
-            s_data = {**s_query, "type": m_type, "url": item['stream_url'], "updated_at": datetime.utcnow()}
-            streams_col.update_one(s_query, {"$set": s_data}, upsert=True)
-    return {"status": "success"}
+    except Exception as e:
+        print(f"Sync Error: {e}")
+        return None