Update iptv-ckeck.py
This commit is contained in:
+12
-7
@@ -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
|
||||||
if is_link_valid(url):
|
url = url.strip() # 移除URL两端的空白字符
|
||||||
output_file.write(line)
|
if is_link_valid(url):
|
||||||
|
output_file.write(line)
|
||||||
|
progress.update(1) # 更新进度条
|
||||||
|
|
||||||
# 执行函数
|
# 执行函数
|
||||||
process_file(input_file, output_file)
|
process_file(input_file, output_file)
|
||||||
|
|||||||
Reference in New Issue
Block a user