frxz751113/iptv
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update iptv流畅度检测.py

Browse Source
This commit is contained in:
frxz751113
2024-08-15 13:21:31 +08:00
committed by GitHub
parent 7fd464d3f4
commit 59c2823b06
1 changed files with 93 additions and 93 deletions
Download Patch File Download Diff File Expand all files Collapse all files
py/iptv流畅度检测.py
+93 -93
View File
@@ -777,98 +777,98 @@ from opencc import OpenCC
import base64 import base64
import cv2 import cv2
#################################################### 对整理好的频道列表测试HTTP连接 #################################################### 对整理好的频道列表测试HTTP连接
# 函数：获取视频分辨率\\ # 函数：获取视频分辨率#$
def get_video_resolution(video_path, timeout=2):\\ def get_video_resolution(video_path, timeout=2):#$
# 使用OpenCV创建视频捕获对象\\ # 使用OpenCV创建视频捕获对象#$
cap = cv2.VideoCapture(video_path)\\ cap = cv2.VideoCapture(video_path)#$
# 检查视频是否成功打开\\ # 检查视频是否成功打开#$
if not cap.isOpened():\\ if not cap.isOpened():#$
return None\\ return None#$
# 获取视频的宽度和高度\\ # 获取视频的宽度和高度#$
width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))\\ width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))#$
height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))\\ height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))#$
# 释放视频捕获对象\\ # 释放视频捕获对象#$
cap.release()\\ cap.release()#$
# 返回视频的分辨率\\ # 返回视频的分辨率#$
return (width, height)\\ return (width, height)#$
\\ #$
# 函数：处理每一行\\ # 函数：处理每一行#$
def process_line(line, output_file, order_list, valid_count, invalid_count, total_lines):\\ def process_line(line, output_file, order_list, valid_count, invalid_count, total_lines):#$
# 去除行尾的空白字符并按逗号分割行\\ # 去除行尾的空白字符并按逗号分割行#$
parts = line.strip().split(',')\\ parts = line.strip().split(',')#$
# 如果行包含特定的标签'#genre#'，则直接写入新文件\\ # 如果行包含特定的标签'#genre#'，则直接写入新文件#$
if '#genre#' in line:\\ if '#genre#' in line:#$
with threading.Lock(): # 使用线程锁保证写入操作的原子性\\ with threading.Lock(): # 使用线程锁保证写入操作的原子性#$
output_file.write(line)\\ output_file.write(line)#$
print(f"已写入genre行：{line.strip()}")\\ print(f"已写入genre行：{line.strip()}")#$
# 如果分割后的部分数量为2，则继续处理\\ # 如果分割后的部分数量为2，则继续处理#$
elif len(parts) == 2:\\ elif len(parts) == 2:#$
channel_name, channel_url = parts\\ channel_name, channel_url = parts#$
# 获取视频的分辨率\\ # 获取视频的分辨率#$
resolution = get_video_resolution(channel_url, timeout=2)\\ resolution = get_video_resolution(channel_url, timeout=2)#$
# 如果分辨率有效且高度大于等于720p\\ # 如果分辨率有效且高度大于等于720p#$
if resolution and resolution[1] >= 720:\\ if resolution and resolution[1] >= 720:#$
with threading.Lock(): # 使用线程锁\\ with threading.Lock(): # 使用线程锁#$
output_file.write(f"{channel_name}[{resolution[1]}p],{channel_url}\n")\\ output_file.write(f"{channel_name}[{resolution[1]}p],{channel_url}\n")#$
# 将频道名、分辨率和URL添加到列表中\\ # 将频道名、分辨率和URL添加到列表中#$
order_list.append((channel_name, resolution[1], channel_url))\\ order_list.append((channel_name, resolution[1], channel_url))#$
# 有效计数增加\\ # 有效计数增加#$
valid_count[0] += 1\\ valid_count[0] += 1#$
print(f"Channel '{channel_name}' accepted with resolution {resolution[1]}p at URL {channel_url}.")\\ print(f"Channel '{channel_name}' accepted with resolution {resolution[1]}p at URL {channel_url}.")#$
else:\\ else:#$
# 如果分辨率不满足条件，无效计数增加\\ # 如果分辨率不满足条件，无效计数增加#$
invalid_count[0] += 1\\ invalid_count[0] += 1#$
# 打印当前处理进度\\ # 打印当前处理进度#$
with threading.Lock():\\ with threading.Lock():#$
print(f"有效: {valid_count[0]}, 无效: {invalid_count[0]}, 总数: {total_lines}, 进度: {(valid_count[0] + invalid_count[0]) / total_lines * 100:.2f}%")\\ print(f"有效: {valid_count[0]}, 无效: {invalid_count[0]}, 总数: {total_lines}, 进度: {(valid_count[0] + invalid_count[0]) / total_lines * 100:.2f}%")#$
\\ #$
# 函数：多线程工作\\ # 函数：多线程工作#$
def worker(task_queue, output_file, order_list, valid_count, invalid_count, total_lines):\\ def worker(task_queue, output_file, order_list, valid_count, invalid_count, total_lines):#$
# 循环直到队列为空\\ # 循环直到队列为空#$
while True:\\ while True:#$
try:\\ try:#$
# 从队列中获取任务，超时时间为1秒\\ # 从队列中获取任务，超时时间为1秒#$
line = task_queue.get(timeout=1)\\ line = task_queue.get(timeout=1)#$
# 处理获取的任务\\ # 处理获取的任务#$
process_line(line, output_file, order_list, valid_count, invalid_count, total_lines)\\ process_line(line, output_file, order_list, valid_count, invalid_count, total_lines)#$
except Queue.Empty: # 如果队列为空，捕获异常\\ except Queue.Empty: # 如果队列为空，捕获异常#$
break\\ break#$
finally:\\ finally:#$
# 标记任务已完成\\ # 标记任务已完成#$
task_queue.task_done()\\ task_queue.task_done()#$
\\ #$
# 主函数\\ # 主函数#$
def main(source_file_path, output_file_path):\\ def main(source_file_path, output_file_path):#$
# 初始化列表和计数器\\ # 初始化列表和计数器#$
order_list = []\\ order_list = []#$
valid_count = [0]\\ valid_count = [0]#$
invalid_count = [0]\\ invalid_count = [0]#$
task_queue = Queue()\\ task_queue = Queue()#$
# 使用with语句打开源文件并读取所有行\\ # 使用with语句打开源文件并读取所有行#$
with open(source_file_path, 'r', encoding='utf-8') as source_file:\\ with open(source_file_path, 'r', encoding='utf-8') as source_file:#$
lines = source_file.readlines()\\ lines = source_file.readlines()#$
# 使用with语句打开输出文件准备写入\\ # 使用with语句打开输出文件准备写入#$
with open(output_file_path, 'w', encoding='utf-8') as output_file:\\ with open(output_file_path, 'w', encoding='utf-8') as output_file:#$
# 创建线程池，最大工作线程数为64\\ # 创建线程池，最大工作线程数为64#$
with ThreadPoolExecutor(max_workers=264) as executor:\\ with ThreadPoolExecutor(max_workers=264) as executor:#$
# 为线程池中的每个线程提交worker函数\\ # 为线程池中的每个线程提交worker函数#$
for _ in range(264):\\ for _ in range(264):#$
executor.submit(worker, task_queue, output_file, order_list, valid_count, invalid_count, len(lines))\\ executor.submit(worker, task_queue, output_file, order_list, valid_count, invalid_count, len(lines))#$
# 将所有行放入任务队列\\ # 将所有行放入任务队列#$
for line in lines:\\ for line in lines:#$
task_queue.put(line)\\ task_queue.put(line)#$
# 等待队列中的所有任务完成\\ # 等待队列中的所有任务完成#$
task_queue.join()\\ task_queue.join()#$
# 打印任务完成的统计信息\\ # 打印任务完成的统计信息#$
print(f"任务完成，有效频道数：{valid_count[0]}, 无效频道数：{invalid_count[0]}, 总频道数：{len(lines)}")\\ print(f"任务完成，有效频道数：{valid_count[0]}, 无效频道数：{invalid_count[0]}, 总频道数：{len(lines)}")#$
\\ #$
# 程序入口点\\ # 程序入口点#$
if __name__ == "__main__":\\ if __name__ == "__main__":#$
# 定义源文件和输出文件的路径\\ # 定义源文件和输出文件的路径#$
source_file_path = '酒店优选.txt' # 替换为你的源文件路径\\ source_file_path = '酒店优选.txt' # 替换为你的源文件路径#$
output_file_path = '酒店优选.txt' # 替换为你的输出文件路径\\ output_file_path = '酒店优选.txt' # 替换为你的输出文件路径#$
# 调用主函数\\ # 调用主函数#$
main(source_file_path, output_file_path)\\ main(source_file_path, output_file_path)#$
####################### 提示用户输入文件名（拖入文件操作）打开用户指定的文件对不规范频道名再次替换 ####################### 提示用户输入文件名（拖入文件操作）打开用户指定的文件对不规范频道名再次替换
@@ -1116,7 +1116,7 @@ for line in lines:
with open('酒店优选.txt', 'w', encoding="utf-8") as file: with open('酒店优选.txt', 'w', encoding="utf-8") as file:
file.writelines(unique_lines) file.writelines(unique_lines)
#任务结束,删除不必要的过程文件 #任务结束,删除不必要的过程文件
files_to_remove = ['去重.txt', "2.txt", "iptv.txt", "e.txt", "a0.txt", "a.txt", "a1.txt", "b.txt", "c.txt", "c1.txt", "c2.txt", "d.txt", "f.txt", "o1.txt", "o.txt", "酒店源.txt"] files_to_remove = ['去重.txt', "2.txt", "iptv.txt", "e.txt", "a0.txt", "a.txt", "a1.txt", "b.txt", "c.txt", "c1.txt", "c2.txt", "d.txt", "f.txt", "o1.txt", "o.txt", "酒店源#.txt"]
for file in files_to_remove: for file in files_to_remove:
if os.path.exists(file): if os.path.exists(file):
os.remove(file) os.remove(file)