# -*- coding: utf-8 -*-
"""
Created on Wed Dec 13 13:49:52 2017
@author:KillerTwo
"""
import requests
import os
hero_list_url = 'http://pvp.qq.com/web201605/js/herolist.json'
hero_skin_root_url = 'http://game.gtimg.cn/images/yxzj/img201606/skin/hero-info/'
skin_base_dir = 'C:\\Users\\lwt27\\Pictures\\image\\heroskin\\'

def get_ename(hero_json):#傳入獲取到的python對(duì)象，如hero_list_json
 '''獲取英雄名稱對(duì)應(yīng)英雄編號(hào)的一個(gè)字典,例如{小喬:106,...}'''
 cname_ename = {}
 for hero in hero_json:
 cname_ename[hero['cname']] = hero['ename']
 return cname_ename
def get_skin_name(hero_json): #傳入從網(wǎng)頁獲取到的json轉(zhuǎn)換為python字典的對(duì)象
 '''獲取英雄名稱對(duì)應(yīng)的皮膚的所有皮膚名稱的字典,例如
 {'小喬':'戀之微風(fēng)|萬圣前夜|天鵝之夢(mèng)|純白花嫁|繽紛獨(dú)角獸',...}'''
 cname_skin_name = {}
 for hero in hero_json:
 cname_skin_name[hero['cname']] = hero['skin_name']
 return cname_skin_name
def get_hero_skin_count(cname_skin_name): #傳入英雄名稱對(duì)應(yīng)皮膚名稱的字典
 '''獲取每個(gè)英雄對(duì)應(yīng)的皮膚的個(gè)數(shù),例如{'小喬':5,...}'''
 cname_skin_count = {} 
 for item in cname_skin_name.items():
 cname_skin_count[item[0]] = len(item[1].split('|'))
 return cname_skin_count
def get_skin_name_url(skin_base_rul,cname_skin_count,cname_ename):
 #傳入皮膚根地址和名稱對(duì)應(yīng)皮膚數(shù)量的字典和名稱對(duì)應(yīng)編號(hào)的字典
 '''返回英雄名稱對(duì)應(yīng)的所有皮膚的url地址列表的字典,例如{小喬:[skin_url1,skin_url2],...}'''
 cname_url_list = {}
 for cname,count in cname_skin_count.items():
 #print(cname)
 #print(count)
 #print(skin_base_rul)
 #print(cname_ename[cname])
 base_url = skin_base_rul+str(cname_ename[cname])+'/'+str(cname_ename[cname])+'-bigskin-'
 #print(base_url)
 skin_url_list = [str(base_url)+str(num)+'.jpg' for num in range(1,count+1)]
 cname_url_list[cname] = skin_url_list
 return cname_url_list
#print()
d = get_skin_name_url(hero_skin_root_url,get_hero_skin_count(get_skin_name(hero_list_json)),get_ename(hero_list_json))
#print(d)
def get_cname_skin_name(cname_skin_name):#傳入名稱對(duì)應(yīng)皮膚名稱字符串的字典
 cname_skin_name_dict = {}  #返回名稱對(duì)應(yīng)【皮膚名稱的列表】的字典
 for cname,skin_name_list in cname_skin_name.items():
 skin_list = [name for name in skin_name_list.split('|')]
 cname_skin_name_dict[cname] = skin_list
 return cname_skin_name_dict
 
#s = get_skin_name(hero_list_json)
#print(s)
#f = get_cname_skin_name(s)
#print(f)
def get_hero_skin(cname_url_list,cname_skin_name):#傳入名稱對(duì)應(yīng)【皮膚名稱列表】的字典和名稱對(duì)應(yīng)皮膚url列表的字典
 # """獲取每個(gè)英雄的圖片"""
 for cname,skin_url in cname_url_list.items():
 
 if mkdir(skin_base_dir+cname):#創(chuàng)建指定目錄
  os.chdir(skin_base_dir+cname) #進(jìn)入到創(chuàng)建的目錄
  
  for i in range(len(skin_url)):
  file_name = cname_skin_name[cname][i]+'.jpg'
  r = requests.get(skin_url[i])
  with open(file_name,'wb') as f:
   f.write(r.content)
#創(chuàng)建目錄
def mkdir(path):
 # 引入模塊
 import os
 # 去除首位空格
 path=path.strip()
 # 去除尾部 \ 符號(hào)
 path=path.rstrip("\\")
 # 判斷路徑是否存在
 # 存在 True
 # 不存在 False
 isExists=os.path.exists(path)
 # 判斷結(jié)果
 if not isExists:
 # 如果不存在則創(chuàng)建目錄
 # 創(chuàng)建目錄操作函數(shù)
 os.makedirs(path)
 print(path+' 創(chuàng)建成功')
 return True
 else:
 # 如果目錄存在則不創(chuàng)建，并提示目錄已存在
 print(path+' 目錄已存在')
 return False
 return 
if __name__ == '__main__':
 
 hero_list_body = requests.get(hero_list_url) #請(qǐng)求英雄列表
 hero_list_json = hero_list_body.json() #將英雄列表的獲取的json數(shù)據(jù)轉(zhuǎn)換為python對(duì)象
 cname_ename = {} #英雄名稱對(duì)應(yīng)英雄編號(hào)的字典
 cname__skin_name = {} #英雄名稱對(duì)應(yīng)皮膚名稱字符串的字典
 cname_skin_count = {} #英雄名稱對(duì)應(yīng)皮膚數(shù)量的字典
 
 cname_skin_name_str_list = get_skin_name(hero_list_json)
 cname_skin_name_list = get_cname_skin_name(cname_skin_name_str_list)
 cname_skin_count = get_hero_skin_count(cname_skin_name_str_list)
 cname_ename = get_ename(hero_list_json)
 cnam_skin_url_list = get_skin_name_url(hero_skin_root_url,cname_skin_count,cname_ename)
 get_hero_skin(cnam_skin_url_list,cname_skin_name_list)