Upload vtv_api.py

vtv_api.py

@@ -13,14 +13,18 @@ UA = {
     "Accept-Language": "vi-VN,vi;q=0.9",
 }
 
-# ===== NGUỒN PHÁT TỪ Love4vn M3U (FLV/HLS) =====
+# ===== NGUỒN PHÁT: HARDCODE VTV6 + M3U fallback cho các kênh khác =====
+HARDCODED_URLS = {
+    "vtv6": "https://live.fptplay53.net/live/media/v6abr/live247-hls-avc/v6abr-avc1_5600000=10000-mp4a_131600=20000.m3u8",
+}
+
 CHANNEL_URLS = {
     "vtv1":  None,
     "vtv2":  None,
     "vtv3":  None,
     "vtv4":  None,
     "vtv5":  None,
-    "vtv6":  None,
+    "vtv6":  HARDCODED_URLS["vtv6"],  # hardcode VTV6
     "vtv7":  None,
     "vtv8":  None,
     "vtv9":  None,
@@ -87,7 +91,12 @@ def _fetch_m3u():
                     }
                     if ch_id in ch_map:
                         ch_id = ch_map[ch_id]
+                    # Chỉ cập nhật nếu chưa có hoặc hardcode không ghi đè
                     if ch_id in CHANNEL_URLS and ch_id not in all_urls:
+                        # KHÔNG ghi đè hardcode (VTV6)
+                        if CHANNEL_URLS[ch_id] and CHANNEL_URLS[ch_id] == HARDCODED_URLS.get(ch_id):
+                            current_channel = None
+                            continue
                         # Ưu tiên FLV hơn HLS/m3u8
                         if ".flv" in s:
                             all_urls[ch_id] = s
@@ -103,10 +112,13 @@ def _fetch_m3u():
     with _channel_urls_lock:
         found = 0
         for ch_id in CHANNEL_URLS:
+            # KHÔNG ghi đè hardcode
+            if CHANNEL_URLS[ch_id] and HARDCODED_URLS.get(ch_id):
+                continue
             if ch_id in all_urls:
                 CHANNEL_URLS[ch_id] = all_urls[ch_id]
                 found += 1
-        print(f"[M3U] Found {found}/10 channels")
+        print(f"[M3U] Found {found}/10 channels (hardcoded: vtv6)")
 
 def get_channel_url(channel_id):
     _fetch_m3u()
@@ -124,7 +136,7 @@ def proxy_stream(url: str = Query(...)):
     """Proxy FLV/HLS stream — tránh CORS & geo-block"""
     try:
         req_headers = dict(UA)
-        req_headers["Referer"] = "https://k2s.tv/"
+        req_headers["Referer"] = "https://live.fptplay53.net/"
         r = requests.get(url, headers=req_headers, timeout=30, stream=True)
         if r.status_code != 200:
             return Response(content=f"stream error: {r.status_code}", status_code=502)
@@ -159,7 +171,7 @@ def proxy_flv(url: str = Query(...)):
     """Proxy FLV stream với Range support"""
     try:
         req_headers = dict(UA)
-        req_headers["Referer"] = "https://k2s.tv/"
+        req_headers["Referer"] = "https://live.fptplay53.net/"
         r = requests.get(url, headers=req_headers, timeout=30, stream=True)
         if r.status_code != 200:
             return Response(content=f"flv error: {r.status_code}", status_code=502)