124 lines
4.8 KiB
Python
124 lines
4.8 KiB
Python
from bs4 import BeautifulSoup
|
||
import requests
|
||
import sys
|
||
import time
|
||
|
||
def parse_list():
|
||
headers = {
|
||
'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.7',
|
||
'Accept-Language': 'en,zh-CN;q=0.9,zh;q=0.8',
|
||
'Connection': 'keep-alive',
|
||
'Referer': 'http://tonkiang.us/',
|
||
'Upgrade-Insecure-Requests': '1',
|
||
'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/124.0.0.0 Safari/537.36',
|
||
}
|
||
r = requests.get('http://tonkiang.us/', headers=headers, verify=False)
|
||
soup = BeautifulSoup(r.text, 'html.parser')
|
||
parse_result = {}
|
||
for box in soup.select('div.box'):
|
||
box_name = box.find('div').string.strip()
|
||
items = [x.string for x in box.select('span > a')]
|
||
parse_result[box_name] = items
|
||
print(f'打印parse_result：{parse_result}')
|
||
return parse_result
|
||
|
||
def search_ip(ip: str):
|
||
|
||
headers = {
|
||
'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.7',
|
||
'Accept-Language': 'en,zh-CN;q=0.9,zh;q=0.8',
|
||
'Connection': 'keep-alive',
|
||
'Referer': 'http://tonkiang.us/',
|
||
'Upgrade-Insecure-Requests': '1',
|
||
'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/124.0.0.0 Safari/537.36',
|
||
}
|
||
|
||
params = {
|
||
's': ip,
|
||
}
|
||
r = requests.get('http://tonkiang.us/hoteliptv.php', params=params, headers=headers, verify=False)
|
||
soup = BeautifulSoup(r.text, 'html.parser')
|
||
group_name = []
|
||
group_ip = []
|
||
for a in soup.select('div > i'):
|
||
g_name = a.string.strip()
|
||
if '组播' in g_name:
|
||
index = g_name.index('上线') + 2
|
||
if '酒店' in g_name:
|
||
index = g_name.index('上线') + 2
|
||
g_namestring = g_name[index:]
|
||
group_name.append(g_namestring.strip())
|
||
for b in soup.select('div.channel > a > b'):
|
||
b.img.extract()
|
||
group_ip.append(b.string.strip())
|
||
return group_name,group_ip
|
||
|
||
def parse(address: str):
|
||
headers = {
|
||
'Accept': '*/*',
|
||
'Accept-Language': 'en,zh-CN;q=0.9,zh;q=0.8',
|
||
'Connection': 'keep-alive',
|
||
'Referer': 'http://tonkiang.us/hotellist.html?s=223.10.16.11:8085',
|
||
'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/124.0.0.0 Safari/537.36',
|
||
'X-Requested-With': 'XMLHttpRequest',
|
||
}
|
||
|
||
r = requests.get(f'http://tonkiang.us/alllist.php?s={address}&c=false', headers=headers, verify=False)
|
||
parse_result = []
|
||
soup = BeautifulSoup(r.text, 'html.parser')
|
||
for result in soup.select('div.result'):
|
||
name_div = result.find('div', style="float: left;")
|
||
m3u8_div = result.find('td', style="padding-left: 6px;")
|
||
if name_div and m3u8_div:
|
||
parse_result.append({
|
||
"name": name_div.string,
|
||
"m3u": m3u8_div.string.strip()
|
||
})
|
||
# print(f'打印parse_result:{parse_result}')
|
||
return parse_result
|
||
|
||
def gen_m3u(group):
|
||
with open("iptv.m3u", "w+", encoding='utf-8') as f:
|
||
f.write('#EXTM3U\n')
|
||
f.write(f'#EXT-X-VERSION {int(time.time())}\n')
|
||
f.write('\n')
|
||
index = 1
|
||
for (group_name, parse_result) in group.items():
|
||
for one in parse_result:
|
||
f.write(f'''#EXTINF:-1,tvg-id="{index}-{one['name']}" tvg-name="{index}-{one['name']}" group-title="{group_name}",{one['name']}\n''')
|
||
f.write(one['m3u'])
|
||
f.write('\n')
|
||
index = index + 1
|
||
|
||
print("m3u生成成功！")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
result = {}
|
||
ps = []
|
||
ps_name = []
|
||
parse_list = parse_list()
|
||
for (group_name, items) in parse_list.items():
|
||
if group_name == 'Multicast IP':
|
||
for ip in items:
|
||
mu_name,mu_ip = search_ip(ip)
|
||
ps.extend(mu_ip)
|
||
ps_name.extend(mu_name)
|
||
for (group_name, items) in parse_list.items():
|
||
if group_name == 'Hotel IPTV':
|
||
for ip in items:
|
||
hot_name, hot_ip = search_ip(ip)
|
||
ps.extend(hot_ip)
|
||
ps_name.extend(hot_name)
|
||
# print(f'ps：{ps}')
|
||
print(f'ps_name：{ps_name},{len(ps_name)}')
|
||
for x in [x for x in ps if x is not None]:
|
||
try:
|
||
rs = parse(x)
|
||
for y in [y for y in ps_name if y is not None]:
|
||
result[y] = rs
|
||
# print(f'打印result:{result}')
|
||
except Exception as e:
|
||
print(e)
|
||
gen_m3u(result)
|