6/18/ds/PY/py_ali.py

405 lines
11 KiB
Python
Raw Normal View History

2024-07-12 09:44:07 +03:00
#coding=utf-8
#!/usr/bin/python
import sys
sys.path.append('..')
from base.spider import Spider
import json
import requests
import time
import re
class Spider(Spider): # 元类 默认的元类 type
def getName(self):
return "阿里云盘"
def init(self,extend=""):
print("============{0}============".format(extend))
pass
def homeContent(self,filter):
result = {}
return result
def homeVideoContent(self):
result = {}
return result
def categoryContent(self,tid,pg,filter,extend):
result = {}
return result
def searchContent(self,key,quick):
result = {}
return result
def isVideoFormat(self,url):
pass
def manualVideoCheck(self):
pass
def playerContent(self,flag,id,vipFlags):
if flag == 'AliYun':
return self.originContent(flag,id,vipFlags)
elif flag == 'AliYun原画':
return self.fhdContent(flag,id,vipFlags)
else:
return {}
def fhdContent(self,flag,id,vipFlags):
self.login()
ids = id.split('+')
shareId = ids[0]
shareToken = ids[1]
fileId = ids[2]
category = ids[3]
url = self.getDownloadUrl(shareId,shareToken,fileId,category)
print(url)
noRsp = requests.get(url,headers=self.header, allow_redirects=False,verify = False)
realUrl = ''
if 'Location' in noRsp.headers:
realUrl = noRsp.headers['Location']
if 'location' in noRsp.headers and len(realUrl) == 0 :
realUrl = noRsp.headers['location']
newHeader = {
"user-agent":"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/94.0.4606.54 Safari/537.36",
"referer":"https://www.aliyundrive.com/",
}
result = {
'parse':'0',
'playUrl':'',
'url':realUrl,
'header':newHeader
}
return result
def originContent(self,flag,id,vipFlags):
self.login()
ids = id.split('+')
shareId = ids[0]
shareToken = ids[1]
fileId = ids[2]
url = '{0}?do=push_agent&api=python&type=m3u8&share_id={1}&file_id={2}'.format(self.localProxyUrl,shareId,fileId)
result = {
'parse':'0',
'playUrl':'',
'url':url,
'header':''
}
# shareToken = self.getToken(shareId,'')
# self.getMediaSlice(shareId,shareToken,fileId)
# map = {
# 'share_id':'p1GJYEqgeb2',
# 'file_id':'62ed1b95b1048d60ffc246669f5e0999e90b8c2f',
# 'media_id':'1'
# }
# self.proxyMedia(map)
return result
def detailContent(self,array):
tid = array[0]
# shareId = self.regStr(href,'www.aliyundrive.com\\/s\\/([^\\/]+)(\\/folder\\/([^\\/]+))?')
# todo =========================================================================================
m = re.search('www.aliyundrive.com\\/s\\/([^\\/]+)(\\/folder\\/([^\\/]+))?', tid)
col = m.groups()
shareId = col[0]
fileId = col[2]
infoUrl = 'https://api.aliyundrive.com/adrive/v3/share_link/get_share_by_anonymous'
infoForm = {'share_id':shareId}
infoRsp = requests.post(infoUrl,json = infoForm,headers=self.header)
infoJo = json.loads(infoRsp.text)
infoJa = []
if 'file_infos' in infoJo:
infoJa = infoJo['file_infos']
if len(infoJa) <= 0 :
return ''
fileInfo = {}
# todo
fileInfo = infoJa[0]
print(fileId)
if fileId == None or len(fileId) <= 0:
fileId = fileInfo['file_id']
vodList = {
'vod_id':tid,
'vod_name':infoJo['share_name'],
'vod_pic':infoJo['avatar'],
'vod_content':tid,
'vod_play_from':'AliYun原画'
}
fileType = fileInfo['type']
if fileType != 'folder':
if fileType != 'file' or fileInfo['category'] != video:
return ''
fileId = 'root'
shareToken = self.getToken(shareId,'')
hashMap = {}
self.listFiles(hashMap,shareId,shareToken,fileId)
sortedMap = sorted(hashMap.items(), key=lambda x: x[0])
arrayList = []
playList = []
for sm in sortedMap:
arrayList.append(sm[0]+'$'+sm[1])
playList.append('#'.join(arrayList))
playList.append('#'.join(arrayList))
vodList['vod_play_url'] = '$$$'.join(playList)
result = {
'list':[vodList]
}
return result
authorization = ''
timeoutTick = 0
localTime = 0
expiresIn = 0
shareTokenMap = {}
expiresMap = {}
localMedia = {}
header = {
"Referer":"https://www.aliyundrive.com/",
"User-Agent":"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/94.0.4606.54 Safari/537.36"
}
localProxyUrl = 'http://127.0.0.1:UndCover/proxy'
def redirectResponse(tUrl):
rsp = requests.get(tUrl, allow_redirects=False,verify = False)
if 'Location' in rsp.headers:
return redirectResponse(rsp.headers['Location'])
else:
return rsp
def getDownloadUrl(self,shareId,token,fileId,category):
lShareId = shareId
lFileId = fileId
params = {
"share_id": lShareId,
"category": "live_transcoding",
"file_id": lFileId,
"template_id": ""
}
customHeader = self.header.copy()
customHeader['x-share-token'] = token
customHeader['authorization'] = self.authorization
url = 'https://api.aliyundrive.com/v2/file/get_share_link_video_preview_play_info'
if category == 'video':
rsp = requests.post(url,json = params,headers=customHeader)
rspJo = json.loads(rsp.text)
lShareId = rspJo['share_id']
lFileId = rspJo['file_id']
jo = {
}
if category == 'video':
jo['share_id'] = lShareId
jo['file_id'] = lFileId
jo['expire_sec'] = 600
if category == 'audio':
jo['share_id'] = lShareId
jo['file_id'] = lFileId
jo['get_audio_play_info'] = True
downloadUrl = 'https://api.aliyundrive.com/v2/file/get_share_link_download_url'
downloadRsp = requests.post(downloadUrl,json = jo,headers=customHeader)
resultJo = json.loads(downloadRsp.text)
return resultJo['download_url']
def getMediaSlice(self,shareId,token,fileId):
params = {
"share_id": shareId,
"category": "live_transcoding",
"file_id": fileId,
"template_id": ""
}
customHeader = self.header.copy()
customHeader['x-share-token'] = token
customHeader['authorization'] = self.authorization
url = 'https://api.aliyundrive.com/v2/file/get_share_link_video_preview_play_info'
rsp = requests.post(url,json = params,headers=customHeader)
rspJo = json.loads(rsp.text)
quality = ['FHD','HD','SD']
videoList = rspJo['video_preview_play_info']['live_transcoding_task_list']
highUrl = ''
for q in quality:
if len(highUrl) > 0:
break
for video in videoList:
if(video['template_id'] == q):
highUrl = video['url']
break
if len(highUrl) == 0:
highUrl = videoList[0]['url']
noRsp = requests.get(highUrl,headers=self.header, allow_redirects=False,verify = False)
m3u8Url = ''
if 'Location' in noRsp.headers:
m3u8Url = noRsp.headers['Location']
if 'location' in noRsp.headers and len(m3u8Url) == 0 :
m3u8Url = noRsp.headers['location']
m3u8Rsp = requests.get(m3u8Url,headers=self.header)
m3u8Content = m3u8Rsp.text
tmpArray = m3u8Url.split('/')[0:-1]
host = '/'.join(tmpArray) + '/'
m3u8List = []
mediaMap = {}
slices = m3u8Content.split("\n")
count = 0
for slice in slices:
tmpSlice = slice
if 'x-oss-expires' in tmpSlice:
count = count + 1
mediaMap[str(count)] = host+tmpSlice
tmpSlice = "{0}?do=push_agent&api=python&type=media&share_id={1}&file_id={2}&media_id={3}".format(self.localProxyUrl,shareId,fileId,count)
m3u8List.append(tmpSlice)
self.localMedia[fileId] = mediaMap
return '\n'.join(m3u8List)
def proxyMedia(self,map):
shareId = map['share_id']
fileId = map['file_id']
mediaId = map['media_id']
shareToken = self.getToken(shareId,'')
refresh = False
url = ''
ts = 0
if fileId in self.localMedia:
fileMap = self.localMedia[fileId]
if mediaId in fileMap:
url = fileMap[mediaId]
if len(url) > 0:
ts = int(self.regStr(url,"x-oss-expires=(\\d+)&"))
# url = self.localMedia[fileId][mediaId]
# ts = int(self.regStr(url,"x-oss-expires=(\\d+)&"))
self.localTime = int(time.time())
if ts - self.localTime <= 60:
self.getMediaSlice(shareId,shareToken,fileId)
url = self.localMedia[fileId][mediaId]
action = {
'url':url,
'header':self.header,
'param':'',
'type':'stream',
'after':''
}
print(action)
return [200, "video/MP2T", action, ""]
def proxyM3U8(self,map):
shareId = map['share_id']
fileId = map['file_id']
shareToken = self.getToken(shareId,'')
content = self.getMediaSlice(shareId,shareToken,fileId)
action = {
'url':'',
'header':'',
'param':'',
'type':'string',
'after':''
}
return [200, "application/octet-stream", action, content]
def localProxy(self,param):
typ = param['type']
if typ == "m3u8":
return self.proxyM3U8(param)
if typ == "media":
return self.proxyMedia(param)
return None
def getToken(self,shareId,sharePwd):
self.localTime = int(time.time())
shareToken = ''
if shareId in self.shareTokenMap:
shareToken = self.shareTokenMap[shareId]
# todo
expire = self.expiresMap[shareId]
if len(shareToken) > 0 and expire - self.localTime > 600:
return shareToken
params = {
'share_id':shareId,
'share_pwd':sharePwd
}
url = 'https://api.aliyundrive.com/v2/share_link/get_share_token'
rsp = requests.post(url,json = params,headers=self.header)
jo = json.loads(rsp.text)
newShareToken = jo['share_token']
self.expiresMap[shareId] = self.localTime + int(jo['expires_in'])
self.shareTokenMap[shareId] = newShareToken
print(self.expiresMap)
print(self.shareTokenMap)
return newShareToken
def listFiles(self,map,shareId,shareToken,fileId):
url = 'https://api.aliyundrive.com/adrive/v3/file/list'
newHeader = self.header.copy()
newHeader['x-share-token'] = shareToken
params = {
'image_thumbnail_process':'image/resize,w_160/format,jpeg',
'image_url_process':'image/resize,w_1920/format,jpeg',
'limit':200,
'order_by':'updated_at',
'order_direction':'DESC',
'parent_file_id':fileId,
'share_id':shareId,
'video_thumbnail_process':'video/snapshot,t_1000,f_jpg,ar_auto,w_300'
}
maker = ''
arrayList = []
for i in range(1,51):
if i >= 2 and len(maker) == 0:
break
params['marker'] = maker
rsp = requests.post(url,json = params,headers=newHeader)
jo = json.loads(rsp.text)
ja = jo['items']
for jt in ja:
if jt['type'] == 'folder':
arrayList.append(jt['file_id'])
else:
if 'video' in jt['mime_type'] or 'video' in jt['category']:
repStr = jt['name'].replace("#", "_").replace("$", "_")
map[repStr] = shareId + "+" + shareToken + "+" + jt['file_id'] + "+" + jt['category']
# print(repStr,shareId + "+" + shareToken + "+" + jt['file_id'])
maker = jo['next_marker']
i = i + 1
for item in arrayList:
self.listFiles(map,shareId,shareToken,item)
def login(self):
self.localTime = int(time.time())
url = 'https://api.aliyundrive.com/token/refresh'
if len(self.authorization) == 0 or self.timeoutTick - self.localTime <= 600:
form = {
'refresh_token':'81461f2e9d0b4ffb86612db403824b8a'
}
rsp = requests.post(url,json = form,headers=self.header)
jo = json.loads(rsp.text)
self.authorization = jo['token_type'] + ' ' + jo['access_token']
self.expiresIn = int(jo['expires_in'])
self.timeoutTick = self.localTime + self.expiresIn
# print(self.authorization)
# print(self.timeoutTick)
# print(self.localTime)
# print(self.expiresIn)