def sync(self, cookie=None):
# 检验必要的参数
if self.aid is None:
raise Util.ParameterError('缺少获取视频信息的必要参数')
# 发送网络请求
http_request = {
'info_obj': Util.ALBUM,
'params': {
'aid': str(self.aid)
},
'cookie': cookie
}
json_data = Util.http_get(**http_request)
# 修改对象信息
self.num = json_data['data']['videos']
self.type = json_data['data']['tname']
self.cover = json_data['data']['pic']
self.name = json_data['data']['title']
self.time = json_data['data']['ctime']
self.desc = json_data['data']['desc']
self.view = json_data['data']['stat']['view']
self.danmu = json_data['data']['stat']['danmaku']
self.reply = json_data['data']['stat']['reply']
self.favorite = json_data['data']['stat']['favorite']
self.coin = json_data['data']['stat']['coin']
self.share = json_data['data']['stat']['share']
self.like = json_data['data']['stat']['like']
self.video_info = list()
for page in json_data['data']['pages']:
self.video_info.append((page['cid'], page['part']))
# 返回稿件信息
return copy.deepcopy(vars(self))
def get_album_list(self, cookie=None):
# 检验必要的参数
if self.uid is None or self.cid is None:
raise Util.ParameterError('缺少获取频道列表的必要参数')
# 发送网络请求
http_request = {
'info_obj': Util.CHANNEL,
'params': {
'mid': str(self.uid),
'cid': str(self.cid),
'pn': 1, # 当前页码下标
'ps': 30, # 每页视频数量
'order': 0 # 默认排序
},
'cookie': cookie
}
album_list = []
while True:
json_data = Util.http_get(**http_request)
# 修改对象信息
self.name = json_data['data']['list']['name']
self.cover = json_data['data']['list']['cover']
self.count = str(json_data['data']['list']['count'])
# 循环获取列表
album_list.extend([Video.Album(av['aid']) for av in json_data['data']['list']['archives']])
if len(album_list) < int(json_data['data']['page']['count']):
http_request['params']['pn'] += 1
else:
break
return album_list
def get_album_list(self, cookie=None):
# 检验必要的参数
if self.uid is None:
raise Util.ParameterError('缺少获取视频列表的必要参数')
# 发送网络请求
http_request = {
'info_obj': Util.USER_VIDEO,
'params': {
'mid': str(self.uid),
'pagesize': 30,
'tid': 0,
'page': 1,
'order': 'pubdate'
},
'cookie': cookie
}
album_list = []
while True:
json_data = Util.http_get(**http_request)
# 循环获取列表
album_list.extend(
[Video.Album(av['aid']) for av in json_data['data']['vlist']])
if len(album_list) < int(json_data['data']['count']):
http_request['params']['page'] += 1
else:
break
# 返回视频列表
return album_list
def sync(self, cookie=None):
# 检验必要的参数
if self.uid is None:
raise Util.ParameterError('缺少获取用户信息的必要参数')
# 发送网络请求
http_request = {
'info_obj': Util.USER,
'params': {
'mid': str(self.uid),
'jsonp': 'jsonp'
},
'cookie': cookie
}
json_data = Util.http_get(**http_request)
# 修改对象信息
self.name = json_data['data']['name']
self.sex = json_data['data']['sex']
self.face = json_data['data']['face']
self.sign = json_data['data']['sign']
self.level = json_data['data']['level']
self.birthday = json_data['data']['birthday']
self.title = json_data['data']['official']['title']
self.vip = bool(json_data['data']['vip']['status'])
# 返回用户信息
return copy.deepcopy(vars(self))
def fetch_all(self,
cookie=None,
name_pattern=Util.Config.SET_AS_CODE,
quality=None):
"""
自动获取用户与频道名下的所有视频信息
:param cookie: 权限信息
:param quality: 下载视频质量
:param name_pattern: 路径命名方式
:return: 所有视频的av号列表
"""
av_list = []
self.info_list = []
# 确定对象类型
if isinstance(self.obj, Space.User):
obj_code = self.obj.uid
elif isinstance(self.obj, Space.Channel):
obj_code = self.obj.cid
else:
raise Util.ParameterError('该类型对象无法自动加载')
# 逐级确定储存路径
if name_pattern == Util.Config.SET_AS_NAME:
self.obj.sync()
obj_name = Util.legalize_name(self.obj.name)
else:
obj_name = obj_code
album_list = self.obj.get_album_list()
for album in album_list:
if name_pattern == Util.Config.SET_AS_NAME:
album.sync()
album_name = Util.legalize_name(album.name)
else:
album_name = album.aid
video_list = album.get_video_list()
for video in video_list:
video.sync(cookie, quality)
if name_pattern == Util.Config.SET_AS_NAME:
video_name = Util.legalize_name(video.name)
else:
video_name = video.aid
video_name = '{}_P{}_{}'.format(video_name, video.page,
video.quality[1])
self.info_list.append({
'obj': self.obj,
'obj_name': obj_name,
'album': album,
'album_name': album_name,
'video': video,
'video_name': video_name
})
av_list.append(album.aid)
av_list = list(set(av_list))
return av_list
def fetch_all(self,
cookie=None,
name_pattern=Util.Config.SET_AS_CODE,
quality=None):
"""
自动获取用户与频道名下的所有视频信息
:param cookie: 权限信息
:param quality: 下载视频质量
:param name_pattern: 路径命名方式
:return: 所有视频的av号列表
"""
av_list = []
self.info_list = []
# 确定对象类型
if isinstance(self.obj, Space.User):
obj_code = self.obj.uid
elif isinstance(self.obj, Space.Channel):
obj_code = self.obj.cid
else:
raise Util.ParameterError('该类型对象无法自动加载')
# 逐级确定储存路径
if name_pattern == Util.Config.SET_AS_CODE:
obj_name = obj_code
elif name_pattern == Util.Config.SET_AS_NAME:
self.obj.sync()
obj_name = Util.legalize_name(self.obj.name)
elif name_pattern == Util.Config.SET_AS_PAGE:
self.obj.sync()
obj_name = Util.legalize_name(self.obj.name)
else:
obj_name = "unknown"
album_list = self.obj.get_album_list()
for album in tqdm(album_list, desc="Fetching..."):
try:
album.sync(cookie)
except Util.RunningError:
print("404 or 403 on: %s, skipped" % album.aid)
continue
album_name = album.album_name(name_pattern)
video_list = album.get_video_list()
for video in video_list:
video.sync(cookie, quality)
video_name = video.video_name(name_pattern)
self.info_list.append({
'obj': (obj_name, self.obj),
'album': (album_name, album),
'video': (video_name, video)
})
av_list.append(album.aid)
av_list = list(set(av_list))
return av_list
def sync(self, cookie=None, quality=None):
# 检验必要的参数
if self.album.aid is None or self.cid is None:
raise Util.ParameterError('缺少获取视频信息的必要参数')
if quality is None:
quality = Util.Config.Quality.V1080P60
# 发送网络请求
http_request = {
'info_obj': Util.VIDEO,
'params': {
'avid': str(self.album.aid),
'cid': str(self.cid),
'qn': quality[0],
'otype': 'json',
'fnver': 0,
'fnval': 16
},
'cookie': cookie
}
json_data = Util.http_get(**http_request)
# 自动识别不同的数据
self.format = json_data['data']['format']
self.length = json_data['data']['timelength']
self.quality = Util.Config.Quality.INDEX[json_data['data']['quality']]
if 'dash' in json_data['data']:
self.level = 'new_version'
video_obj = json_data['data']['dash']['video'][0]
audio_obj = json_data['data']['dash']['audio'][0]
self.height = video_obj['height']
self.width = video_obj['width']
self.video = list([video_obj['baseUrl']])
self.audio = list([audio_obj['baseUrl']])
if video_obj['backup_url']:
for backup in video_obj['backup_url']:
self.video.append(backup)
if audio_obj['backup_url']:
for backup in audio_obj['backup_url']:
self.video.append(backup)
elif 'durl' in json_data['data']:
self.level = 'old_version'
video_obj = json_data['data']['durl'][-1]
for item in json_data['data']['durl']:
self.video.append(item['url'])
#self.video = list([video_obj['url']])
if item['backup_url']:
for backup in video_obj['backup_url']:
self.video.append(backup)
# 返回视频信息
return copy.deepcopy(vars(self))
def get_video_list(self, cookie=None):
# 检验必要的参数
if self.aid is None:
raise Util.ParameterError('缺少获取视频信息的必要参数')
if self.cid_list is None:
self.sync(cookie)
video_list = []
for index, cid in enumerate(self.cid_list):
cv = Video.Video(self.aid, cid, self.name, index + 1)
video_list.append(cv)
return video_list
def get_video_list(self, cookie=None):
# 检验必要的参数
if self.aid is None:
raise Util.ParameterError('缺少获取视频信息的必要参数')
if self.video_info is None:
self.sync(cookie)
video_list = []
for index, info in enumerate(self.video_info):
video = Video.Video(self, info[0], info[1], index + 1)
video_list.append(video)
return video_list
def set_ratio(self, ratio):
if ~isinstance(ratio, list):
raise Util.ParameterError('参数类型异常')
self.ratio = ratio
def set_page(self, page):
if ~isinstance(page, list):
raise Util.ParameterError('参数类型异常')
self.page = page
def set_width(self, width):
if ~isinstance(width, list):
raise Util.ParameterError('参数类型异常')
self.width = width
def set_height(self, height):
if ~isinstance(height, list):
raise Util.ParameterError('参数类型异常')
self.height = height
def set_length(self, length):
if ~isinstance(length, list):
raise Util.ParameterError('参数类型异常')
self.length = length
def set_quality(self, quality):
if ~isinstance(quality, list):
raise Util.ParameterError('参数类型异常')
self.quality = quality
