import os, time, requests, json, re
from retrying import retry
from urllib import parse
 
class HonorOfKings:
 '''
  This is a main Class, the file contains all documents.
  One document contains paragraphs that have several sentences
  It loads the original file and converts the original file to new content
  Then the new content will be saved by this class
 '''
 def __init__(self, save_path='./heros'):
  self.save_path = save_path
  self.time = str(time.time()).split('.')
  self.url = 'https://apps.game.qq.com/cgi-bin/ams/module/ishow/V1.0/query/workList_inc.cgi?activityId=2735&sVerifyCode=ABCD&sDataType=JSON&iListNum=20&totalpage=0&page={}&iOrder=0&iSortNumClose=1&iAMSActivityId=51991&_everyRead=true&iTypeId=2&iFlowId=267733&iActId=2735&iModuleId=2735&_=%s' % self.time[0]
 
 def hello(self):
  '''
  This is a welcome speech
  :return: self
  '''
  print("*" * 50)
  print(' ' * 18 + '王者榮耀壁紙下載')
  print(' ' * 5 + '作者: Felix  Date: 2020-05-20 13:14')
  print("*" * 50)
  return self
 
 def run(self):
  '''
  The program entry
  '''
  print('↓' * 20 + ' 格式選擇: ' + '↓' * 20)
  print('1.縮略圖 2.1024x768 3.1280x720 4.1280x1024 5.1440x900 6.1920x1080 7.1920x1200 8.1920x1440')
  size = input('請輸入您想下載的格式序號，默認6：')
  size = size if size and int(size) in [1,2,3,4,5,6,7,8] else 6
 
  print('---下載開始...')
  page = 0
  offset = 0
  total_response = self.request(self.url.format(page)).text
  total_res = json.loads(total_response)
  total_page = --int(total_res['iTotalPages'])
  print('---總共 {} 頁...' . format(total_page))
  while True:
if offset > total_page:
 break
url = self.url.format(offset)
response = self.request(url).text
result = json.loads(response)
now = 0
for item in result["List"]:
 now += 1
 hero_name = parse.unquote(item['sProdName']).split('-')[0]
 hero_name = re.sub(r'[【】:.<>|·@#$%^&() ]', '', hero_name)
 print('---正在下載第 {} 頁 {} 英雄 進度{}/{}...' . format(offset, hero_name, now, len(result["List"])))
 hero_url = parse.unquote(item['sProdImgNo_{}'.format(str(size))])
 save_path = self.save_path + '/' + hero_name
 save_name = save_path + '/' + hero_url.split('/')[-2]
 if not os.path.exists(save_path):
  os.makedirs(save_path)
 if not os.path.exists(save_name):
  with open(save_name, 'wb') as f:response_content = self.request(hero_url.replace("/200", "/0")).contentf.write(response_content)
offset += 1
  print('---下載完成...')
 
 @retry(stop_max_attempt_number=3)
 def request(self, url):
  '''
  Send a request
  :param url: the url of request
  :param timeout: the time of request
  :return: the result of request
  '''
  response = requests.get(url, timeout=10)
  assert response.status_code == 200
  return response
 
if __name__ == "__main__":
 HonorOfKings().hello().run()