Merge pull request Jack-Cherish#25 from steven7851/patch-2

Jack-Cherish · web-flow · commit d508afde8e2a · 2018-05-28T20:41:47.000+08:00
share_desc change and replace invalid characters
diff --git a/douyin_pro_2.py b/douyin_pro_2.py
@@ -43,11 +43,14 @@ def get_video_urls(self, user_id):
 		html = json.loads(req.text)
 		for each in html['aweme_list']:
 			share_desc = each['share_info']['share_desc']
+			if os.name == 'nt':
+				for c in r'\/:*?"<>|':
+					share_desc = share_desc.replace(c, '')
 			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:
+			if share_desc in ['抖音-原创音乐短视频社区', 'TikTok']:
 				video_names.append(tc + '.mp4')
 			else:
 				video_names.append(tc + '-' + share_desc + '.mp4')
