生成图片验证码

# ********************************************************************************
# PIL是python2版本的图像处理库，不过现在用Pillow比PIL强大，是python3的处理库
from PIL import Image, ImageDraw, ImageFont
from django.http import HttpResponse
#导入随机数模块
import random
from io import BytesIO

def verify_code(request):
        # 1，定义变量，用于画面的背景色、宽、高
        # random.randrange(20, 100)意思是在20到100之间随机找一个数
        bgcolor = (random.randrange(20, 100), random.randrange(20, 100), 159)
        width = 100
        height = 30
        # 2，创建画面对象
        im = Image.new('RGB', (width, height), bgcolor)
        # 3，创建画笔对象
        draw = ImageDraw.Draw(im)
        # 4，调用画笔的point()函数绘制噪点，防止攻击
        for i in range(0, 100):
            # 噪点绘制的范围
            xy = (random.randrange(0, width), random.randrange(0, height))
            # 噪点的随机颜色
            fill = (random.randrange(0, 255), 255, random.randrange(0, 255))
            # 绘制出噪点
            draw.point(xy, fill=fill)
        # 5，定义验证码的备选值
        str1 = 'ABCD123EFGHJK456LMNPQRS789TUVWXYZ0'
        # 6，随机选取4个值作为验证码
        rand_str = ''
        for i in range(0, 4):
            rand_str += str1[random.randrange(0, len(str1))]
        # 7，构造字体对象，ubuntu的字体路径为“/usr/share/fonts/truetype/freefont”
        #arial.ttf window下的字体
        font = ImageFont.truetype('arial.ttf', 23)
        # 8，构造字体颜色
        fontcolor = (255, random.randrange(0, 255), random.randrange(0, 255))
        print(rand_str)
        # 9，绘制4个字
        draw.text((5, 2), rand_str[0], font=font, fill=fontcolor)
        draw.text((25, 2), rand_str[1], font=font, fill=fontcolor)
        draw.text((50, 2), rand_str[2], font=font, fill=fontcolor)
        draw.text((75, 2), rand_str[3], font=font, fill=fontcolor)
        # 9，用完画笔，释放画笔
        del draw
        # 10，存入session，用于做进一步验证
        request.session['verifycode'] = rand_str.lower()
        # 11，内存文件操作
        buf = BytesIO()
        # 12，将图片保存在内存中，文件类型为png
        im.save(buf, 'png')
        # print(buf.getvalue())
        # 13，将内存中的图片数据返回给客户端，MIME类型为图片png
        return HttpResponse(buf.getvalue(), 'image/png')

# 获取验证码图片的视图
def get_valid_img(request):
    # 获取随机颜色的函数
    def get_random_color():
        return random.randint(0, 255), random.randint(0, 255), random.randint(0, 255)

    # 生成一个图片对象
    img_obj = Image.new(
        'RGB',
        (220, 35),
        get_random_color()
    )
    # 在生成的图片上写字符
    # 生成一个图片画笔对象
    draw_obj = ImageDraw.Draw(img_obj)
    # 加载字体文件， 得到一个字体对象
    font_obj = ImageFont.truetype('arial.ttf', 28)
    # 开始生成随机字符串并且写到图片上
    tmp_list = []
    for i in range(5):
        u = chr(random.randint(65, 90))  # 生成大写字母
        l = chr(random.randint(97, 122))  # 生成小写字母
        n = str(random.randint(0, 9))  # 生成数字，注意要转换成字符串类型

        tmp = random.choice([u, l, n])
        tmp_list.append(tmp)
        draw_obj.text((20 + 40 * i, 0), tmp, fill=get_random_color(), font=font_obj)

    # 保存到session
    request.session["valid_code"] = "".join(tmp_list)
    # 加干扰线
    width = 220  # 图片宽度（防止越界）
    height = 35
    for i in range(5):
        x1 = random.randint(0, width)
        x2 = random.randint(0, width)
        y1 = random.randint(0, height)
        y2 = random.randint(0, height)
        draw_obj.line((x1, y1, x2, y2), fill=get_random_color())

    # 加干扰点
    for i in range(40):
        draw_obj.point((random.randint(0, width), random.randint(0, height)), fill=get_random_color())
        x = random.randint(0, width)
        y = random.randint(0, height)
        draw_obj.arc((x, y, x+4, y+4), 0, 90, fill=get_random_color())

    # 不需要在硬盘上保存文件，直接在内存中加载就可以
    io_obj = BytesIO()
    # 将生成的图片数据保存在io对象中
    img_obj.save(io_obj, "png")
    # 从io对象里面取上一步保存的数据
    data = io_obj.getvalue()
    print(data)
    return render(request, 'index.html', {"data":data})

# ********************************************************************************