From 50143448eb2d554a846870e4dad22ec8683cf072 Mon Sep 17 00:00:00 2001
From: frxz751113 <156018267+frxz751113@users.noreply.github.com>
Date: Fri, 9 Aug 2024 12:08:57 +0800
Subject: [PATCH] Create iptv-ckeck.py

---
 py/iptv-ckeck.py | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)
 create mode 100644 py/iptv-ckeck.py

diff --git a/py/iptv-ckeck.py b/py/iptv-ckeck.py
new file mode 100644
index 0000000..fd82161
--- /dev/null
+++ b/py/iptv-ckeck.py
@@ -0,0 +1,34 @@
+import requests
+from tqdm import tqdm
+
+# 要检测的文件和结果文件的文件名
+input_file = '综合源.txt'
+output_file = 'valid_sources.txt'
+
+def is_link_valid(url):
+    try:
+        response = requests.head(url, allow_redirects=True, timeout=5)
+        # 检查HTTP状态码是否表示成功
+        return 200 <= response.status_code < 300
+    except requests.RequestException as e:
+        print(f"请求错误: {e}")
+        return False
+
+def process_file(input_filename, output_filename):
+    with open(input_filename, 'r', encoding='utf-8') as file:
+        lines = file.readlines()
+
+    # 使用tqdm管理进度条
+    with tqdm(total=len(lines)) as progress:
+        with open(output_filename, 'w', encoding='utf-8') as output_file:
+            for line in lines:
+                if line.strip() and not line.startswith('#'):
+                    url = line.strip()
+                    # 更新进度条，leave=True参数确保进度条不会在每次迭代时刷新
+                    progress.update(1)
+                    if is_link_valid(url):
+                        output_file.write(line)
+
+# 执行函数
+process_file(input_file, output_file)
+print(f"有效链接已写入到 {output_file}")