Python 使用Pillow模块生成验证码

PIL <span style="color: #0000ff;">def<span style="color: #000000;"> get_chars_str():
<span style="color: #800000;">'''<span style="color: #800000;">
:return:验证码字符集合
<span style="color: #800000;">'''<span style="color: #000000;">
_letter_cases = <span style="color: #800000;">"<span style="color: #800000;">abcdefghjkmnpqrstuvwxy<span style="color: #800000;">" <span style="color: #008000;">#<span style="color: #008000;"> 小写字母，去除可能干扰的i，l，o，z
_upper_cases = _letter_cases.upper() <span style="color: #008000;">#<span style="color: #008000;"> 大写字母
_numbers = <span style="color: #800000;">''.join(map(str,range(3,10))) <span style="color: #008000;">#<span style="color: #008000;"> 数字
init_chars = <span style="color: #800000;">''<span style="color: #000000;">.join((_letter_cases,_upper_cases,_numbers))
<span style="color: #0000ff;">return<span style="color: #000000;"> init_chars

<span style="color: #0000ff;">def create_validate_code(size=(120,30<span style="color: #000000;">),chars=<span style="color: #000000;">get_chars_str(),img_type=<span style="color: #800000;">"<span style="color: #800000;">GIF<span style="color: #800000;">"<span style="color: #000000;">,mode=<span style="color: #800000;">"<span style="color: #800000;">RGB<span style="color: #800000;">"<span style="color: #000000;">,bg_color=(255,255,255<span style="color: #000000;">),fg_color=(0,font_size=18<span style="color: #000000;">,font_type=<span style="color: #800000;">"<span style="color: #800000;">utils/Monaco.ttf<span style="color: #800000;">"<span style="color: #000000;">,length=4<span style="color: #000000;">,draw_lines=<span style="color: #000000;">True,n_line=(1,2<span style="color: #000000;">),draw_points=<span style="color: #000000;">True,point_chance=2<span style="color: #000000;">):
<span style="color: #800000;">"""<span style="color: #800000;">
@todo: 生成验证码图片
@param size: 图片的大小，格式（宽，高），默认为(120,30)
@param chars: 允许的字符集合，格式字符串
@param img_type: 图片保存的格式，默认为GIF，可选的为GIF，JPEG，TIFF，PNG
@param mode: 图片模式，默认为RGB
@param bg_color: 背景颜色，默认为白色
@param fg_color: 前景色，验证码字符颜色，默认为蓝色#0000FF
@param font_size: 验证码字体大小
@param font_type: 验证码字体，默认为 ae_AlArabiya.ttf
@param length: 验证码字符个数
@param draw_lines: 是否划干扰线
@param n_lines: 干扰线的条数范围，格式元组，默认为(1,2)，只有draw_lines为True时有效
@param draw_points: 是否画干扰点
@param point_chance: 干扰点出现的概率，大小范围[0,100]
@return: [0]: PIL Image实例
@return: [1]: 验证码图片中的字符串
<span style="color: #800000;">"""<span style="color: #000000;">

width,height </span>= size  <span style="color: #008000;"&gt;#</span><span style="color: #008000;"&gt; 宽高</span>
<span style="color: #008000;"&gt;#</span><span style="color: #008000;"&gt; 创建图形</span>
img =<span style="color: #000000;"&gt; Image.new(mode,size,bg_color)
draw </span>= ImageDraw.Draw(img)  <span style="color: #008000;"&gt;#</span><span style="color: #008000;"&gt; 创建画笔</span>

<span style="color: #0000ff;"&gt;def</span><span style="color: #000000;"&gt; get_chars():
    </span><span style="color: #800000;"&gt;"""</span><span style="color: #800000;"&gt;生成给定长度的字符串，返回列表格式</span><span style="color: #800000;"&gt;"""</span>
    <span style="color: #0000ff;"&gt;return</span><span style="color: #000000;"&gt; random.sample(chars,length)

</span><span style="color: #0000ff;"&gt;def</span><span style="color: #000000;"&gt; create_lines():
    </span><span style="color: #800000;"&gt;"""</span><span style="color: #800000;"&gt;绘制干扰线</span><span style="color: #800000;"&gt;"""</span><span style="color: #000000;"&gt;
    line_num </span>= random.randint(*n_line)  <span style="color: #008000;"&gt;#</span><span style="color: #008000;"&gt; 干扰线条数</span>

    <span style="color: #0000ff;"&gt;for</span> i <span style="color: #0000ff;"&gt;in</span><span style="color: #000000;"&gt; range(line_num):
        </span><span style="color: #008000;"&gt;#</span><span style="color: #008000;"&gt; 起始点</span>
        begin = (random.randint(0,size[0]),random.randint(0,size[1<span style="color: #000000;"&gt;]))
        </span><span style="color: #008000;"&gt;#</span><span style="color: #008000;"&gt; 结束点</span>
        end = (random.randint(0,size[1<span style="color: #000000;"&gt;]))
        draw.line([begin,end],fill</span>=<span style="color: #000000;"&gt;(0,0))

</span><span style="color: #0000ff;"&gt;def</span><span style="color: #000000;"&gt; create_points():
    </span><span style="color: #800000;"&gt;"""</span><span style="color: #800000;"&gt;绘制干扰点</span><span style="color: #800000;"&gt;"""</span><span style="color: #000000;"&gt;
    chance </span>= min(100,max(0,int(point_chance)))  <span style="color: #008000;"&gt;#</span><span style="color: #008000;"&gt; 大小限制在[0,100]</span>

    <span style="color: #0000ff;"&gt;for</span> w <span style="color: #0000ff;"&gt;in</span><span style="color: #000000;"&gt; range(width):
        </span><span style="color: #0000ff;"&gt;for</span> h <span style="color: #0000ff;"&gt;in</span><span style="color: #000000;"&gt; range(height):
            tmp </span>= random.randint(0,100<span style="color: #000000;"&gt;)
            </span><span style="color: #0000ff;"&gt;if</span> tmp > 100 -<span style="color: #000000;"&gt; chance:
                draw.point((w,h),0))

</span><span style="color: #0000ff;"&gt;def</span><span style="color: #000000;"&gt; create_strs():
    </span><span style="color: #800000;"&gt;"""</span><span style="color: #800000;"&gt;绘制验证码字符</span><span style="color: #800000;"&gt;"""</span><span style="color: #000000;"&gt;
    c_chars </span>=<span style="color: #000000;"&gt; get_chars()
    strs </span>= <span style="color: #800000;"&gt;'</span><span style="color: #800000;"&gt; %s </span><span style="color: #800000;"&gt;'</span> % <span style="color: #800000;"&gt;'</span> <span style="color: #800000;"&gt;'</span>.join(c_chars)  <span style="color: #008000;"&gt;#</span><span style="color: #008000;"&gt; 每个字符前后以空格隔开</span>

<span style="color: #000000;">
font =<span style="color: #000000;"> ImageFont.truetype(font_type,font_size)
font_width,font_height =<span style="color: #000000;"> font.getsize(strs)

    draw.text(((width </span>- font_width) / 3,(height - font_height) / 3<span style="color: #000000;"&gt;),strs,font</span>=font,fill=<span style="color: #000000;"&gt;fg_color)

    </span><span style="color: #0000ff;"&gt;return</span> <span style="color: #800000;"&gt;''</span><span style="color: #000000;"&gt;.join(c_chars)

</span><span style="color: #0000ff;"&gt;if</span><span style="color: #000000;"&gt; draw_lines:
    create_lines()
</span><span style="color: #0000ff;"&gt;if</span><span style="color: #000000;"&gt; draw_points:
    create_points()
strs </span>=<span style="color: #000000;"&gt; create_strs()

</span><span style="color: #008000;"&gt;#</span><span style="color: #008000;"&gt; 图形扭曲参数</span>
params = [1 - float(random.randint(1,2)) / 100<span style="color: #000000;"&gt;,</span>1 - float(random.randint(1,10)) / 100<span style="color: #000000;"&gt;,float(random.randint(</span>1,2)) / 500<span style="color: #000000;"&gt;,</span>0.001<span style="color: #000000;"&gt;,2)) / 500<span style="color: #000000;"&gt;
          ]
img </span>= img.transform(size,Image.PERSPECTIVE,params)  <span style="color: #008000;"&gt;#</span><span style="color: #008000;"&gt; 创建扭曲</span>

<span style="color: #000000;">
img = img.filter(ImageFilter.EDGE_ENHANCE_MORE) <span style="color: #008000;">#<span style="color: #008000;"> 滤镜，边界加强（阈值更大）

<span style="color: #0000ff;"&gt;return</span> img,strs</pre>