stellar-getdanmu/qq_danmu.py at main · simplecelery/stellar-getdanmu · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
import json
import requests
from requests.packages.urllib3.exceptions import InsecureRequestWarning
import re
import bs4
import csv
import random

class qq_danmu():
    def __init__(self, url):
        self.url = url
        self.duration = 0
        self.vid = ''
        self.medianame = ''
        requests.packages.urllib3.disable_warnings(InsecureRequestWarning)

    def get_danmu_by_url(self):
        danmudata = []
        self.get_vid()
        if self.vid == '':
            return danmudata
        tid = self.get_tid()
        print(tid)
        if tid == '':
            return danmudata
        jsonlist = []
        self.get_danmu(tid,jsonlist)
        random.shuffle(jsonlist)
        jsonout = {'danmu_type':'qq','danmu':jsonlist}
        self.medianame = re.sub('[’!"#$%&\'()*+,-./:;<=>?@，。?★、…【】《》？“”‘’！[\\]^_`{|}~\s]+', "_", self.medianame)
        danmudata.append({'title':self.medianame,'data':jsonout})
        return danmudata

    def get_vid(self):
        print(self.url)
        res = requests.get(self.url,verify=False)
        statuecode = res.status_code
        print(statuecode)
        if statuecode == 200:
            bs = bs4.BeautifulSoup(res.content.decode('UTF-8','ignore'),'html.parser')
            jsonstr = ''
            for item in bs.find_all('script'):
                try:
                    data = re.findall(r"var VIDEO_INFO = (.+)",str(item))
                    if len(data) > 0:
                        jsonstr = data[0]
                        jsondata = json.loads(jsonstr)
                        self.vid = jsondata['vid']
                        self.duration = int(jsondata['duration'])
                        self.medianame = jsondata['title']
                        break
                    data = re.findall(r"window.__pinia=(.+)",str(item))
                    if len(data) > 0:
                        print("a---------")
                        tmp = data[0]
                        sidx1 = tmp.find('"videoInfo":')
                        sidx = tmp.find("{",sidx1)
                        einx = tmp.find("}",sidx)
                        infostr = tmp[sidx : einx + 1]
                        jsondata = json.loads(infostr)
                        self.vid = jsondata['vid']
                        self.duration = 1000
                        self.medianame = jsondata['title']
                        break
                except:
                    self.vid = ""

    def get_tid(self):
        #getting targetid
        vid = self.vid
        print("getting targetid")
        url = 'https://access.video.qq.com/danmu_manage/regist?vappid=97767206&vsecret=c0bdcbae120669fff425d0ef853674614aa659c605a613a4&raw=1'
        headers = {
            'User-Agent':"PostmanRuntime/7.19.0",
            'Postman-Token':'ecb9a8ff-480b-4c63-af22-da18d327375d',
            'Host':'access.video.qq.com',
            'Content-Type':'application/json',
            'Accept':'*/*',
            'Cache-Control':'no-cache',
            'Accept-Encoding':'gzip, deflate',
            'Connection':'keep-alive'
            }
        payload ={"wRegistType":2,"vecIdList":[vid],"wSpeSource":0,"bIsGetUserCfg":1,"mapExtData":{vid:{"strCid":"wu1e7mrffzvibjy","strLid":""}}}
        r=requests.post(url,headers=headers,data=json.dumps(payload))
        id_compile = re.compile("targetid=([\d]*)&vid")
        targetid = id_compile.findall(r.text)
        return targetid[0]

    def get_danmu(self,targetid,jsonlist):
        timestamp = 0
        while timestamp < self.duration:
            url = 'http://mfm.video.qq.com/danmu?target_id={}&timestamp={}'.format(targetid,timestamp)
            timestamp = timestamp + 30
            res = requests.get(url,verify=False)
            if res.status_code == 200:
                jsondata = json.loads(res.text, strict=False)
                if jsondata:
                    for item in jsondata['comments']:
                        tp = int(item['timepoint']) * 1000
                        newitem = {'tp':tp,'msg':item['content']}
                        jsonlist.append(newitem)


if __name__ == '__main__':
    #r = input('请输入qq视频网页地址：\n')
    dm = qq_danmu("https://v.qq.com/x/cover/mzc00200ccaqs78/y00429ym62l.html")
    print(dm.get_danmu_by_url())