Python3爬虫使用requests爬取lol英雄皮肤

本人博客:https://xiaoxiablogs.top

此次爬取lol英雄皮肤一共有两个版本，分别是多线程版本和非多线程版本。

多线程版本

# !/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time     : 2/25/2020 2:24 PM
# @Author   : XiaoXia
# @Blog     : https://xiaoxiablogs.top
# @File     : lol_hero_photo.py
import datetime

import requests
import simplejson
import os
import threading


# 多线程版本
class HeroImage(threading.Thread):
    # lol英雄获取英雄皮肤列表网站
    url_demo = "https://game.gtimg.cn/images/lol/act/img/js/hero/"
    # 设置ua
    ua = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/80.0.3987.100 Safari/537.36"
    headers = {
        'User-Agent': ua
    }

    def __init__(self, hero_id, hero_name):
        threading.Thread.__init__(self)
        self.hero_id = hero_id
        self.hero_name = hero_name.replace("/", "")

    def run(self):
        print("{}的皮肤爬取开始了!!!".format(self.hero_name))
        hero_images_list = self.getImagesUrl()
        self.dirIsExist()
        for hero_images in hero_images_list:
            self.saveImage(hero_images["url"], hero_images['name'].replace("/", ""))
        print("{}皮肤爬取完成!!!".format(self.hero_name))

    def dirIsExist(self):
        """
        判断文件夹是否存在,不存在则创建
        """
        if not os.path.exists("./hero/"):
            os.mkdir("./hero/")
        path = "./hero/{}/".format(self.hero_name)
        if not os.path.exists(path):
            os.mkdir(path)

    def getImagesUrl(self) -> list:
        """
        获取皮肤照片链接
        :return: 皮肤照片数组
        """
        response = self.getJson(self.url_demo + self.hero_id + ".js")
        images = simplejson.loads(response.text)['skins']
        image_list = []
        '''
        skinId: 图片的编号
        name: 皮肤名称
        mainImg: 图片地址
        '''
        for image in images:
            image_dic = {
                "name": image['name'],
                "url": image['mainImg']
            }
            # 由于其中还有一些炫彩模型，所以要去除掉
            if image_dic['url']:
                image_list.append(image_dic)

        return image_list

    def saveImage(self, url: str, image_name: str):
        """
        通过链接获取图片并且将图片保存到相应的目录下
        :param path: 保存目录
        :param image_name: 图片名称
        :param url: 图片地址
        """
        response = requests.get(url, headers=self.headers)
        image_path = "./hero/{}/{}.jpg".format(self.hero_name, image_name)

        with response:
            # 得到图片的二进制文件
            image_file = response.content
            with open(image_path, "wb+") as f:
                f.write(image_file)
                f.flush()

    @staticmethod
    def getJson(hero_url: str) -> requests.models.Response:
        """
            获取json响应
            :param hero_url: 英雄列表的获取链接
            :return:
            """
        response = requests.get(hero_url, headers=HeroImage.headers)
        return response


if __name__ == "__main__":
    # 用于计算程序运行时间的，不需要可直接删除该语句
    start_time = datetime.datetime.now()
    # lol英雄列表
    hero_list = "https://game.gtimg.cn/images/lol/act/img/js/heroList/hero_list.js"
    jsons = HeroImage.getJson(hero_list)
    heros = simplejson.loads(jsons.text)["hero"]
    for hero in heros:
        '''
        编号: heroId
        称号: name
        英文名: alias
        中文名: title
        '''
        name = hero['name'] + '-' + hero['title']

        name = name.replace("/", "")
        thread = HeroImage(hero['heroId'], name)
        thread.start()
        print(threading.active_count())
    # 用于计算程序运行时间的，不需要可直接删除该循环
    while True:
        if threading.active_count() <= 1:
            print("全部爬取完毕")
            end_time = datetime.datetime.now()
            print("总用时为:", end_time-start_time)
            break

非多线程版本

# !/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time     : 2/25/2020 2:24 PM
# @Author   : XiaoXia
# @Blog     : https://xiaoxiablogs.top
# @File     : lol_hero_photo.py
import datetime

import requests
from lxml import etree
from pprint import pprint
import simplejson
import os

# lol英雄网站
url_demo = "https://game.gtimg.cn/images/lol/act/img/js/hero/"
# lol英雄列表
hero_list = "https://game.gtimg.cn/images/lol/act/img/js/heroList/hero_list.js"
# 设置ua
ua = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/80.0.3987.100 Safari/537.36"
headers = {
    'User-Agent': ua
}


def dirIsExist(dir_name: str):
    """
    判断文件夹是否存在,不存在则创建
    :param dir_name: 文件夹名称
    """
    if not os.path.exists("./hero/"):
        os.mkdir("./hero/")
    path = "./hero/{}/".format(dir_name)
    if not os.path.exists(path):
        os.mkdir(path)


def getJson(hero_url: str) -> requests.models.Response:
    """
    获取json响应
    :param hero_url: 英雄列表的获取链接
    :return:
    """
    response = requests.get(hero_url)
    return response


def getImagesUrl(hero_id: str) -> list:
    """
    获取皮肤照片链接
    :param hero_id: 英雄编号
    :return: 皮肤照片数组
    """
    response = getJson(url_demo + hero_id + ".js")
    images = simplejson.loads(response.text)['skins']
    image_list = []
    '''
    skinId: 图片的编号
    name: 皮肤名称
    mainImg: 图片地址
    '''
    for image in images:
        image_dic = {
            "name": image['name'],
            "url": image['mainImg']
        }
        # 由于其中还有一些炫彩模型，所以要去除掉
        if image_dic['url']:
            image_list.append(image_dic)

    return image_list


def saveImage(url: str, image_name: str, path: str):
    """
    通过链接获取图片并且将图片保存到相应的目录下
    :param path: 保存目录
    :param image_name: 图片名称
    :param url: 图片地址
    """
    response = requests.get(url, headers=headers)
    image_path = path + image_name + ".jpg"

    with response:
        # 得到图片的二进制文件
        image_file = response.content
        with open(image_path, "wb+") as f:
            f.write(image_file)
            f.flush()


if __name__ == "__main__":
    # 该语句是用于计算程序运行时间的，不需要可以删除
    start_time = datetime.datetime.now()
    jsons = getJson(hero_list)
    heros = simplejson.loads(jsons.text)["hero"]
    for hero in heros:
        '''
        编号: heroId
        称号: name
        英文名: alias
        中文名: title
        '''
        name = hero['name'] + '-' + hero['title']

        name = name.replace("/", "")
        # 获取每个英雄的皮肤名称及链接列表
        image_lists = getImagesUrl(hero['heroId'])
        # 创建该英雄的文件夹
        dirIsExist(name)
        for img in image_lists:
            # 联盟中有K/DA的皮肤，所以需要将/去掉
            print(img["name"].replace("/", ""))
            saveImage(img['url'], img["name"].replace("/", ""), './hero/{}/'.format(name))
    print("全部爬取完毕")
    # 下面部分是用于计算程序运行时间的，不需要可以删除
    end_time = datetime.datetime.now()
    print("总用时为:", end_time - start_time)