Create 网络收集.py
This commit is contained in:
frxz751113
GitHub
+682
@@ -0,0 +1,682 @@
|
|||||||
|
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', '151:99', '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
|
||||||
|
import re
|
||||||
|
from tqdm import tqdm
|
||||||
|
|
||||||
|
# 初始化字典
|
||||||
|
detected_ips = {}
|
||||||
|
|
||||||
|
# 存储文件路径，使用通配符表示文件名以'网络收集'开头并以'.txt'结尾
|
||||||
|
file_path_pattern = '*网络收集.txt'
|
||||||
|
output_file_path = None
|
||||||
|
|
||||||
|
def get_ip_key(url):
|
||||||
|
"""从 URL 中提取 IP 地址，并构造一个唯一的键"""
|
||||||
|
start = url.find('://') + 3
|
||||||
|
end = url.find('/', start)
|
||||||
|
return url[start:end] if end!= -1 else None
|
||||||
|
|
||||||
|
def process_file(file_path):
|
||||||
|
# 打开输入文件
|
||||||
|
with open(file_path, 'r', encoding='utf-8') as file:
|
||||||
|
lines = file.readlines()
|
||||||
|
|
||||||
|
# 获取总行数用于进度条
|
||||||
|
total_lines = len(lines)
|
||||||
|
|
||||||
|
# 临时输出文件路径
|
||||||
|
temp_output_file_path = file_path + '.temp'
|
||||||
|
|
||||||
|
# 写入通过检测的行到临时新文件
|
||||||
|
with open(temp_output_file_path, 'a', encoding='utf-8') as output_file:
|
||||||
|
for i, line in tqdm(enumerate(lines), total=total_lines, desc="Processing", unit='line'):
|
||||||
|
if 'genre' in line:
|
||||||
|
output_file.write(line)
|
||||||
|
continue
|
||||||
|
parts = line.split(',', 1)
|
||||||
|
if len(parts) == 2:
|
||||||
|
channel_name, url = parts
|
||||||
|
channel_name = channel_name.strip()
|
||||||
|
url = url.strip()
|
||||||
|
ip_key = get_ip_key(url)
|
||||||
|
if ip_key and ip_key in detected_ips:
|
||||||
|
if detected_ips[ip_key]['status'] == 'ok':
|
||||||
|
output_file.write(line)
|
||||||
|
elif ip_key:
|
||||||
|
cap = cv2.VideoCapture(url)
|
||||||
|
start_time = time.time()
|
||||||
|
frame_count = 0
|
||||||
|
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:
|
||||||
|
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']}")
|
||||||
|
|
||||||
|
# 执行去重操作
|
||||||
|
seen_urls = set()
|
||||||
|
seen_lines_with_genre = set()
|
||||||
|
output_lines = []
|
||||||
|
with open(temp_output_file_path, 'r', encoding='utf-8') as f:
|
||||||
|
lines = f.readlines()
|
||||||
|
print("去重前的行数：", len(lines))
|
||||||
|
for line in lines:
|
||||||
|
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
|
||||||
|
if urls and urls[0] not in seen_urls:
|
||||||
|
seen_urls.add(urls[0])
|
||||||
|
output_lines.append(line)
|
||||||
|
if genre_line:
|
||||||
|
output_lines.append(line)
|
||||||
|
|
||||||
|
# 将结果写回原文件
|
||||||
|
with open(file_path, 'w', encoding='utf-8') as f:
|
||||||
|
f.writelines(output_lines)
|
||||||
|
print("去重后的行数：", len(output_lines))
|
||||||
|
|
||||||
|
# 获取当前目录下符合模式的文件并处理
|
||||||
|
import os
|
||||||
|
directory = '.'
|
||||||
|
for filename in os.listdir(directory):
|
||||||
|
if filename.startswith('网络收集') and filename.endswith('.txt'):
|
||||||
|
output_file_path = filename
|
||||||
|
process_file(output_file_path)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
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 process_file(file_path_pattern):
|
||||||
|
import os
|
||||||
|
for filename in os.listdir('.'):
|
||||||
|
if filename.startswith('网络收集') and filename.endswith('.txt'):
|
||||||
|
input_file = filename
|
||||||
|
output_file = filename
|
||||||
|
with open(input_file, "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]:
|
||||||
|
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
|
||||||
|
status = "有效" if is_valid else "无效"
|
||||||
|
if "genre" in line.lower() or status == "有效":
|
||||||
|
results.append((channel_name.strip(), channel_url.strip(), status))
|
||||||
|
# 写入文件
|
||||||
|
with open(output_file, "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__":
|
||||||
|
process_file('*网络收集.txt')
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
import re
|
||||||
|
from pypinyin import lazy_pinyin
|
||||||
|
import datetime
|
||||||
|
|
||||||
|
def process_file(file_path_pattern):
|
||||||
|
import os
|
||||||
|
for filename in os.listdir('.'):
|
||||||
|
if filename.startswith('网络收集') and filename.endswith('.txt'):
|
||||||
|
input_file = filename
|
||||||
|
output_file = filename
|
||||||
|
# 打开一个 utf-8 编码的文本文件
|
||||||
|
with open(input_file, "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(output_file, "w", encoding="utf-8") as file:
|
||||||
|
for line in sorted_lines:
|
||||||
|
file.write(line)
|
||||||
|
|
||||||
|
# 正则表达式匹配从'//'开始到第一个'/'或第一个'::'结束的部分
|
||||||
|
ip_or_domain_pattern = re.compile(r'//([^/:]*:[^/:]*::[^/:]*|[^/]*)')
|
||||||
|
# 用于存储每个 IP 或域名及其对应的行列表
|
||||||
|
ip_or_domain_to_lines = {}
|
||||||
|
# 用于生成分类名的字母和数字计数器
|
||||||
|
alphabet_counter = 0 # 字母计数器，从 0 开始
|
||||||
|
number_counter = 1 # 数字计数器，从 1 开始
|
||||||
|
# 读取原始文件内容
|
||||||
|
with open(output_file, '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, 'w', encoding='utf-8') as output_file:
|
||||||
|
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 # 递增字母计数器
|
||||||
|
|
||||||
|
now = datetime.datetime.utcnow() + datetime.timedelta(hours=8)
|
||||||
|
current_time = now.strftime("%m/%d %H:%M")
|
||||||
|
# 打开文本文件并将时间添加到开头
|
||||||
|
with open(output_file, 'r+', encoding='utf-8') as f:
|
||||||
|
content = f.read()
|
||||||
|
f.seek(0, 0)
|
||||||
|
f.write(f'{current_time}更新,#genre#\n')
|
||||||
|
f.write(f'虚情的爱,https://vd2.bdstatic.com/mda-mi1dd05gmhwejdwn/sc/cae_h264/1630576203346678103/mda-mi1dd05gmhwejdwn.mp4\n{content}')
|
||||||
|
|
||||||
|
process_file('*网络收集.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文件！")
|
||||||
|
|
||||||
|
|
||||||
Reference in New Issue
Block a user