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