diff --git a/TikTokMulti.py b/TikTokMulti.py index ac9c5c9..e0f3943 100644 --- a/TikTokMulti.py +++ b/TikTokMulti.py @@ -4,7 +4,7 @@ @Description:TikTokMulti.py @Date :2021/05/25 00:14:28 @Author :JohnserfSeed -@version :1.2 +@version :1.2.3 @License :(C)Copyright 2019-2021, Liugroup-NLPR-CASIA @Mail :johnserfseed@gmail.com ''' @@ -26,7 +26,7 @@ def __init__(self): print("#" * 110) print( """ - TikTokDownload V1.2.2 + TikTokDownload V1.2.3 使用说明： 1、运行软件前先打开目录下 conf.ini 文件按照要求进行配置 2、批量下载可直接修改配置文件，单一视频下载请直接打开粘贴视频链接即可 @@ -39,6 +39,7 @@ def __init__(self): print("#" * 110) print('\r') + #检测配置文件 if os.path.isfile("conf.ini") == True: pass else: @@ -58,9 +59,9 @@ def __init__(self): self.cf.set("mode", "mode", "post") with open("conf.ini","a+") as f: self.cf.write(f) - print('----生成成功----') + print('----生成成功----\r') except: - input('----生成失败,正在为您下载配置文件----') + input('----生成失败,正在为您下载配置文件----\r') r =requests.get('https://gitee.com/johnserfseed/TikTokDownload/raw/main/conf.ini') with open("conf.ini", "a+") as conf: conf.write(r.content) @@ -108,10 +109,9 @@ def judge_link(self): #获取解码后原地址 r = requests.get(url = self.Find(self.uid)[0]) multi_url = 'https://www.douyin.com/user/' - #multi_url = 'https://www.iesdouyin.com/share/user/' + #multi_url = 'https://www.iesdouyin.com/share/user/' #旧 #判断输入的是不是用户主页 - #if r.url[:27] == multi_url: if r.url[:28] == multi_url: print('----为您下载多个视频----\r') #获取用户sec_uid @@ -119,7 +119,7 @@ def judge_link(self): key = re.findall('/user/(.*?)\?',str(r.url))[0] if not key: key = r.url[28:83] - print('----'+'用户的sec_id='+key+'----') + print('----'+'用户的sec_id='+key+'----\r') else: print('----为您下载单个视频----\r') print(r.url) @@ -139,17 +139,14 @@ def judge_link(self): def get_data(self,api_post_url,max_cursor): #尝试次数 index = 0 - #存储api数据 result = [] while result == []: index += 1 print('----正在进行第 %d 次尝试----\r' % index) time.sleep(0.3) - response = requests.get(url = api_post_url,headers=self.headers) - #print(api_post_url) + response = requests.get(url = api_post_url,headers = self.headers) html = json.loads(response.content.decode()) - #print(html) if self.end == False: #下一页值 self.nickname = html['aweme_list'][0]['author']['nickname'] @@ -165,13 +162,11 @@ def get_data(self,api_post_url,max_cursor): self.next_data(max_cursor) #self.end = True print('----此页无数据，为您跳过----\r') - return result,max_cursor #下一页 def next_data(self,max_cursor): - #获取解码后原地址 r = requests.get(url = self.Find(self.uid)[0]) @@ -182,10 +177,13 @@ def next_data(self,max_cursor): #构造下一次访问链接 api_naxt_post_url = 'https://www.iesdouyin.com/web/api/v2/aweme/%s/?sec_uid=%s&count=%s&max_cursor=%s&aid=1128&_signature=RuMN1wAAJu7w0.6HdIeO2EbjDc&dytk=' % (self.mode,key,str(self.count),max_cursor) - index = 0 result = [] - while result == []: + while self.end == False: + #回到首页，则结束 + if max_cursor == 0: + self.end = True + return index += 1 print('----正在对',max_cursor,'页进行第 %d 次尝试----\r' % index) time.sleep(0.3) @@ -201,12 +199,10 @@ def next_data(self,max_cursor): else: self.end == True print('----',max_cursor,'页抓获数据失败----\r') - #print('----',max_cursor,'页抓获数据失败----\r') #sys.exit() #处理视频信息 def video_info(self,result,max_cursor): - #作者信息 author_list = [] @@ -220,23 +216,22 @@ def video_info(self,result,max_cursor): nickname = [] #封面大图 - dynamic_cover = [] + #dynamic_cover = [] + for i2 in range(self.count): try: author_list.append(str(result[i2]['desc'])) video_list.append(str(result[i2]['video']['play_addr']['url_list'][0])) aweme_id.append(str(result[i2]['aweme_id'])) nickname.append(str(result[i2]['author']['nickname'])) - dynamic_cover.append(str(result[i2]['video']['dynamic_cover']['url_list'][0])) + #dynamic_cover.append(str(result[i2]['video']['dynamic_cover']['url_list'][0])) except Exception as error: - pass #print(error) - #input('视频信息处理失败...') - #sys.exit() - self.videos_download(author_list,video_list,aweme_id,nickname,dynamic_cover,max_cursor) - return self,author_list,video_list,aweme_id,nickname,dynamic_cover,max_cursor + pass + self.videos_download(author_list,video_list,aweme_id,nickname,max_cursor) + return self,author_list,video_list,aweme_id,nickname,max_cursor - def videos_download(self,author_list,video_list,aweme_id,nickname,dynamic_cover,max_cursor): + def videos_download(self,author_list,video_list,aweme_id,nickname,max_cursor): for i in range(self.count): try: #创建并检测下载目录是否存在 @@ -244,83 +239,63 @@ def videos_download(self,author_list,video_list,aweme_id,nickname,dynamic_cover, except: #有目录不再创建 pass + try: jx_url = f'https://www.iesdouyin.com/web/api/v2/aweme/iteminfo/?item_ids={aweme_id[i]}' #官方接口 js = json.loads(requests.get(url = jx_url,headers=self.headers).text) music_url = str(js['item_list'][0]['music']['play_url']['url_list'][0]) music_title = str(js['item_list'][0]['music']['author']) - if self.musicarg == "yes": - #保留音频 - music=requests.get(music_url) - #保存视频 - start = time.time() #下载开始时间 - size = 0 #初始化已下载大小 - chunk_size = 1024 #每次下载的数据大小 + if self.musicarg == "yes": #保留音频 + music=requests.get(music_url) #保存音频 + start = time.time() #下载开始时间 + size = 0 #初始化已下载大小 + chunk_size = 1024 #每次下载的数据大小 content_size = int(music.headers['content-length']) # 下载文件总大小 try: - if music.status_code == 200: #判断是否响应成功 + if music.status_code == 200: #判断是否响应成功 print('[ 音频 ]:'+author_list[i]+'[文件 大小]:{size:.2f} MB'.format(size = content_size / chunk_size /1024)) #开始下载，显示下载文件大小 m_url = self.save + self.mode + "\\" + nickname[i] + '\\' + re.sub(r'[\\/:*?"<>|\r\n]+', "_", music_title) + '_' + author_list[i] + '.mp3' - with open(m_url,'wb') as file: #显示进度条 + with open(m_url,'wb') as file: #显示进度条 for data in music.iter_content(chunk_size = chunk_size): file.write(data) size +=len(data) print('\r'+'[下载进度]:%s%.2f%%' % ('>'*int(size*50/ content_size), float(size / content_size * 100)) ,end=' ') - end = time.time() #下载结束时间 + end = time.time() #下载结束时间 print('\n' + '[下载完成]:耗时: %.2f秒\n' % (end - start)) #输出下载用时时间 except: - input('下载音频出错!') - #print('音频 ',music_title,'-',author_list[i],' 下载中\r') - #m_url = self.save + self.mode + "\\" + nickname[i] + '\\' + re.sub(r'[\\/:*?"<>|\r\n]+', "_", music_title) + '_' + author_list[i] + '.mp3' - #print(m_url) - #with open(m_url,'wb') as f: - # f.write(music.content) + input('下载音频出错!\r') except Exception as error: #print(error) - #if music_url == '': - print('该页视频没有'+str(self.count)+'个,已为您跳过') + print('该页音频没有'+str(self.count)+'个,已为您跳过\r') break - #print('该音频目前不可用\r') - #else: - # pass + try: - video = requests.get(video_list[i]) - #保存视频 - start = time.time() #下载开始时间 - size = 0 #初始化已下载大小 - chunk_size = 1024 #每次下载的数据大小 + video = requests.get(video_list[i]) #保存视频 + start = time.time() #下载开始时间 + size = 0 #初始化已下载大小 + chunk_size = 1024 #每次下载的数据大小 content_size = int(video.headers['content-length']) # 下载文件总大小 try: - if video.status_code == 200: #判断是否响应成功 + if video.status_code == 200: #判断是否响应成功 print('[ 视频 ]:'+author_list[i]+'[文件 大小]:{size:.2f} MB'.format(size = content_size / chunk_size /1024)) #开始下载，显示下载文件大小 v_url = self.save + self.mode + "\\" + nickname[i] + '\\' + re.sub(r'[\\/:*?"<>|\r\n]+', "_", author_list[i]) + '.mp4' - with open(v_url,'wb') as file: #显示进度条 + with open(v_url,'wb') as file: #显示进度条 for data in video.iter_content(chunk_size = chunk_size): file.write(data) size +=len(data) print('\r'+'[下载进度]:%s%.2f%%' % ('>'*int(size*50/ content_size), float(size / content_size * 100)) ,end=' ') - end = time.time() #下载结束时间 + end = time.time() #下载结束时间 print('\n' + '[下载完成]:耗时: %.2f秒\n' % (end - start)) #输出下载用时时间 except: - input('下载视频出错!') - #print('视频 ',author_list[i],' 下载中\r') - #v_url = self.save + self.mode + "\\" + nickname[i] + '\\' + re.sub(r'[\\/:*?"<>|\r\n]+', "_", author_list[i]) + '.mp4' - #with open(v_url,'wb') as f: - # f.write(video.content) - - #保存视频动态封面 - #dynamic = requests.get(dynamic_cover[i]) - #with open(self.save + self.mode + '\\'+ nickname[i] + '\\' + re.sub(r'[\\/:*?"<>|\r\n]+', "_", author_list[i]) + '.webp','wb') as f: - # f.write(dynamic.content) + input('下载视频出错!\r') except Exception as error: - #pass - print(error) - input('缓存失败，请检查！') - #sys.exit() + #print(error) + print('该页视频没有'+str(self.count)+'个,已为您跳过\r') + break self.next_data(max_cursor) #主模块执行 if __name__ == "__main__": RTK = TikTok() - input('[ 完成 ]:已完成批量下载，输入任意键后退出') + input('[ 完成 ]:已完成批量下载，输入任意键后退出:') sys.exit() \ No newline at end of file