simple fix

Author: steven7851

douyin_pro_2.py

@@ -1,9 +1,7 @@
 # -*- coding:utf-8 -*-
-from splinter.driver.webdriver.chrome import Options, Chrome
-from splinter.browser import Browser
 from contextlib import closing
 import requests, json, time, re, os, sys, time
-from bs4 import BeautifulSoup
+from datetime import datetime, timezone
 
 class DouYin(object):
 def __init__(self, width = 500, height = 300):
@@ -35,23 +33,25 @@ def get_video_urls(self, user_id):
 }
 while unique_id != user_id:
 search_url = 'https://api.amemv.com/aweme/v1/discover/search/?cursor=0&keyword=%s&count=10&type=1&retry_type=no_retry&iid=17900846586&device_id=34692364855&ac=wifi&channel=xiaomi&aid=1128&app_name=aweme&version_code=162&version_name=1.6.2&device_platform=android&ssmix=a&device_type=MI+5&device_brand=Xiaomi&os_api=24&os_version=7.0&uuid=861945034132187&openudid=dc451556fc0eeadb&manifest_version_code=162&resolution=1080*1920&dpi=480&update_version_code=1622' % user_id
-req = requests.get(url = search_url, verify = False)
+req = requests.get(search_url, headers=headers)
 html = json.loads(req.text)
 aweme_count = html['user_list'][0]['user_info']['aweme_count']
 uid = html['user_list'][0]['user_info']['uid']
 nickname = html['user_list'][0]['user_info']['nickname']
 unique_id = html['user_list'][0]['user_info']['unique_id']
-user_url = 'https://www.douyin.com/aweme/v1/aweme/post/?user_id=%s&max_cursor=0&count=%s' % (uid, aweme_count)
-req = requests.get(url = user_url, headers=headers, verify = False)
+user_url = 'https://www.amemv.com/aweme/v1/aweme/post/?user_id=%s&max_cursor=0&count=%s' % (uid, aweme_count)
+req = requests.get(user_url, headers=headers)
 html = json.loads(req.text)
-i = 1
 for each in html['aweme_list']:
 share_desc = each['share_info']['share_desc']
+unix_timestamp = each['create_time']
+utc_time = datetime.fromtimestamp(unix_timestamp, timezone.utc)
+local_time = utc_time.astimezone()
+tc = local_time.strftime('%Y-%m-%d-%H-%M-%S')
 if '抖音-原创音乐短视频社区' == share_desc:
-video_names.append(str(i) + '.mp4')
-i += 1
+video_names.append(tc + '.mp4')
 else:
-video_names.append(share_desc + '.mp4')
+video_names.append(tc + '-' + share_desc + '.mp4')
 video_urls.append(each['share_info']['share_url'])
 
 return video_names, video_urls, nickname
@@ -64,17 +64,15 @@ def get_download_url(self, video_url, watermark_flag):
 Returns:
 download_url: 带水印的视频下载地址
 """
-req = requests.get(url = video_url, verify = False)
-bf = BeautifulSoup(req.text, 'lxml')
-script = bf.find_all('script')[-1]
-video_url_js = re.findall('var data = \[(.+)\];', str(script))[0]
-video_html = json.loads(video_url_js)
+req = requests.get(video_url)
+_playaddr_re = re.compile(r'playAddr: "(.+)",')
+playaddr = _playaddr_re.search(req.text)
 # 带水印视频
 if watermark_flag == True:
-download_url = video_html['video']['play_addr']['url_list'][0]
+download_url = playaddr.group(1)
 # 无水印视频
 else:
-download_url = video_html['video']['play_addr']['url_list'][0].replace('playwm','play')
+download_url = playaddr.group(1).replace('playwm','play')
 return download_url
 
 def video_downloader(self, video_url, video_name, watermark_flag=False):
@@ -89,7 +87,7 @@ def video_downloader(self, video_url, video_name, watermark_flag=False):
 """
 size = 0
 video_url = self.get_download_url(video_url, watermark_flag=watermark_flag)
-with closing(requests.get(video_url, stream=True, verify = False)) as response:
+with closing(requests.get(video_url, stream=True)) as response:
 chunk_size = 1024
 content_size = int(response.headers['content-length']) 
 if response.status_code == 200:
@@ -127,7 +125,10 @@ def run(self):
 video_name = video_names[num].replace('/', '')
 else:
 video_name = video_names[num]
-self.video_downloader(video_urls[num], os.path.join(nickname, video_name), watermark_flag)
+if os.path.isfile(os.path.join(nickname, video_name)):
+print('视频已存在')
+else:
+self.video_downloader(video_urls[num], os.path.join(nickname, video_name), watermark_flag)
 print('\n')
 print('下载完成!')