Upload vtv_api.py

vtv_api.py

@@ -1,6 +1,7 @@
 """
 VTV Channels API - Backend endpoints for VTV1-VTV10 + VTVPrime
 Fetches stream URLs from xemtv.us PHP endpoints (primary)
+Alternative: xemtivitop.com (backup)
 Fallback: FPTPlay CDN → VTVGo CDN → xemtv.net (legacy)
 EPG data scraped from https://vtv.vn/lich-phat-song.htm
 """
@@ -35,6 +36,20 @@ XEMTV_US_ENDPOINTS = {
     "vtvprime": "https://xemtv.us/tv/vtvprime.php",
 }
 
+# ===== ALTERNATIVE: xemtivitop.com (backup source) =====
+XEMTIVITOP_ENDPOINTS = {
+    "vtv1":  "https://www.xemtivitop.com/2018/10/vtv1-online.html?m=1",
+    "vtv2":  "https://www.xemtivitop.com/2018/10/vtv2-online.html?m=1",
+    "vtv3":  "https://www.xemtivitop.com/2018/10/vtv3-online.html?m=1",
+    "vtv4":  "https://www.xemtivitop.com/2018/10/vtv4-online.html?m=1",
+    "vtv5":  "https://www.xemtivitop.com/2018/10/vtv5-online.html?m=1",
+    "vtv6":  "https://www.xemtivitop.com/2018/10/vtv6-online.html?m=1",
+    "vtv7":  "https://www.xemtivitop.com/2018/10/vtv7-online.html?m=1",
+    "vtv8":  "https://www.xemtivitop.com/2018/10/vtv8-online.html?m=1",
+    "vtv9":  "https://www.xemtivitop.com/2018/10/vtv9-online.html?m=1",
+    "vtv10": "https://www.xemtivitop.com/2020/05/vtv10-can-tho-online.html?m=1",
+}
+
 # ===== LEGACY: xemtv.net (may return 403, keep as last resort) =====
 XEMTV_LEGACY_ENDPOINTS = {
     "vtv1":  "https://hd.xemtv.net/kenh/vtv1.php",
@@ -135,6 +150,22 @@ def fetch_xemtv_us_stream(channel_id):
         pass
     return None
 
+def fetch_xemtivitop_stream(channel_id):
+    """Fetch stream from xemtivitop.com blogspot pages"""
+    page_url = XEMTIVITOP_ENDPOINTS.get(channel_id)
+    if not page_url:
+        return None
+    try:
+        headers = {**UA, "Referer": "https://www.xemtivitop.com/"}
+        r = requests.get(page_url, headers=headers, timeout=15, allow_redirects=True, verify=False)
+        if r.status_code == 200:
+            m3u8 = extract_m3u8_from_html(r.text)
+            if m3u8:
+                return m3u8
+    except:
+        pass
+    return None
+
 def fetch_xemtv_legacy_stream(channel_id):
     php_url = XEMTV_LEGACY_ENDPOINTS.get(channel_id)
     if not php_url:
@@ -181,7 +212,6 @@ def fetch_vtvgo_stream(channel_id):
     return None
 
 def normalize_fptplay_url(url):
-    """Replace old/broken FPTPlay URLs with new working ones"""
     if not url:
         return url
     old_to_new = {
@@ -213,12 +243,6 @@ def normalize_fptplay_url(url):
     return old_to_new.get(url, url)
 
 def fetch_vtv_stream(channel_id):
-    """Fetch VTV stream with multi-source fallback chain:
-    1. xemtv.us (primary - new domain, most reliable)
-    2. FPTPlay CDN (fallback - new URLs)
-    3. VTVGo CDN (fallback)
-    4. xemtv.net legacy (last resort)
-    """
     channel_id = channel_id.lower().strip()
     name_map = {
         'vtvct': 'vtv10', 'vtv-can-tho': 'vtv10', 'vtv can tho': 'vtv10',
@@ -233,32 +257,34 @@ def fetch_vtv_stream(channel_id):
         return cached
 
     if channel_id == 'vtvprime':
-        url = fetch_xemtv_us_stream('vtvprime') or fetch_xemtv_legacy_stream('vtvprime')
+        url = fetch_xemtv_us_stream('vtvprime') or fetch_xemtivitop_stream('vtvprime') or fetch_xemtv_legacy_stream('vtvprime')
         if url:
             url = normalize_fptplay_url(url)
         _set_cache(channel_id, url)
         return url
 
-    # Source 1: xemtv.us (primary)
     url = fetch_xemtv_us_stream(channel_id)
     if url:
         url = normalize_fptplay_url(url)
         _set_cache(channel_id, url)
         return url
 
-    # Source 2: FPTPlay CDN
+    url = fetch_xemtivitop_stream(channel_id)
+    if url:
+        url = normalize_fptplay_url(url)
+        _set_cache(channel_id, url)
+        return url
+
     url = fetch_fptplay_stream(channel_id)
     if url:
         _set_cache(channel_id, url)
         return url
 
-    # Source 3: VTVGo CDN
     url = fetch_vtvgo_stream(channel_id)
     if url:
         _set_cache(channel_id, url)
         return url
 
-    # Source 4: xemtv.net legacy (last resort)
     url = fetch_xemtv_legacy_stream(channel_id)
     if url:
         url = normalize_fptplay_url(url)
@@ -314,8 +340,8 @@ def proxy_vtv_m3u8(url: str = Query(...)):
             return Response(status_code=502, content="upstream error")
         content = r.text
         lines = content.split('\n')
+        base_url = url.rsplit('/', 1)[0] + '/' if '/' in url else url
         rewritten = []
-        base_url = url.rsplit('/', 1)[0] + '/'
         for line in lines:
             line = line.strip()
             if not line or line.startswith('#'):
@@ -351,7 +377,7 @@ def proxy_vtv_segment(url: str = Query(...)):
 
 _epg_cache = {}
 _epg_cache_time = 0
-_EPG_CACHE_TTL = 600  # Giảm từ 30 phút xuống 10 phút, đảm bảo dữ liệu mới
+_EPG_CACHE_TTL = 600
 
 VTV_CHANNEL_MAP = {
     "vtv1": "vtv1", "vtv2": "vtv2", "vtv3": "vtv3", "vtv4": "vtv4",
@@ -366,7 +392,6 @@ def _fetch_epg_from_vtv():
     if _epg_cache and now_ts - _epg_cache_time < _EPG_CACHE_TTL:
         return _epg_cache
     epg_data = {}
-    # Lấy thời gian VN hiện tại để truyền vào parse_time
     now_vn = datetime.now(VN_TZ)
     try:
         headers = {
@@ -411,7 +436,6 @@ def _fetch_epg_from_vtv():
                     title = title_span.get_text(strip=True)
                 if not time_str or not title:
                     continue
-                # Truyền reference_date để xử lý quy tắc ngày truyền hình VTV
                 start_dt = _parse_time(time_str, reference_date=now_vn)
                 if not start_dt:
                     continue
@@ -433,16 +457,6 @@ def _fetch_epg_from_vtv():
     return epg_data
 
 def _parse_time(time_str, reference_date=None):
-    """
-    Parse giờ từ lịch phát sóng VTV (đã là giờ VN UTC+7) sang datetime có timezone.
-    
-    VTV hiển thị lịch theo ngày dương lịch (không phải ngày truyền hình).
-    Ví dụ: Lịch ngày 17/06 sẽ hiển thị tất cả chương trình từ 00:00 đến 23:59 ngày 17/06.
-    
-    Logic:
-    - Giờ 00:00-04:59: Có thể là đêm khuya của ngày hôm trước HOẶC sáng sớm của ngày mới
-    - Giờ 05:00-23:59: Luôn thuộc ngày hiện tại
-    """
     if not time_str:
         return None
     time_str = time_str.strip().replace("h", ":").replace("H", ":")
@@ -450,35 +464,20 @@ def _parse_time(time_str, reference_date=None):
     if m:
         try:
             hour, minute = int(m.group(1)), int(m.group(2))
-            
             if reference_date:
                 base_date = reference_date
             else:
-                # Lấy ngày hiện tại theo giờ VN (UTC+7)
                 now_vn = datetime.now(VN_TZ)
                 base_date = now_vn
-            
             from datetime import timedelta
-            
-            # Xác định ngày cho giờ program
             if hour < 5:
-                # Giờ 00:00-04:59: Cần xem giờ hiện tại để quyết định
                 if base_date.hour < 5:
-                    # Nếu hiện tại cũng < 5:00 (đang trong khoảng sáng sớm)
-                    # → Giờ program thuộc CÙNG NGÀY với giờ hiện tại
                     tv_date = base_date
                 else:
-                    # Nếu hiện tại >= 5:00 (đã qua sáng sớm)
-                    # → Giờ 00:00-04:59 thuộc NGÀY HÔM TRƯỚC
                     tv_date = base_date - timedelta(days=1)
             else:
-                # Giờ 05:00-23:59: Luôn thuộc ngày hiện tại
                 tv_date = base_date
-            
-            # Tạo datetime với giờ từ lịch
             result = tv_date.replace(hour=hour, minute=minute, second=0, microsecond=0)
-            
-            # Đảm bảo result có timezone
             if result.tzinfo is None:
                 result = result.replace(tzinfo=VN_TZ)
             return result
@@ -494,18 +493,13 @@ def _get_epg_for_channel(channel_id):
     now = datetime.now(VN_TZ)
     today = now.date()
     result = []
-    
-    # Lọc chỉ lấy programs của ngày hôm nay (theo lịch VTV)
     today_programmes = []
     for p in programmes:
         start_dt = p.get("start_dt")
         if start_dt and start_dt.date() == today:
             today_programmes.append(p)
-    
-    # Nếu không có programs cho hôm nay, dùng tất cả (fallback)
     if not today_programmes:
         today_programmes = programmes
-    
     for i, p in enumerate(today_programmes):
         start_dt = p.get("start_dt")
         stop_dt = None
@@ -543,4 +537,4 @@ def api_vtv_epg_refresh():
     return JSONResponse({
         "status": "refreshed", "channels": len(epg_data),
         "total_programmes": sum(len(v) for v in epg_data),
-    })
+    })