From c52d01ccd6081d3cf263df1d2562bda1053f39d6 Mon Sep 17 00:00:00 2001
From: frxz751113 <156018267+frxz751113@users.noreply.github.com>
Date: Mon, 12 Aug 2024 18:24:52 +0800
Subject: [PATCH] Update iptv-jd.py

---
 py/iptv-jd.py | 82 +++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 82 insertions(+)

diff --git a/py/iptv-jd.py b/py/iptv-jd.py
index a7998e3..9966f62 100644
--- a/py/iptv-jd.py
+++ b/py/iptv-jd.py
@@ -1466,3 +1466,85 @@ for file in files_to_remove:
     else:              # 如果文件不存在，则提示异常并打印提示信息
         print(f"文件 {file} 不存在，跳过删除。")
 print("任务运行完毕，分类频道列表可查看文件夹内综合源.txt文件！")
+
+
+import cv2
+import time
+from tqdm import tqdm
+
+# 初始化检测结果字典
+detected_ips = {}
+
+# 存储文件路径
+file_path = "酒店源.txt"
+output_file_path = "流畅源.txt"
+
+def get_ip_key(url):
+    """从URL中提取IP地址，并构造一个唯一的键"""
+    # 找到'//'到第三个'.'之间的字符串
+    start = url.find('://') + 3  # '://'.length 是 3
+    end = start
+    dot_count = 0
+    while dot_count < 3:
+        end = url.find('.', end)
+        if end == -1:  # 如果没有找到第三个'.'，就结束
+            break
+        dot_count += 1
+    return url[start:end] if dot_count == 3 else None
+
+# 打开输入文件和输出文件
+with open(file_path, 'r', encoding='utf-8') as file:
+    lines = file.readlines()
+
+# 获取总行数用于进度条
+total_lines = len(lines)
+
+# 写入通过检测的行到新文件
+with open(output_file_path, 'w', encoding='utf-8') as output_file:
+    # 使用tqdm显示进度条
+    for i, line in tqdm(enumerate(lines), total=total_lines, desc="Processing", unit='line'):
+        # 检查是否包含 'genre'
+        if 'genre' in line:
+            output_file.write(line)
+            continue
+
+        # 分割频道名称和URL，并去除空白字符
+        parts = line.split(',', 1)
+        if len(parts) == 2:
+            channel_name, url = parts
+            channel_name = channel_name.strip()
+            url = url.strip()
+
+            # 构造IP键
+            ip_key = get_ip_key(url)
+            if ip_key and ip_key in detected_ips:
+                # 如果IP键已存在，根据之前的结果决定是否写入新文件
+                if detected_ips[ip_key]['status'] == 'ok':
+                    output_file.write(line)
+            elif ip_key:  # 新IP键，进行检测
+                # 进行检测
+                cap = cv2.VideoCapture(url)
+                start_time = time.time()
+                frame_count = 0
+
+                # 尝试捕获5秒内的帧
+                while frame_count < 60 and (time.time() - start_time) < 5:
+                    ret, frame = cap.read()
+                    if not ret:
+                        break
+                    frame_count += 1
+
+                # 释放资源
+                cap.release()
+
+                # 根据捕获的帧数判断状态并记录结果
+                if frame_count >= 50:
+                    detected_ips[ip_key] = {'status': 'ok'}
+                    output_file.write(line)  # 写入检测通过的行
+                else:
+                    detected_ips[ip_key] = {'status': 'fail'}
+
+# 打印检测结果
+for ip_key, result in detected_ips.items():
+    print(f"IP Key: {ip_key}, Status: {result['status']}")
+