From 67347c8c9c374fa24e9782572d17318d8a1d1454 Mon Sep 17 00:00:00 2001 From: frxz751113 <156018267+frxz751113@users.noreply.github.com> Date: Mon, 12 Aug 2024 17:19:06 +0800 Subject: [PATCH] Update iptv_udp.py --- py/iptv_udp.py | 54 +++++++++----------------------------------------- 1 file changed, 9 insertions(+), 45 deletions(-) diff --git a/py/iptv_udp.py b/py/iptv_udp.py index e9aa7cb..d0978aa 100644 --- a/py/iptv_udp.py +++ b/py/iptv_udp.py @@ -279,7 +279,7 @@ def main(): # 移除文件 # os.remove('playlist/' + file_path) # 写入合并后的txt文件 - with open("playlist/IPTV_UDP.txt", "w", encoding="utf-8") as output: + with open("playlist/IPTV_UDP", "w", encoding="utf-8") as output: output.write('\n\n'.join(file_contents)) # 写入更新日期时间 # file.write(f"{now_today}更新,#genre#\n") @@ -291,56 +291,20 @@ def main(): output.write(f"{now.strftime("%Y-%m-%d")},url\n") output.write(f"{now.strftime("%H:%M:%S")},url\n") output.close() - print(f"电视频道成功写入IPTV_UDP.txt") + print(f"电视频道成功写入IPTV_UDP") main() -######################################################################################################################################################################################## - -for line in fileinput.input("playlist/IPTV_UDP.txt", inplace=True): #打开文件，并对其进行关键词原地替换 +for line in fileinput.input("playlist/IPTV_UDP", inplace=True): #打开文件，并对其进行关键词原地替换 line = line.replace("CHC电影", "CHC影迷电影") line = line.replace("高清电影", "影迷电影") print(line, end="") #设置end=""，避免输出多余的换行符 -print(f"文件已替换完成") - -######################################################################################################################################################################################## - - - -#从整理好的文本中按类别进行特定关键词提取# -keywords = ['CCTV', '剧场', '影院', '电影'] # 需要提取的关键字列表 -pattern = '|'.join(keywords) # 创建正则表达式模式，匹配任意一个关键字 -#pattern = r"^(.*?),(?!#genre#)(.*?)$" #以分类直接复制 -with open('playlist/IPTV_UDP.txt', 'r', encoding='utf-8') as file, open('c0.txt', 'w', encoding='utf-8') as c0: #定义临时文件名 - c0.write('组播央视,#genre#\n') #写入临时文件名$GD - for line in file: - if re.search(pattern, line): # 如果行中有任意关键字 - c0.write(line) # 将该行写入输出文件 #定义临时文件 - -import re -def find_number_before_comma(line): - # 从行的开始到逗号前的部分，找到第一个数字 - before_comma = line.split(',')[0] if ',' in line else line - numbers = re.findall(r'\d+', before_comma) - return int(numbers[0]) if numbers else None -def custom_sort_key(line): - # 返回一个元组，元组的第一个元素用于数字排序，第二个元素用于字典排序 - return (find_number_before_comma(line), line) -with open('c0.txt', 'r', encoding='utf-8') as file: - lines = file.readlines() -sorted_lines = sorted(lines, key=custom_sort_key) -# 写入新文件 -with open('c00.txt', 'w', encoding='utf-8') as file: - for line in sorted_lines: - file.write(line) - - #从整理好的文本中按类别进行特定关键词提取# keywords = ['爱动漫', '爱怀旧', '爱经典', '爱科幻', '爱幼教', '爱青春', '爱院线', '爱悬疑'] # 需要提取的关键字列表 pattern = '|'.join(keywords) # 创建正则表达式模式，匹配任意一个关键字 #pattern = r"^(.*?),(?!#genre#)(.*?)$" #以分类直接复制 -with open('playlist/IPTV_UDP.txt', 'r', encoding='utf-8') as file, open('c1.txt', 'w', encoding='utf-8') as c1: #定义临时文件名 - c1.write('组播央视,#genre#\n') #写入临时文件名$GD +with open('playlist/IPTV_UDP', 'r', encoding='utf-8') as file, open('c1.txt', 'w', encoding='utf-8') as c1: #定义临时文件名 + c1.write('\niHOT系列,#genre#\n') #写入临时文件名$GD for line in file: if '$GD' not in line and '4K' not in line: if re.search(pattern, line): # 如果行中有任意关键字 @@ -353,7 +317,7 @@ keywords = ['河北', '石家庄', '丰宁', '临漳', '井陉', '井陉矿区', '邱县', '隆化', '雄县', '阜平', '高碑店', '高邑', '魏县', '黄骅', '饶阳', '赵县', '睛彩河北', '滦南', '玉田', '崇礼', '平泉', '容城', '文安', '三河', '清河'] # 需要提取的关键字列表 pattern = '|'.join(keywords) # 创建正则表达式模式，匹配任意一个关键字 #pattern = r"^(.*?),(?!#genre#)(.*?)$" #以分类直接复制 -with open('playlist/IPTV_UDP.txt', 'r', encoding='utf-8') as file, open('f.txt', 'w', encoding='utf-8') as f: #定义临时文件名 +with open('playlist/IPTV_UDP', 'r', encoding='utf-8') as file, open('f.txt', 'w', encoding='utf-8') as f: #定义临时文件名 f.write('\n河北频道,#genre#\n') #写入临时文件名 for line in file: if 'CCTV' not in line and '卫视' not in line and 'CHC' not in line and '4K' not in line and 'genre' not in line: @@ -365,7 +329,7 @@ keywords = ['河南', '焦作', '开封', '卢氏', '洛阳', '孟津', '安阳' '新密', '新乡', '信阳', '新郑', '延津', '叶县', '义马', '永城', '禹州', '原阳', '镇平', '郑州', '周口'] # 需要提取的关键字列表 pattern = '|'.join(keywords) # 创建正则表达式模式，匹配任意一个关键字 #pattern = r"^(.*?),(?!#genre#)(.*?)$" #以分类直接复制 -with open('playlist/IPTV_UDP.txt', 'r', encoding='utf-8') as file, open('f1.txt', 'w', encoding='utf-8') as f1: #定义临时文件名 +with open('playlist/IPTV_UDP', 'r', encoding='utf-8') as file, open('f1.txt', 'w', encoding='utf-8') as f1: #定义临时文件名 f1.write('\n河南频道,#genre#\n') #写入临时文件名 for line in file: if 'CCTV' not in line and '卫视' not in line and 'CHC' not in line and '4K' not in line and 'genre' not in line: @@ -465,7 +429,7 @@ print('删除的行已保存到:', deleted_lines_file_path) #合并所有频道文件# # 读取要合并的频道文件，并生成临时文件#合并所有频道文件# file_contents = [] -file_paths = ["a0.txt", "港澳.txt", "df0.txt", "c00.txt", "c1.txt", "f.txt", "f1.txt"] # 替换为实际的文件路径列表# +file_paths = ["a0.txt", "港澳.txt", "df0.txt", "f.txt", "f1.txt"] # 替换为实际的文件路径列表# for file_path in file_paths: # with open(file_path, 'r', encoding="utf-8") as file: # content = file.read() @@ -577,7 +541,7 @@ def txt_to_m3u(input_file, output_file): # 将txt文件转换为m3u文件 txt_to_m3u('综合源.txt', '综合源.m3u') #任务结束，删除不必要的过程文件# -files_to_remove = ["TW.txt", "a.txt", "a0.txt", "港澳.txt", "playlist/IPTV_UDP.txt", "df0.txt", "sr1.txt", "c0.txt", "c00.txt", "c1.txt", \ +files_to_remove = ["TW.txt", "a.txt", "a0.txt", "港澳.txt", "playlist/IPTV_UDP", "df0.txt", "sr1.txt", "c1.txt", \ "f.txt", "f1.txt", "playlist/酒店源#.txt"] for file in files_to_remove: if os.path.exists(file):