Update iptv-check.py

This commit is contained in:
frxz751113
2024-08-10 03:15:22 +08:00
committed by GitHub
parent 7ace6baf90
commit 0833de8b46
-24
View File
@@ -542,30 +542,6 @@ def filter_lines(input_file, output_file):
filter_lines("iptv.txt", "iptv.txt") filter_lines("iptv.txt", "iptv.txt")
print(f"文件已过滤完成") print(f"文件已过滤完成")
######################################################################################################################################################################################## ########################################################################################################################################################################################
#################文本排序
# 打开原始文件读取内容,并写入新文件
with open('iptv.txt', 'r', encoding='utf-8') as file:
lines = file.readlines()
# 定义一个函数,用于提取每行的第一个数字
def extract_first_number(line):
match = re.search(r'\d+', line)
return int(match.group()) if match else float('inf')
# 对列表中的行进行排序
# 按照第一个数字的大小排列,如果不存在数字则按中文拼音排序
sorted_lines = sorted(lines, key=lambda x: (not 'CCTV' in x, extract_first_number(x) if 'CCTV' in x else lazy_pinyin(x.strip())))
# 将排序后的行写入新的utf-8编码的文本文件,文件名基于原文件名
output_file_path = "sorted_" + os.path.basename(file_path)
# 写入新文件
with open('iptv.txt', "w", encoding="utf-8") as file:
for line in sorted_lines:
file.write(line)
print(f"文件已排序并保存为: {output_file_path}")
######################################################################################################################################################################################## ########################################################################################################################################################################################