mirror of
https://github.com/xhlove/GetDanMu.git
synced 2025-12-16 16:15:56 +08:00
93 lines
3.0 KiB
Python
93 lines
3.0 KiB
Python
#!/usr/bin/env python3.7
|
||
# coding=utf-8
|
||
'''
|
||
# 作者: weimo
|
||
# 创建日期: 2020-01-04 19:14:41
|
||
# 上次编辑时间 : 2020-01-11 17:23:32
|
||
# 一个人的命运啊,当然要靠自我奋斗,但是...
|
||
'''
|
||
|
||
import json
|
||
import requests
|
||
|
||
from zlib import decompress
|
||
from xmltodict import parse
|
||
|
||
from basic.vars import iqiyiplayer
|
||
from pfunc.dump_to_ass import check_file, write_one_video_subtitles
|
||
from pfunc.request_info import get_vinfos, get_vinfos_by_url, get_vinfo_by_tvid
|
||
|
||
|
||
def get_danmu_by_tvid(name, duration, tvid):
|
||
# http://cmts.iqiyi.com/bullet/41/00/10793494100_300_3.z
|
||
if tvid.__class__ == int:
|
||
tvid = str(tvid)
|
||
api_url = "http://cmts.iqiyi.com/bullet/{}/{}/{}_{}_{}.z"
|
||
timestamp = 300
|
||
index = 0
|
||
max_index = duration // timestamp + 1
|
||
comments = []
|
||
while index < max_index:
|
||
url = api_url.format(tvid[-4:-2], tvid[-2:], tvid, timestamp, index + 1)
|
||
# print(url)
|
||
try:
|
||
r = requests.get(url, headers=iqiyiplayer).content
|
||
except Exception as e:
|
||
print("error info -->", e)
|
||
continue
|
||
try:
|
||
raw_xml = decompress(bytearray(r), 15+32).decode('utf-8')
|
||
except Exception as e:
|
||
index += 1
|
||
continue
|
||
try:
|
||
entry = parse(raw_xml)["danmu"]["data"]["entry"]
|
||
except Exception as e:
|
||
index += 1
|
||
continue
|
||
# with open("raw_xml.json", "w", encoding="utf-8") as f:
|
||
# f.write(json.dumps(parse(raw_xml), ensure_ascii=False, indent=4))
|
||
if entry.__class__ != list:
|
||
entry = [entry]
|
||
for comment in entry:
|
||
if comment.get("list") is None:
|
||
continue
|
||
bulletInfo = comment["list"]["bulletInfo"]
|
||
if bulletInfo.__class__ != list:
|
||
bulletInfo = [bulletInfo]
|
||
for info in bulletInfo:
|
||
color = [info["color"]]
|
||
comments.append([info["content"], color, int(comment["int"])])
|
||
print("已下载{:.2f}%".format(index * timestamp * 100 / duration))
|
||
index += 1
|
||
comments = sorted(comments, key=lambda _: _[-1])
|
||
return comments
|
||
|
||
|
||
def main(args):
|
||
vinfos = []
|
||
if args.tvid:
|
||
vi = get_vinfo_by_tvid(args.tvid)
|
||
if vi:
|
||
vinfos.append(vi)
|
||
if args.aid:
|
||
vi = get_vinfos(args.aid)
|
||
if vi:
|
||
vinfos += vi
|
||
if args.tvid == "" and args.aid == "" and args.url == "":
|
||
args.url = input("请输入iqiyi链接：\n")
|
||
if args.url:
|
||
vi = get_vinfos_by_url(args.url)
|
||
if vi:
|
||
vinfos += vi
|
||
subtitles = {}
|
||
for name, duration, tvid in vinfos:
|
||
print(name, "开始下载...")
|
||
flag, file_path = check_file(name, skip=args.y)
|
||
if flag is False:
|
||
print("跳过{}".format(name))
|
||
continue
|
||
comments = get_danmu_by_tvid(name, duration, tvid)
|
||
comments = write_one_video_subtitles(file_path, comments, args)
|
||
subtitles.update({file_path:comments})
|
||
return subtitles |