frxz751113
666 lines
26 KiB
Python
666 lines
26 KiB
Python
from lxml import etree
|
||
import time
|
||
import datetime
|
||
from datetime import datetime, timedelta # 确保 timedelta 被导入
|
||
import concurrent.futures
|
||
from selenium import webdriver
|
||
from selenium.webdriver.chrome.options import Options
|
||
from concurrent.futures import ThreadPoolExecutor
|
||
import requests
|
||
import re
|
||
import os
|
||
import threading
|
||
from queue import Queue
|
||
import queue
|
||
from datetime import datetime
|
||
import fileinput
|
||
from tqdm import tqdm
|
||
from pypinyin import lazy_pinyin
|
||
from opencc import OpenCC
|
||
import base64
|
||
import cv2
|
||
from bs4 import BeautifulSoup
|
||
from urllib.parse import urlparse
|
||
from translate import Translator # 导入Translator类,用于文本翻译
|
||
## 定义txt文件的URL列表
|
||
urls = [
|
||
'https://dimaston.github.io/live.m3u', #假m3u
|
||
'https://raw.githubusercontent.com/gaotianliuyun/gao/master/list.txt', #暂时保留
|
||
'http://117.72.68.25:9230/latest.txt',
|
||
'http://ttkx.live:55/lib/kx2024.txt',
|
||
'https://raw.githubusercontent.com/kimwang1978/tvbox/main/%E5%A4%A9%E5%A4%A9%E5%BC%80%E5%BF%83/lives/%E2%91%AD%E5%BC%80%E5%BF%83%E7%BA%BF%E8%B7%AF.txt',
|
||
'https://raw.githubusercontent.com/Fairy8o/IPTV/main/PDX-V4.txt',
|
||
'http://gg.gg/cctvgg',
|
||
'https://raw.githubusercontent.com/ddhola/file/d7afb504b1ba4fef31813e1166cb892215a9c063/0609test',
|
||
'https://raw.githubusercontent.com/vbskycn/iptv/2738b3bec8c298f57e0e2052b155846ab6ea3787/dsyy/hd.txt',
|
||
'https://raw.githubusercontent.com/frxz751113/AAAAA/main/IPTV/TW.txt',
|
||
'https://raw.githubusercontent.com/ljlfct01/ljlfct01.github.io/main/list.%E8%87%AA%E7%94%A8',
|
||
'https://notabug.org/qcfree/TVBox-api/raw/main/live.txt', #通用源
|
||
'https://47.93.222.173/gonglijun010/ys0826/-/raw/05f5829362d550d96c6c3196b3451b6a3effde67/fmml_ipv4_0206.txt?inline=false',
|
||
'https://notabug.org/bxtv/a/raw/master/315.txt',
|
||
'https://raw.githubusercontent.com/xiangjiao169/yf2025/main/yf169.txt',
|
||
'http://ttkx.live:55/lib/kx2024.txt', #天天开心
|
||
'https://raw.githubusercontent.com/KAN314go/A/e81a1c22cd1b9f459bc363bd916c13133e235510/tv/%E5%AE%89%E5%8D%9A8K.txt',
|
||
'',
|
||
'',
|
||
'',
|
||
'',
|
||
'',
|
||
'',
|
||
'',
|
||
'',
|
||
'',
|
||
'',
|
||
'',
|
||
'',
|
||
'',
|
||
'',
|
||
'',
|
||
'',
|
||
'',
|
||
'',
|
||
'',
|
||
'',
|
||
'',
|
||
'',
|
||
'',
|
||
''
|
||
]
|
||
# 合并文件的函数
|
||
def merge_txt_files(urls, output_filename='汇总.txt'):
|
||
try:
|
||
with open(output_filename, 'w', encoding='utf-8') as outfile:
|
||
for url in urls:
|
||
try:
|
||
response = requests.get(url)
|
||
response.raise_for_status() # 确保请求成功
|
||
# 尝试将响应内容解码为UTF-8，如果失败则尝试其他编码
|
||
try:
|
||
content = response.content.decode('utf-8')
|
||
except UnicodeDecodeError:
|
||
content = response.content.decode('gbk') # 尝试GBK编码
|
||
outfile.write(content + '\n')
|
||
except requests.RequestException as e:
|
||
print(f'Error downloading {url}: {e}')
|
||
except IOError as e:
|
||
print(f'Error writing to file: {e}')
|
||
|
||
# 调用函数
|
||
merge_txt_files(urls)
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
#简体转繁体
|
||
# 创建一个OpenCC对象,指定转换的规则为繁体字转简体字
|
||
converter = OpenCC('t2s.json')#繁转简
|
||
#converter = OpenCC('s2t.json')#简转繁
|
||
# 打开txt文件
|
||
with open('汇总.txt', 'r', encoding='utf-8') as file:
|
||
traditional_text = file.read()
|
||
# 进行繁体字转简体字的转换
|
||
simplified_text = converter.convert(traditional_text)
|
||
# 将转换后的简体字写入txt文件
|
||
with open('汇总.txt', 'w', encoding='utf-8') as file:
|
||
file.write(simplified_text)
|
||
|
||
|
||
|
||
with open('汇总.txt', 'r', encoding="utf-8") as file:
|
||
# 读取所有行并存储到列表中
|
||
lines = file.readlines()
|
||
#定义替换规则的字典对频道名替换
|
||
replacements = {
|
||
"CCTV-1高清测试": "",
|
||
"CCTV-2高清测试": "",
|
||
"CCTV-7高清测试": "",
|
||
"CCTV-10高清测试": "",
|
||
"中央": "CCTV",
|
||
"高清""": "",
|
||
"HD": "",
|
||
"标清": "",
|
||
"amc": "AMC",
|
||
"CCTV1综合": "CCTV1",
|
||
"CCTV2财经": "CCTV2",
|
||
"CCTV3综艺": "CCTV3",
|
||
"国际": "",
|
||
"5体育": "5",
|
||
"6电影": "6",
|
||
"军农": "",
|
||
"8影视": "8",
|
||
"9纪录": "9",
|
||
"0科教": "0",
|
||
"2社会与法": "2",
|
||
"3新闻": "3",
|
||
"4少儿": "4",
|
||
"5音乐": "5",
|
||
"": "",
|
||
"": "",
|
||
"": "",
|
||
"": "",
|
||
"": "",
|
||
"": "",
|
||
"咪咕": "",
|
||
"": "",
|
||
"超清": "",
|
||
"频道": "",
|
||
"CCTV-": "CCTV",
|
||
"CCTV_": "CCTV",
|
||
" ": "",
|
||
"CCTV风云剧场": "风云剧场",
|
||
"CCTV第一剧场": "第一剧场",
|
||
"CCTV怀旧剧场": "怀旧剧场",
|
||
"熊猫影院": "熊猫电影",
|
||
"熊猫爱生活": "熊猫生活",
|
||
"爱宠宠物": "宠物生活",
|
||
"[ipv6]": "",
|
||
"专区": "",
|
||
"卫视超": "卫视",
|
||
"CCTV风云剧场": "风云剧场",
|
||
"CCTV第一剧场": "第一剧场",
|
||
"CCTV怀旧剧场": "怀旧剧场",
|
||
"IPTV": "",
|
||
"PLUS": "+",
|
||
"＋": "+",
|
||
"(": "",
|
||
")": "",
|
||
"CAV": "",
|
||
"美洲": "",
|
||
"北美": "",
|
||
"12M": "",
|
||
"高清测试CCTV-1": "",
|
||
"高清测试CCTV-2": "",
|
||
"高清测试CCTV-7": "",
|
||
"高清测试CCTV-10": "",
|
||
"LD": "",
|
||
"HEVC20M": "",
|
||
"S,": ",",
|
||
"测试": "",
|
||
"CCTW": "CCTV",
|
||
"试看": "",
|
||
"测试": "",
|
||
" ": "",
|
||
"测试cctv": "CCTV",
|
||
"CCTV1综合": "CCTV1",
|
||
"CCTV2财经": "CCTV2",
|
||
"CCTV3综艺": "CCTV3",
|
||
"CCTV4国际": "CCTV4",
|
||
"CCTV4中文国际": "CCTV4",
|
||
"CCTV4欧洲": "CCTV4",
|
||
"CCTV5体育": "CCTV5",
|
||
"CCTV5+体育": "CCTV5+",
|
||
"CCTV6电影": "CCTV6",
|
||
"CCTV7军事": "CCTV7",
|
||
"CCTV7军农": "CCTV7",
|
||
"CCTV7农业": "CCTV7",
|
||
"CCTV7国防军事": "CCTV7",
|
||
"CCTV8电视剧": "CCTV8",
|
||
"CCTV8影视": "CCTV8",
|
||
"CCTV8纪录": "CCTV9",
|
||
"CCTV9记录": "CCTV9",
|
||
"CCTV9纪录": "CCTV9",
|
||
"CCTV10科教": "CCTV10",
|
||
"CCTV11戏曲": "CCTV11",
|
||
"CCTV12社会与法": "CCTV12",
|
||
"CCTV13新闻": "CCTV13",
|
||
"CCTV新闻": "CCTV13",
|
||
"CCTV14少儿": "CCTV14",
|
||
"央视14少儿": "CCTV14",
|
||
"CCTV少儿超": "CCTV14",
|
||
"CCTV15音乐": "CCTV15",
|
||
"CCTV音乐": "CCTV15",
|
||
"CCTV16奥林匹克": "CCTV16",
|
||
"CCTV17农业农村": "CCTV17",
|
||
"CCTV17军农": "CCTV17",
|
||
"CCTV17农业": "CCTV17",
|
||
"CCTV5+体育赛视": "CCTV5+",
|
||
"CCTV5+赛视": "CCTV5+",
|
||
"CCTV5+体育赛事": "CCTV5+",
|
||
"CCTV5+赛事": "CCTV5+",
|
||
"CCTV5+体育": "CCTV5+",
|
||
"CCTV5赛事": "CCTV5+",
|
||
"凤凰中文台": "凤凰中文",
|
||
"凤凰资讯台": "凤凰资讯",
|
||
"(CCTV4K测试）": "CCTV4K",
|
||
"上海东方卫视": "上海卫视",
|
||
"东方卫视": "上海卫视",
|
||
"内蒙卫视": "内蒙古卫视",
|
||
"福建东南卫视": "东南卫视",
|
||
"广东南方卫视": "南方卫视",
|
||
"湖南金鹰卡通": "金鹰卡通",
|
||
"炫动卡通": "哈哈炫动",
|
||
"卡酷卡通": "卡酷少儿",
|
||
"卡酷动画": "卡酷少儿",
|
||
"BRTVKAKU少儿": "卡酷少儿",
|
||
"优曼卡通": "优漫卡通",
|
||
"优曼卡通": "优漫卡通",
|
||
"嘉佳卡通": "佳嘉卡通",
|
||
"世界地理": "地理世界",
|
||
"CCTV世界地理": "地理世界",
|
||
"BTV北京卫视": "北京卫视",
|
||
"BTV冬奥纪实": "冬奥纪实",
|
||
"东奥纪实": "冬奥纪实",
|
||
"卫视台": "卫视",
|
||
"湖南电视台": "湖南卫视",
|
||
"少儿科教": "少儿",
|
||
"影视剧": "影视",
|
||
"电视剧": "影视",
|
||
"CCTV1CCTV1": "CCTV1",
|
||
"CCTV2CCTV2": "CCTV2",
|
||
"CCTV7CCTV7": "CCTV7",
|
||
"CCTV10CCTV10": "CCTV10"
|
||
}
|
||
with open('汇总.txt', 'w', encoding='utf-8') as new_file:
|
||
for line in lines:
|
||
# 去除行尾的换行符
|
||
line = line.rstrip('\n')
|
||
# 分割行，获取逗号前的字符串
|
||
parts = line.split(',', 1)
|
||
if len(parts) > 0:
|
||
# 替换逗号前的字符串
|
||
before_comma = parts[0]
|
||
for old, new in replacements.items():
|
||
before_comma = before_comma.replace(old, new)
|
||
# 将替换后的逗号前部分和逗号后部分重新组合成一行，并写入新文件
|
||
new_line = f'{before_comma},{parts[1]}\n' if len(parts) > 1 else f'{before_comma}\n'
|
||
new_file.write(new_line)
|
||
|
||
|
||
|
||
|
||
|
||
|
||
# 打开文本文件进行读取
|
||
def read_and_process_file(input_filename, output_filename, encodings=['utf-8', 'gbk']):
|
||
for encoding in encodings:
|
||
try:
|
||
with open(input_filename, 'r', encoding=encoding) as file:
|
||
lines = file.readlines()
|
||
break
|
||
except UnicodeDecodeError:
|
||
continue
|
||
else:
|
||
raise ValueError(f"Cannot decode file '{input_filename}' with any of the provided encodings")
|
||
|
||
with open(output_filename, 'w', encoding='utf-8') as outfile:
|
||
for line in lines:
|
||
if '$' in line:
|
||
processed_line = line.split('$')[0].rstrip('\n')
|
||
outfile.write(processed_line + '\n')
|
||
else:
|
||
outfile.write(line)
|
||
|
||
# 调用函数
|
||
read_and_process_file('汇总.txt', '汇总.txt') # 修改输出文件名以避免覆盖原始文件
|
||
|
||
###################################################################去重#####################################
|
||
def remove_duplicates(input_file, output_file):
|
||
# 用于存储已经遇到的URL和包含genre的行
|
||
seen_urls = set()
|
||
seen_lines_with_genre = set()
|
||
# 用于存储最终输出的行
|
||
output_lines = []
|
||
# 打开输入文件并读取所有行
|
||
with open(input_file, 'r', encoding='utf-8') as f:
|
||
lines = f.readlines()
|
||
print("去重前的行数：", len(lines))
|
||
# 遍历每一行
|
||
for line in lines:
|
||
# 使用正则表达式查找URL和包含genre的行,默认最后一行
|
||
urls = re.findall(r'[https]?[http]?[rtsp]?[rtmp]?://(?:[a-zA-Z]|[0-9]|[$-_@.&+]|[!*\\(\\),]|(?:%[0-9a-fA-F][0-9a-fA-F]))+', line)
|
||
genre_line = re.search(r'\bgenre\b', line, re.IGNORECASE) is not None
|
||
# 如果找到URL并且该URL尚未被记录
|
||
if urls and urls[0] not in seen_urls:
|
||
seen_urls.add(urls[0])
|
||
output_lines.append(line)
|
||
# 如果找到包含genre的行，无论是否已被记录，都写入新文件
|
||
if genre_line:
|
||
output_lines.append(line)
|
||
# 将结果写入输出文件
|
||
with open(output_file, 'w', encoding='utf-8') as f:
|
||
f.writelines(output_lines)
|
||
print("去重后的行数：", len(output_lines))
|
||
# 使用方法
|
||
remove_duplicates('汇总.txt', '2.txt')
|
||
|
||
|
||
|
||
|
||
|
||
|
||
######################################################################################提取goodiptv
|
||
import re
|
||
import os
|
||
# 定义一个包含所有要排除的关键词的列表
|
||
excluded_keywords = [
|
||
'epg', 'mitv', 'udp', 'rtp', 'P2p', 'p2p', 'p3p', 'P2P', '新闻综合', 'P3p', 'jdshipin', '9930/qilu', 'shandong', 'goodiptv', '购物', '[', 'P3P', '腔', '曲', '//1', '/hls/', '春节'
|
||
# 在这里添加需要排除的关键词
|
||
]
|
||
|
||
# 定义一个包含所有要提取的关键词的列表
|
||
extract_keywords = ['1905', '凤凰卫视', '人间卫视', '亚洲卫视', '香港卫视', '神乐', '翡翠台', '凤凰香港', '凤凰中文', '凤凰资讯', 'AXN', 'AMC', '香蕉', '电影台', '大爱', '东森', 'MTV',
|
||
'华视', '中天', '天良', '翡翠台', '美亚', '星影', '纬来', '天映', '无线', '华剧台', '华丽台', '剧台', 'Movie', '八大', '采昌', '靖天', '美亚', '民視', '探索', '旅游', '影视2',
|
||
'影视3', '中视', '豬哥亮', 'TVB', '东森', '公视', '华视', '寰宇', '戏剧', '靖', '龙华', '龙祥', '民视', '三立', '中视', '猪哥亮', '综艺', '新闻', '影迷', '影剧', '电视剧'
|
||
# 在这里添加需要提取的关键词
|
||
]
|
||
|
||
# 读取文件并处理每一行
|
||
with open('2.txt', 'r', encoding='utf-8') as file:
|
||
lines = file.readlines()
|
||
|
||
# 创建或打开一个输出文件用于写入处理后的数据
|
||
with open('2.txt', 'w', encoding='utf-8') as outfile:
|
||
for line in lines:
|
||
# 首先检查行是否包含任何提取关键词
|
||
if any(keyword in line for keyword in extract_keywords):
|
||
# 如果包含提取关键词，进一步检查行是否不包含任何排除关键词
|
||
if not any(keyword in line for keyword in excluded_keywords):
|
||
outfile.write(line) # 写入符合条件的行到文件
|
||
|
||
|
||
###############################################################
|
||
import re
|
||
def parse_file(input_file_path, output_file_name):
|
||
# 正则表达式匹配从'//'开始到第一个'/'或第一个'::'结束的部分
|
||
ip_or_domain_pattern = re.compile(r'//([^/:]*:[^/:]*::[^/:]*|[^/]*)')
|
||
# 用于存储每个IP或域名及其对应的行列表
|
||
ip_or_domain_to_lines = {}
|
||
# 读取原始文件内容
|
||
with open(input_file_path, 'r', encoding='utf-8') as file:
|
||
for line in file:
|
||
line = line.strip()
|
||
# 如果行是分类标签行，则跳过
|
||
if ",#genre#" in line:
|
||
continue
|
||
# 检查行是否包含IP或域名
|
||
match = ip_or_domain_pattern.search(line)
|
||
if match:
|
||
# 提取匹配到的IP或域名
|
||
matched_text = match.group(1)
|
||
# 去除IP或域名后的剩余部分，只保留匹配到的IP或域名
|
||
ip_or_domain = matched_text.split('://')[-1].split('/')[0].split('::')[0]
|
||
# 将行添加到对应的IP或域名列表中
|
||
if ip_or_domain not in ip_or_domain_to_lines:
|
||
ip_or_domain_to_lines[ip_or_domain] = []
|
||
ip_or_domain_to_lines[ip_or_domain].append(line)
|
||
############################################################################### 过滤掉小于1500字节的IP或域名段
|
||
filtered_ip_or_domain_to_lines = {ip_or_domain: lines for ip_or_domain, lines in ip_or_domain_to_lines.items()
|
||
if sum(len(line) for line in lines) >= 500}
|
||
# 如果没有满足条件的IP或域名段，则不生成文件
|
||
if not filtered_ip_or_domain_to_lines:
|
||
print("没有满足条件的IP或域名段，不生成文件。")
|
||
return
|
||
# 合并所有满足条件的IP或域名的行到一个文件
|
||
with open(output_file_name, 'w', encoding='utf-8') as output_file:
|
||
for ip_or_domain, lines in filtered_ip_or_domain_to_lines.items():
|
||
# 写入IP或域名及其对应的行到输出文件
|
||
output_file.write(f"频道,#genre#\n")
|
||
for line in lines:
|
||
output_file.write(line + '\n')
|
||
output_file.write('\n') # 在每个小段后添加一个空行作为分隔
|
||
# 调用函数并传入文件路径和输出文件名
|
||
parse_file('2.txt', '2.txt')
|
||
|
||
|
||
|
||
import cv2
|
||
import time
|
||
from tqdm import tqdm
|
||
# 初始化2字典
|
||
detected_ips = {}
|
||
# 存储文件路径
|
||
file_path = "2.txt"
|
||
output_file_path = "网络收集.txt"
|
||
def get_ip_key(url):
|
||
"""从URL中提取IP地址，并构造一个唯一的键"""
|
||
# 找到'//'到第一个'/'之间的字符串
|
||
start = url.find('://') + 3 # '://'.length 是 3
|
||
end = url.find('/', start) # 找到第一个'/'的位置
|
||
return url[start:end] if end != -1 else None
|
||
# 打开输入文件和输出文件
|
||
with open(file_path, 'r', encoding='utf-8') as file:
|
||
lines = file.readlines()
|
||
# 获取总行数用于进度条
|
||
total_lines = len(lines)
|
||
# 写入通过检测的行到新文件
|
||
with open(output_file_path, 'a', encoding='utf-8') as output_file:
|
||
# 使用tqdm显示进度条
|
||
for i, line in tqdm(enumerate(lines), total=total_lines, desc="Processing", unit='line'):
|
||
# 检查是否包含 'genre'
|
||
if 'genre' in line:
|
||
output_file.write(line)
|
||
continue
|
||
# 分割频道名称和URL,并去除空白字符
|
||
parts = line.split(',', 1)
|
||
if len(parts) == 2:
|
||
channel_name, url = parts
|
||
channel_name = channel_name.strip()
|
||
url = url.strip()
|
||
# 构造IP键
|
||
ip_key = get_ip_key(url)
|
||
# 检查IP键是否存在
|
||
if ip_key and ip_key in detected_ips:
|
||
# 如果IP键已存在,根据之前的结果决定是否写入新文件
|
||
if detected_ips[ip_key]['status'] == 'ok':
|
||
output_file.write(line)
|
||
elif ip_key: # 新IP键,进行检测
|
||
# 进行检测
|
||
cap = cv2.VideoCapture(url) #支持http(s)/rts(m)p协议
|
||
start_time = time.time()
|
||
frame_count = 0
|
||
# 尝试捕获4秒内的帧
|
||
while frame_count < 50 and (time.time() - start_time) < 3:
|
||
ret, frame = cap.read()
|
||
if not ret:
|
||
break
|
||
frame_count += 1
|
||
# 释放资源
|
||
cap.release()
|
||
|
||
# 根据捕获的帧数判断状态并记录结果
|
||
if frame_count >= 50: # 6秒内超过25帧则写入
|
||
detected_ips[ip_key] = {'status': 'ok'}
|
||
output_file.write(line) # 写入检测通过的行
|
||
else:
|
||
detected_ips[ip_key] = {'status': 'fail'}
|
||
# 打印结果
|
||
for ip_key, result in detected_ips.items():
|
||
print(f"IP Key: {ip_key}, Status: {result['status']}")
|
||
|
||
|
||
|
||
def remove_duplicates(input_file, output_file):
|
||
# 用于存储已经遇到的URL和包含genre的行
|
||
seen_urls = set()
|
||
seen_lines_with_genre = set()
|
||
# 用于存储最终输出的行
|
||
output_lines = []
|
||
# 打开输入文件并读取所有行
|
||
with open(input_file, 'r', encoding='utf-8') as f:
|
||
lines = f.readlines()
|
||
print("去重前的行数：", len(lines))
|
||
# 遍历每一行
|
||
for line in lines:
|
||
# 使用正则表达式查找URL和包含genre的行,默认最后一行
|
||
urls = re.findall(r'[https]?[http]?[P2p]?[mitv]?://(?:[a-zA-Z]|[0-9]|[$-_@.&+]|[!*\\(\\),]|(?:%[0-9a-fA-F][0-9a-fA-F]))+', line)
|
||
genre_line = re.search(r'\bgenre\b', line, re.IGNORECASE) is not None
|
||
# 如果找到URL并且该URL尚未被记录
|
||
if urls and urls[0] not in seen_urls:
|
||
seen_urls.add(urls[0])
|
||
output_lines.append(line)
|
||
# 如果找到包含genre的行,无论是否已被记录,都写入新文件
|
||
if genre_line:
|
||
output_lines.append(line)
|
||
# 将结果写入输出文件
|
||
with open(output_file, 'w', encoding='utf-8') as f:
|
||
f.writelines(output_lines)
|
||
print("去重后的行数：", len(output_lines))
|
||
# 使用方法
|
||
remove_duplicates('网络收集.txt', '网络收集.txt')
|
||
|
||
|
||
|
||
|
||
|
||
############################################################################全部检测，防止IP段失效
|
||
import requests
|
||
import time
|
||
import cv2
|
||
from urllib.parse import urlparse
|
||
from tqdm import tqdm
|
||
|
||
# 测试HTTP连接并尝试下载数据
|
||
def test_connectivity_and_download(url, initial_timeout=3, retry_timeout=5):
|
||
parsed_url = urlparse(url)
|
||
if parsed_url.scheme not in ['http', 'https']:
|
||
# 非HTTP(s)协议，尝试RTSP检测
|
||
return test_rtsp_connectivity(url, retry_timeout)
|
||
else:
|
||
# HTTP(s)协议，使用原始方法
|
||
try:
|
||
with requests.get(url, stream=True, timeout=initial_timeout) as response:
|
||
if 200 <= response.status_code <= 403:
|
||
start_time = time.time()
|
||
while time.time() - start_time < initial_timeout:
|
||
chunk = response.raw.read(256) # 尝试下载1KB数据
|
||
if chunk:
|
||
return True # 成功下载数据
|
||
except requests.RequestException as e:
|
||
print(f"请求异常: {e}")
|
||
pass #这行删掉则会在下载不到数据流的时候进行连通性测试
|
||
return False # 默认返回False
|
||
|
||
print("/" * 80)
|
||
# 测试RTSP连接并尝试读取流
|
||
def test_rtsp_connectivity(url, timeout=5):
|
||
cap = cv2.VideoCapture(url)
|
||
if not cap.isOpened():
|
||
return False
|
||
start_time = time.time()
|
||
while time.time() - start_time < timeout:
|
||
ret, _ = cap.read()
|
||
if ret:
|
||
return True # 成功读取帧
|
||
cap.release()
|
||
return False
|
||
|
||
# 主函数
|
||
def main(输入, 输出):
|
||
with open(输入, "r", encoding="utf-8") as source_file:
|
||
lines = source_file.readlines()
|
||
results = []
|
||
for line_number, line in enumerate(tqdm(lines, desc="检测中")):
|
||
parts = line.strip().split(",")
|
||
if len(parts) == 2 and parts[1]: # 确保有URL，并且URL不为空
|
||
channel_name, channel_url = parts
|
||
try:
|
||
is_valid = test_connectivity_and_download(channel_url)
|
||
except Exception as e:
|
||
print(f"检测URL {channel_url} 时发生错误: {e}")
|
||
is_valid = False # 将异常的URL视为无效
|
||
status = "有效" if is_valid else "无效"
|
||
if "genre" in line.lower() or status == "有效":
|
||
results.append((channel_name.strip(), channel_url.strip(), status))
|
||
# 写入文件
|
||
with open(输出, "w", encoding="utf-8") as output_file:
|
||
for channel_name, channel_url, status in results:
|
||
output_file.write(f"{channel_name},{channel_url}\n")
|
||
|
||
print(f"任务完成, 有效源数量: {len([x for x in results if x[2] == '有效'])}, 无效源数量: {len([x for x in results if x[2] == '无效'])}")
|
||
if __name__ == "__main__":
|
||
输入 = "网络收集.txt" #input('请输入utf-8编码的直播源文件路径:')
|
||
输出 = "网络收集.txt"
|
||
main(输入, 输出)
|
||
|
||
|
||
|
||
|
||
|
||
|
||
import re
|
||
from pypinyin import lazy_pinyin
|
||
# 打开一个utf-8编码的文本文件
|
||
with open("网络收集.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编码的文本文件
|
||
with open("网络收集.txt", "w", encoding="utf-8") as file:
|
||
for line in sorted_lines:
|
||
file.write(line)
|
||
|
||
|
||
|
||
|
||
def parse_file(input_file_path, output_file_name): #
|
||
# 正则表达式匹配从'//'开始到第一个'/'或第一个'::'结束的部分
|
||
ip_or_domain_pattern = re.compile(r'//([^/:]*:[^/:]*::[^/:]*|[^/]*)')
|
||
# 用于存储每个IP或域名及其对应的行列表
|
||
ip_or_domain_to_lines = {}
|
||
# 用于生成分类名的字母和数字计数器
|
||
alphabet_counter = 0 # 字母计数器，从0开始
|
||
number_counter = 1 # 数字计数器，从1开始
|
||
# 读取原始文件内容
|
||
with open(input_file_path, 'r', encoding='utf-8') as file:
|
||
for line in file:
|
||
line = line.strip()
|
||
# 如果行是分类标签行，则跳过
|
||
if ",#genre#" in line:
|
||
continue
|
||
# 检查行是否包含IP或域名
|
||
match = ip_or_domain_pattern.search(line)
|
||
if match:
|
||
# 提取匹配到的IP或域名
|
||
matched_text = match.group(1)
|
||
# 去除IP或域名后的剩余部分，只保留匹配到的IP或域名
|
||
ip_or_domain = matched_text.split('://')[-1].split('/')[0].split('::')[0]
|
||
# 将行添加到对应的IP或域名列表中
|
||
if ip_or_domain not in ip_or_domain_to_lines:
|
||
ip_or_domain_to_lines[ip_or_domain] = []
|
||
ip_or_domain_to_lines[ip_or_domain].append(line)
|
||
# 过滤掉小于1000字节的IP或域名段
|
||
filtered_ip_or_domain_to_lines = {ip_or_domain: lines for ip_or_domain, lines in ip_or_domain_to_lines.items()
|
||
if sum(len(line) for line in lines) >= 300} # 过滤掉小于1000字节的IP或域名段
|
||
# 如果没有满足条件的IP或域名段，则不生成文件
|
||
if not filtered_ip_or_domain_to_lines:
|
||
print("没有满足条件的IP或域名段，不生成文件。")
|
||
return
|
||
# 合并所有满足条件的IP或域名的行到一个文件
|
||
with open(output_file_name, 'w', encoding='utf-8') as output_file: #output_
|
||
for ip_or_domain, lines in filtered_ip_or_domain_to_lines.items():
|
||
# 检查是否需要递增数字计数器
|
||
if alphabet_counter >= 26:
|
||
number_counter += 1
|
||
alphabet_counter = 0 # 重置字母计数器
|
||
# 生成分类名
|
||
genre_name = chr(65 + alphabet_counter) + str(number_counter)
|
||
output_file.write(f"{genre_name},#genre#\n")
|
||
for line in lines:
|
||
output_file.write(line + '\n')
|
||
output_file.write('\n') # 在每个小段后添加一个空行作为分隔
|
||
alphabet_counter += 1 # 递增字母计数器
|
||
# 调用函数并传入文件路径和输出文件名
|
||
parse_file('网络收集.txt', '网络收集.txt')
|
||
|
||
|
||
|
||
|
||
################################################################################################任务结束，删除不必要的过程文件
|
||
files_to_remove = ["2.txt", "汇总.txt"]
|
||
for file in files_to_remove:
|
||
if os.path.exists(file):
|
||
os.remove(file)
|
||
else: # 如果文件不存在，则提示异常并打印提示信息
|
||
print(f"文件 {file} 不存在，跳过删除。")
|
||
print("任务运行完毕，频道列表可查看文件夹内源.txt文件！")
|
||
|
||
|