mirror of
https://github.com/plsy1/iptv.git
synced 2026-11-06 00:56:43 +08:00
58 lines
1.7 KiB
Python
58 lines
1.7 KiB
Python
import os
|
|
import glob
|
|
|
|
def merge_m3u(input_files, output_file,search_group_title="山东频道"):
|
|
first_file = input_files[0]
|
|
if not os.path.exists(first_file):
|
|
print(f"文件 '{first_file}' 不存在，跳过。")
|
|
return
|
|
|
|
with open(first_file, "r") as input:
|
|
lines = input.readlines()
|
|
|
|
insert_position = -1
|
|
for i in range(len(lines) - 1, -1, -1):
|
|
if f'group-title="{search_group_title}"' in lines[i]:
|
|
insert_position = i + 2
|
|
break
|
|
|
|
if insert_position == -1:
|
|
print(f"在文件 '{first_file}' 中没有找到 group-title='{search_group_title}'。")
|
|
insert_position = len(lines)
|
|
|
|
second_file = input_files[1]
|
|
if not os.path.exists(second_file):
|
|
print(f"文件 '{second_file}' 不存在，跳过。")
|
|
return
|
|
|
|
with open(second_file, "r") as input:
|
|
second_file_lines = input.readlines()
|
|
|
|
with open(output_file, "w") as output:
|
|
output.writelines(lines[:insert_position])
|
|
output.writelines(second_file_lines)
|
|
output.write("\n")
|
|
output.writelines(lines[insert_position:])
|
|
|
|
print(f"合并完成，输出到 {output_file}")
|
|
|
|
|
|
m3u_files = glob.glob("multicast/local/*.m3u")
|
|
|
|
city_dict = {}
|
|
for filepath in m3u_files:
|
|
city = os.path.splitext(os.path.basename(filepath))[0]
|
|
if city not in city_dict:
|
|
city_dict[city] = []
|
|
city_dict[city].append(filepath)
|
|
|
|
multicast_dir = "multicast"
|
|
output_dir = "multicast"
|
|
for city, append_files in city_dict.items():
|
|
multicast_file = os.path.join(multicast_dir, f"multicast-{city}.m3u")
|
|
output_file = os.path.join(output_dir, f"multicast-{city}.m3u")
|
|
all_files = [multicast_file] + append_files
|
|
merge_m3u(all_files,output_file)
|
|
|
|
|