Create sync.py

sync.py

@@ -0,0 +1,120 @@
+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 အစုံကို တစ်ခါတည်းယူခြင်း
+    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:
+            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
+        meta = {
+            "_id": f"tmdb:{tmdb_id}",
+            "id": f"tmdb:{tmdb_id}",
+            "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')}",
+            "logo": f"https://image.tmdb.org/t/p/original{res.get('poster_path')}", # Logo path can be refined
+            "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")),
+            "runtime": run_time,
+            "cast": [c["name"] for c in res.get("credits", {}).get("cast", [])[:10]],
+            "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() # နောက်ဆုံးထည့်တာ ရှေ့ဆုံးပြဖို့
+        }
+
+        # TV Show Seasons Detail
+        if not is_movie:
+            videos = []
+            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):
+                    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
+                    })
+            meta["videos"] = videos
+
+        return meta
+    except Exception as e:
+        print(f"TMDB Error: {e}")
+        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)
+
+    # 2. Sync Movies/Series
+    items = data.get("movies_series", [])
+    for item in items:
+        tid = str(item['tmdb_id'])
+        is_movie = (item['type'] == 'movie')
+        
+        # Get Metadata from TMDB
+        full_meta = get_tmdb_data(tid, is_movie)
+        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)
+            
+            # 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'),
+                "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
+            )
+            
+            # 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"}