爬虫进阶:验证码突破--4、python PIL 验证码生成

什么是PIL
PIL:是Python Image Library的缩写,图像处理的模块。主要的类包括Image,ImageFont,ImageDraw,ImageFilter

PIL的导入
首先需要安装一下pillow包

pip install pillow

然后就可以调用PIL里的类了

from PIL import Image
from PIL import ImageFont
from PIL import ImageDraw
from PIL import ImageFilter

PIL常用方法

open()  #打开图片     
new(mode,size,color)   #创建一张空白图片     
save("test.gif","GIF")   #保存(新图片路径和名称,保存格式)     
size()   #获取图片大小     
thumbnail(weight,high)   #缩放图片大小(宽,高)     
show()    #显示图片     
blend(img1,img2,alpha)   #两张图片相加,alpha表示img1和img2的比例参数。     
crop()   #剪切,提取某个矩阵大小的图像。它接收一个四元素的元组作为参数,各元素为(left, upper, right, lower),坐标系统的原点(0, 0)是左上角。     
rotate(45)    #逆时针旋转45度     
transpose()    #旋转图像
    transpose(Image.FLIP_LEFT_RIGHT)       #左右对换。
    transpose(Image.FLIP_TOP_BOTTOM)       #上下对换。
    transpose(Image.ROTATE_90)             #旋转 90 度角。
    transpose(Image.ROTATE_180)            #旋转 180 度角。
    transpose(Image.ROTATE_270)            #旋转 270 度角。
 
paste(im,box)#粘贴box大小的im到原先的图片对象中。     
convert()    #用来将图像转换为不同色彩模式。     
filters()     #滤镜
    BLUR   #虚化
    EMBOSS
resize((128,128))     #resize成128*128像素大小     
convert("RGBA")    #图形类型转换     
getpixel((4,4))   #获取某个像素位置的值     
putpixel((4,4),(255,0,0))    #写入某个像素位置的值   

现在我们生成一些验证码来试试:

 	#coding:utf-8
    from PIL import Image
    from PIL import ImageDraw
    from PIL import ImageFont
    import random
    class ValidCodeImg:
        def __init__(self, width=150, height=30, code_count=5, font_size=32, point_count=20, line_count=3,
                     img_format='png'):
            '''
            可以生成一个经过降噪后的随机验证码的图片
            :param width: 图片宽度 单位px
            :param height: 图片高度 单位px
            :param code_count: 验证码个数
            :param font_size: 字体大小
            :param point_count: 噪点个数
            :param line_count: 划线个数
            :param img_format: 图片格式
            :return 生成的图片的bytes类型的data
            '''
            self.width = width
            self.height = height
            self.code_count = code_count
            self.font_size = font_size
            self.point_count = point_count
            self.line_count = line_count
            self.img_format = img_format
    
        @staticmethod
        def getRandomColor():
            '''获取一个随机颜色(r,g,b)格式的'''
            c1 = random.randint(0, 255)
            c2 = random.randint(0, 255)
            c3 = random.randint(0, 255)
            return (c1, c2, c3)
    
        @staticmethod
        def getRandomStr():
            '''获取一个随机字符串,每个字符的颜色也是随机的'''
            random_num = str(random.randint(0, 9))
            random_low_alpha = chr(random.randint(97, 122))
            random_upper_alpha = chr(random.randint(65, 90))
            random_char = random.choice([random_num, random_low_alpha, random_upper_alpha])
            return random_char
    
        def getValidCodeImg(self):
            # 获取一个Image对象,参数分别是RGB模式。宽150,高30,随机颜色
            image = Image.new('RGB', (self.width, self.height), self.getRandomColor())
    
            # 获取一个画笔对象,将图片对象传过去
            draw = ImageDraw.Draw(image)
    
            # 获取一个font字体对象参数是ttf的字体文件的目录,以及字体的大小
            font = ImageFont.truetype("londrina-solid.ttf", size=self.font_size)
    
            temp = []
            for i in range(self.code_count):
                # 循环5次,获取5个随机字符串
                random_char = self.getRandomStr()
    
                # 在图片上一次写入得到的随机字符串,参数是:定位,字符串,颜色,字体
                draw.text((10 + i * 30, -2), random_char, self.getRandomColor(), font=font)
    
                # 保存随机字符,以供验证用户输入的验证码是否正确时使用
                temp.append(random_char)
            valid_str = "".join(temp)
    
            # 噪点噪线
            # 划线
            for i in range(self.line_count):
                x1 = random.randint(0, self.width)
                x2 = random.randint(0, self.width)
                y1 = random.randint(0, self.height)
                y2 = random.randint(0, self.height)
                draw.line((x1, y1, x2, y2), fill=self.getRandomColor())
    
            # 画点
            for i in range(self.point_count):
                draw.point([random.randint(0, self.width), random.randint(0, self.height)], fill=self.getRandomColor())
                x = random.randint(0, self.width)
                y = random.randint(0, self.height)
                draw.arc((x, y, x + 4, y + 4), 0, 90, fill=self.getRandomColor())
    
            # 在内存生成图片
            from io import BytesIO
            f = BytesIO()
            image.save(f, self.img_format)
            data = f.getvalue()
            f.close()
    
            return data, valid_str
    
    
    if __name__ == '__main__':
        filePath = r"D:/Test/"
        n=0
        while n<1000:
            img = ValidCodeImg()
            data, valid_str = img.getValidCodeImg()
            f = open(filePath+valid_str+'.png', 'wb')
            print(filePath+valid_str+'.png'+" Generated")
            f.write(data)
            f.close()
            n+=1

转载从:https://www.cnblogs.com/6324TV/p/8811249.html

字体文件说明https://www.jianshu.com/p/d3793dd481d9
font = ImageFont.truetype(“londrina-solid.ttf”, size=self.font_size)

你可能感兴趣的:(爬虫)