Update iptv-ckeck.py

This commit is contained in:
frxz751113
2024-08-09 12:17:37 +08:00
committed by GitHub
parent fd7de4d8ca
commit 77a113e309
+10 -5
View File
@@ -5,12 +5,14 @@ from tqdm import tqdm
input_file = '综合源.txt' input_file = '综合源.txt'
output_file = 'valid_sources.txt' output_file = 'valid_sources.txt'
def is_link_valid(url): def is_link_valid(url):
try: try:
response = requests.head(url, allow_redirects=True, timeout=5) response = requests.head(url, allow_redirects=True, timeout=5)
# 检查HTTP状态码是否表示成功 # 检查HTTP状态码是否表示成功
return 200 <= response.status_code < 308 return 200 <= response.status_code < 300
except requests.RequestException as e: except requests.RequestException as e:
# 打印错误信息,但继续执行
print(f"请求错误: {e}") print(f"请求错误: {e}")
return False return False
@@ -19,15 +21,18 @@ def process_file(input_filename, output_filename):
lines = file.readlines() lines = file.readlines()
# 使用tqdm管理进度条 # 使用tqdm管理进度条
with tqdm(total=len(lines)) as progress: with tqdm(total=len(lines), desc="Processing lines") as progress:
with open(output_filename, 'w', encoding='utf-8') as output_file: with open(output_filename, 'w', encoding='utf-8') as output_file:
for line in lines: for line in lines:
# 忽略空行和以#开头的注释行
if line.strip() and not line.startswith('#'): if line.strip() and not line.startswith('#'):
url = line.strip() parts = line.split(',', 1) # 只分割第一部分
# 更新进度条,leave=True参数确保进度条不会在每次迭代时刷新 if len(parts) == 2:
progress.update(1) channel_name, url = parts
url = url.strip() # 移除URL两端的空白字符
if is_link_valid(url): if is_link_valid(url):
output_file.write(line) output_file.write(line)
progress.update(1) # 更新进度条
# 执行函数 # 执行函数
process_file(input_file, output_file) process_file(input_file, output_file)