python批量下载验证码，用来做验证码处理

╰半夏微凉° 2021-12-23 17:21 278阅读 0赞

刚学到爬虫识别验证码，所以自己建一个获取验证码的类，感兴趣的道友，可以看看，代码如下：

import requests
    import time
    import os
    import re
    
    class Pictures:
        """docstring for Pictures"""
        def __init__(self, url, request=None, file_dir=None, headers=None):
            self.url = url
            if not request:
                self.requests = requests.session()
            else:
                self.requests = request
            if not file_dir:
                self.image_dir = './image/'
            else:
                self.image_dir = file_dir
            if not headers:
                self.headers = {
                'Accept':'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8',
                'Accept-Encoding':'gzip,deflate',
                'Accept-Language':'zh-CN,zh;q=0.8',
                'User-Agent':'Mozilla/5.0 (Windows NT 6.2) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/38.0.2125.111 Safari/537.36'
                }
            else:
                self.headers = headers
    
        '''
        保存图片
        '''
        def save_image(self, url=None):
            if url is not None:
                self.url = url
            if not self.url:
                return False
            size = 0
            number = 0
            while size == 0:
                try:
                    img_file = self.requests.get(url=self.url, headers=self.headers)
                except self.requests.exceptions.RequestException as e:
                    raise e
    
                # 不是图片跳过
                if not self.check_image(img_file.headers['Content-Type']):
                    return False
                file_path = self.image_path(img_file.headers)
                # 保存
                with open(file_path, 'wb') as f:
                    f.write(img_file.content)
                # 判断是否正确保存图片
                size = os.path.getsize(file_path)
                if size == 0:
                    os.remove(file_path)
                # 如果该图片获取超过十次则跳过
                number += 1
                if number >= 10:
                    break
            return file_path if (size > 0) else False
    
        '''
        图片保存的路径
        '''
        def image_path(self, header):
            # 文件夹
            if not os.path.exists(self.image_dir):
                os.makedirs(self.image_dir)
            # 文件名
            file_name = str(time.time()).replace('.', '')
            # 文件后缀
            suffix = self.img_type(header)
    
            return self.image_dir + file_name + suffix
    
        '''
        获取图片后缀名
        '''
        def img_type(self,header):
            # 获取文件属性
            image_attr = header['Content-Type']
            pattern = 'image/([a-zA-Z]+)'
            suffix = re.findall(pattern, image_attr, re.IGNORECASE)
            # 获取后缀
            if not suffix:
                suffix = 'png'
            else:
                suffix = suffix[0]
            if re.search('jpeg', suffix, re.IGNORECASE):
                suffix = 'jpg'
    
            return '.' + suffix
    
        # 检查是否为图片类型
        def check_image(self, content_type):
            if 'image' in content_type:
                return True
            else:
                return False
    
    if __name__ == '__main__':
        image = Pictures('http://my.cnki.net/elibregister/CheckCode.aspx')
        for i in range(50):
            image.save_image()

转载于:https://www.cnblogs.com/hardykay/p/11043534.html