'''
在學(xué)習(xí)過(guò)程中有什么不懂得可以加我的
python學(xué)習(xí)交流扣扣qun，934109170
群里有不錯(cuò)的學(xué)習(xí)教程、開發(fā)工具與電子書籍。
與你分享python企業(yè)當(dāng)下人才需求及怎么從零基礎(chǔ)學(xué)習(xí)好python，和學(xué)習(xí)什么內(nèi)容。
'''
# -*- coding:utf-8 -*- 
from requests import get
from filetype import guess
from os import rename
from os import makedirs
from os.path import exists
from json import loads
from contextlib import closing 
# 文件下載器
def Down_load(file_url, file_full_name, now_photo_count, all_photo_count):
  headers = {"User-Agent": "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/63.0.3239.132 Safari/537.36"} 
  # 開始下載圖片
  with closing(get(file_url, headers=headers, stream=True)) as response:
    chunk_size = 1024 # 單次請(qǐng)求最大值
    content_size = int(response.headers['content-length']) # 文件總大小
    data_count = 0 # 當(dāng)前已傳輸?shù)拇笮?
    with open(file_full_name, "wb") as file:
      for data in response.iter_content(chunk_size=chunk_size):
        file.write(data)
        done_block = int((data_count / content_size) * 50)
        data_count = data_count + len(data)
        now_jd = (data_count / content_size) * 100
        print("\r %s：[%s%s] %d%% %d/%d" % (file_full_name, done_block * '█', ' ' * (50 - 1 - done_block), now_jd, now_photo_count, all_photo_count), end=" ")
  # 下載完圖片后獲取圖片擴(kuò)展名，并為其增加擴(kuò)展名
  file_type = guess(file_full_name)
  rename(file_full_name, file_full_name + '.' + file_type.extension)  
# 爬取不同類型圖片
def crawler_photo(type_id, photo_count): 
  # 最新 1, 最熱 2, 女生 3, 星空 4
  if(type_id == 1):
    url = 'https://service.paper.meiyuan.in/api/v2/columns/flow/5c68ffb9463b7fbfe72b0db0?page=1&per_page=' + str(photo_count)
  elif(type_id == 2):
    url = 'https://service.paper.meiyuan.in/api/v2/columns/flow/5c69251c9b1c011c41bb97be?page=1&per_page=' + str(photo_count)
  elif(type_id == 3):
    url = 'https://service.paper.meiyuan.in/api/v2/columns/flow/5c81087e6aee28c541eefc26?page=1&per_page=' + str(photo_count)
  elif(type_id == 4):
    url = 'https://service.paper.meiyuan.in/api/v2/columns/flow/5c81f64c96fad8fe211f5367?page=1&per_page=' + str(photo_count) 
  # 獲取圖片列表數(shù)據(jù)
  headers = {"User-Agent": "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/63.0.3239.132 Safari/537.36"}
  respond = get(url, headers=headers)
  photo_data = loads(respond.content)
  # 已經(jīng)下載的圖片張數(shù)
  now_photo_count = 1 
  # 所有圖片張數(shù)
  all_photo_count = len(photo_data)
  # 開始下載并保存5K分辨率壁紙
  for photo in photo_data:
    # 創(chuàng)建一個(gè)文件夾存放我們下載的圖片
    if not exists('./' + str(type_id)):
      makedirs('./' + str(type_id))
    # 準(zhǔn)備下載的圖片鏈接
    file_url = photo['urls']['raw'] 
    # 準(zhǔn)備下載的圖片名稱,不包含擴(kuò)展名
    file_name_only = file_url.split('/')
    file_name_only = file_name_only[len(file_name_only) -1] 
    # 準(zhǔn)備保存到本地的完整路徑
    file_full_name = './' + str(type_id) + '/' + file_name_only 
    # 開始下載圖片
    Down_load(file_url, file_full_name, now_photo_count, all_photo_count)
    now_photo_count = now_photo_count + 1  
if __name__ == '__main__': 
  # 最新 1, 最熱 2, 女生 3, 星空 4
  # 爬取類型為3的圖片(女生),一共準(zhǔn)備爬取20000張
  wall_paper_id = 1
  wall_paper_count = 10
  while(True):
    # 換行符
    print('\n\n') 
    # 選擇壁紙類型
    wall_paper_id = input("壁紙類型：最新壁紙 1, 最熱壁紙 2, 女生壁紙 3, 星空壁紙 4\n請(qǐng)輸入編號(hào)以便選擇5K超清壁紙類型：")
    # 判斷輸入是否正確
    while(wall_paper_id != str(1) and wall_paper_id != str(2) and wall_paper_id != str(3) and wall_paper_id != str(4)):
      wall_paper_id = input("壁紙類型：最新壁紙 1, 最熱壁紙 2, 女生壁紙 3, 星空壁紙 4\n請(qǐng)輸入編號(hào)以便選擇5K超清壁紙類型：")
    # 選擇要下載的壁紙數(shù)量
    wall_paper_count = input("請(qǐng)輸入要下載的5K超清壁紙的數(shù)量：")
    # 判斷輸入是否正確
    while(int(wall_paper_count) <= 0):
      wall_paper_count = input("請(qǐng)輸入要下載的5K超清壁紙的數(shù)量：") 
    # 開始爬取5K高清壁紙
    print("正在下載5K超清壁紙，請(qǐng)稍等……")
    crawler_photo(int(wall_paper_id), int(wall_paper_count))
    print('\n下載5K高清壁紙成功!')