# coding=utf-8
"""
@author: jiajiknag
程序功能：批量下載抖音視頻
"""
import requests
import bs4
import os
import json
import re
import sys
import time
# 如果一個對象沒有實現上下文，我們就不能把它用于with語句。這個時候，可以用closing()來把該對象變為上下文對象。
# closing-將任意對象變為上下文對象，并支持with語句。
from contextlib import closing
# Urllib3是一個功能強大，條理清晰，用于HTTP客戶端的Python庫
# urllib3.disable_warnings()禁用urllib3警告的方法
requests.packages.urllib3.disable_warnings()
# 創建類Sipder()
class Spider():
 def __init__(self):
  # UA對照表：https://blog.csdn.net/time888/article/details/72822729
  self.headers = {
      # 用戶代理：用于瀏覽器識別的，可以看出自己系統版本，瀏覽器，瀏覽器內核等
      'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/65.0.3325.146 Safari/537.36'
      }
  # 輸出信息-視頻信息
  print('[INFO]:Douyin（抖音） App Video downloader...')
  print('[Version]: V1.0')
  print('[Author]: Jiajikang')

 # 創建函數run（）：外部調用運行
 def run(self):
  # 輸入ID地址（爬去某人抖音視頻的抖音號）
  user_id = input('Enter the ID:')
  try:
   # 因為抖音號是數字所以使用int()驗證是否是數字
   int(user_id)
  # 輸入錯誤時輸出except下的語句
  except:
   print('[Error]:ID error...')
   return
  video_names, video_urls, nickname = self._parse_userID(user_id)
  # os.listdir()方法用于返回指定的文件夾包含的文件或文件夾的名字的列表
  if nickname not in os.listdir():
   # os.mkdir() 方法用于以數字權限模式創建目錄
   os.mkdir(nickname)
  print('[INFO]:Number of Videos <%s>' % len(video_urls))
  for num in range(len(video_names)):
   # %d是一個占位符，標識一個字符串型的數據， %s也是一個占位符，標識一個字符串型的數據
   print('[INFO]:Parsing <No.%d> <Url:%s>' % (num+1, video_urls[num]))
   temp = video_names[num].replace('\\', '')
   video_name = temp.replace('/', '')
   # 調用函數_downloader()
   self._downloader(video_urls[num], os.path.join(nickname, video_name))
   print('\n')
  print('[INFO]:All Done...')

 # 創建函數_downloader()并含有參數：路徑和視頻的url-視頻下載
 def _downloader(self, video_url, path):
  # 定義size并初始化為0
  size = 0
  # 定義一個變量download_url：利用函數_get_download_url()來獲取視頻url
  download_url = self._get_download_url(video_url)
  with closing(requests.get(download_url, headers=self.headers, stream=True, verify=False)) as response:
   chunk_size = 1024
   content_size = int(response.headers['content-length'])
   if response.status_code == 200:
    sys.stdout.write('[File Size]: %0.2f MB\n' % (content_size/chunk_size/1024))
    # 使用寫入的方式打開，如果該文件已存在則將其覆蓋。如果該文件不存在，創建新文件。
    with open(path, 'wb') as f:
     # 遍歷獲取數據
     for data in response.iter_content(chunk_size=chunk_size):
      # 向文件中寫入指定的字符串data
      f.write(data)
      # 計算寫入字符串的長度
      size += len(data)
      # flush() 方法是用來刷新緩沖區的，即將緩沖區中的數據立刻寫入文件，同時清空緩沖區，不需要是被動的等待輸出緩沖區寫入。
      f.flush()
      sys.stdout.write('[Progress]: %0.2f%%' % float(size/content_size*100) + '\r')
      sys.stdout.flush()


 # 創建函數_get_download_url()并含有形參video_url：獲得視頻下載地址
 def _get_download_url(self, video_url):
  # 獲取視頻的下載地址
  # Requests 可以為 HTTPS 請求驗證 SSL 證書，就像 web 瀏覽器一樣。要想檢查某個主機的 SSL 證書，你可以使用 verify 參數:
  # 定義變量res用來接收視頻地址，verify 僅應用于主機證書
  res = requests.get(url=video_url, verify=False)
  # 將res.text 文件利用'lxml'解析成xml文件，了解lxml--https://blog.csdn.net/tanzuozhev/article/details/50442243
  soup = bs4.BeautifulSoup(res.text, 'lxml')
  # 使用find_all來獲取網頁中JavaScript中的script的變量；[-1]去除最后一個字符
  script = soup.find_all('script')[-1]
  # 定義變量date=正則表達式\[(.+)]\,[0]可以使其返回一個字典
  video_url_js = re.findall('var data = \[(.+)\];', str(script))[0]
  # 使用loads()下載
  html = json.loads(video_url_js)
  # 返回，使用[0]是返回一個字典
  return html['video']['play_addr']['url_list'][0]


 # 定義函數_parse_userID()且形參user_id；通過user_id獲取該用戶發布的所有視頻
 def _parse_userID(self, user_id):
  # 獲取所有視頻
  video_names = []
  video_urls = []
  unique_id = ''
  # 當獲取的id不是用戶的id時：
  while unique_id != user_id:
   # 獲取url-下載
   search_url = 'https://api.amemv.com/aweme/v1/discover/search/?keyword={}&count=10&type=1&aid=1128'.format(user_id)
   res = requests.get(url=search_url, verify=False)
   res_dic = json.loads(res.text)
   uid = res_dic['user_list'][0]['user_info']['uid']
   aweme_count = res_dic['user_list'][0]['user_info']['aweme_count']
   nickname = res_dic['user_list'][0]['user_info']['nickname']
   unique_id = res_dic['user_list'][0]['user_info']['unique_id']
  # 用戶的url
  user_url = 'https://www.douyin.com/aweme/v1/aweme/post/?user_id={}&max_cursor=0&count={}'.format(uid, aweme_count)
  # 請求獲取用戶的url
  res = requests.get(url=user_url, verify=False)
  # 下載后去的url轉換的文本
  res_dic = json.loads(res.text)
  i = 1
  # 遍歷下載的文本
  for each in res_dic['aweme_list']:
   share_desc = each['share_info']['share_desc']
   if '抖音-原創音樂短視頻社區' == share_desc:
    video_names.append(str(i) + '.mp4')
    i += 1
   else:
    video_names.append(share_desc + '.mp4')
   video_urls.append(each['share_info']['share_url'])
  return video_names, video_urls, nickname

""" 
if __name__ == '__main__':
 # 創建對象
 sp = Spider()
 sp.run()
"""

# coding=utf-8
"""
@author: jiajiknag
程序功能： 測試抖音視頻的下載
"""
from dou_ying_video_download import Spider
if __name__ == '__main__':
 # 創建類Spider()對象
 sp = Spider()
 # 運行開始下載
 sp.run()