python模組之PIL模組（產生隨機驗證碼圖片）

標籤：http   映像   open   admin   需要   obj   應用   main   頁面   

PIL簡介什麼是PIL

PIL：是Python Image Library的縮寫，影像處理的模組。主要的類包括Image,ImageFont,ImageDraw,ImageFilter

PIL的匯入

首先需要安裝一下pillow包

pip install pillow

然後就可以調用PIL裡的類了

from PIL import Imagefrom PIL import ImageFontfrom PIL import ImageDrawfrom 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   #虛化    EMBOSSresize((128,128))     #resize成128*128像素大小convert("RGBA")    #圖形類型轉換getpixel((4,4))   #擷取某個像素位置的值putpixel((4,4),(255,0,0))    #寫入某個像素位置的值

PIL應用

我們主要用PIL來產生一張驗證碼的隨機圖，下面，我們就一步步來做一個小樣本

1、產生一張固定尺寸固定顏色的圖片

from PIL import Image# 擷取一個Image對象，參數分別是RGB模式。寬150，高30，紅色image = Image.new(‘RGB‘,(150,30),‘red‘)# 儲存到硬碟，名為test.png格式為png的圖片image.save(open(‘test.png‘,‘wb‘),‘png‘)

2、產生一張隨機顏色的圖片

from PIL import Imageimport randomdef getRandomColor():    ‘‘‘擷取一個隨機顏色(r,g,b)格式的‘‘‘    c1 = random.randint(0,255)    c2 = random.randint(0,255)    c3 = random.randint(0,255)    return (c1,c2,c3)# 擷取一個Image對象，參數分別是RGB模式。寬150，高30，隨機顏色image = Image.new(‘RGB‘,(150,30),getRandomColor())# 儲存到硬碟，名為test.png格式為png的圖片image.save(open(‘test.png‘,‘wb‘),‘png‘)

3、產生一張帶有固定字串的隨機顏色的圖片

from PIL import Imagefrom PIL import ImageDrawfrom PIL import ImageFontimport randomdef getRandomColor():    ‘‘‘擷取一個隨機顏色(r,g,b)格式的‘‘‘    c1 = random.randint(0,255)    c2 = random.randint(0,255)    c3 = random.randint(0,255)    return (c1,c2,c3)# 擷取一個Image對象，參數分別是RGB模式。寬150，高30，隨機顏色image = Image.new(‘RGB‘,(150,30),getRandomColor())# 擷取一個畫筆對象，將圖片對象傳過去draw = ImageDraw.Draw(image)# 擷取一個font字型對象參數是ttf的字型檔的目錄，以及字型的大小font=ImageFont.truetype("kumo.ttf",size=32)# 在圖片上寫東西,參數是：定位，字串，顏色，字型draw.text((20,0),‘fuyong‘,getRandomColor(),font=font)# 儲存到硬碟，名為test.png格式為png的圖片image.save(open(‘test.png‘,‘wb‘),‘png‘)

　效果：

 

4、產生一張帶有隨機字串隨機顏色的圖片

from PIL import Imagefrom PIL import ImageDrawfrom PIL import ImageFontimport randomdef getRandomColor():    ‘‘‘擷取一個隨機顏色(r,g,b)格式的‘‘‘    c1 = random.randint(0,255)    c2 = random.randint(0,255)    c3 = random.randint(0,255)    return (c1,c2,c3)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# 擷取一個Image對象，參數分別是RGB模式。寬150，高30，隨機顏色image = Image.new(‘RGB‘,(150,30),getRandomColor())# 擷取一個畫筆對象，將圖片對象傳過去draw = ImageDraw.Draw(image)# 擷取一個font字型對象參數是ttf的字型檔的目錄，以及字型的大小font=ImageFont.truetype("kumo.ttf",size=26)for i in range(5):    # 迴圈5次，擷取5個隨機字串    random_char = getRandomStr()    # 在圖片上一次寫入得到的隨機字串,參數是：定位，字串，顏色，字型    draw.text((10+i*30, 0),random_char , getRandomColor(), font=font)# 儲存到硬碟，名為test.png格式為png的圖片image.save(open(‘test.png‘,‘wb‘),‘png‘)

　　效果：

 

5、產生一張帶有噪點的驗證碼圖片

from PIL import Imagefrom PIL import ImageDrawfrom PIL import ImageFontimport randomdef getRandomColor():    ‘‘‘擷取一個隨機顏色(r,g,b)格式的‘‘‘    c1 = random.randint(0,255)    c2 = random.randint(0,255)    c3 = random.randint(0,255)    return (c1,c2,c3)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# 擷取一個Image對象，參數分別是RGB模式。寬150，高30，隨機顏色image = Image.new(‘RGB‘,(150,30),getRandomColor())# 擷取一個畫筆對象，將圖片對象傳過去draw = ImageDraw.Draw(image)# 擷取一個font字型對象參數是ttf的字型檔的目錄，以及字型的大小font=ImageFont.truetype("kumo.ttf",size=26)for i in range(5):    # 迴圈5次，擷取5個隨機字串    random_char = getRandomStr()    # 在圖片上一次寫入得到的隨機字串,參數是：定位，字串，顏色，字型    draw.text((10+i*30, 0),random_char , getRandomColor(), font=font)# 噪點噪線width=150height=30# 劃線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.line((x1,y1,x2,y2),fill=getRandomColor())# 畫點for i in range(30):    draw.point([random.randint(0, width), random.randint(0, height)], fill=getRandomColor())    x = random.randint(0, width)    y = random.randint(0, height)    draw.arc((x, y, x + 4, y + 4), 0, 90, fill=getRandomColor())
# 儲存到硬碟，名為test.png格式為png的圖片image.save(open(‘test.png‘, ‘wb‘), ‘png‘)

　　效果：

6、對驗證碼圖片產生進行封裝

from PIL import Imagefrom PIL import ImageDrawfrom PIL import ImageFontimport randomclass 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("kumo.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_strif __name__ == ‘__main__‘:    img = ValidCodeImg()    data, valid_str = img.getValidCodeImg()    print(valid_str)    f = open(‘test.png‘, ‘wb‘)    f.write(data)

　　

　　效果：

7、應用到實際開發中

login.html

<input id="valid-inp" name="validcode" class="form-control" type="password" placeholder="請輸入驗證碼" autocomplete="off">             <span id="valid-img"><img id="img" src="/get_valid_img" title="點擊再換一張" alt="驗證碼圖片"></span>

urls.py

from django.conf.urls import urlfrom django.contrib import adminfrom blog import viewsurlpatterns = [    url(r‘^admin/‘, admin.site.urls),    url(r‘^$‘,views.Main.as_view(),name=‘main‘),    url(r‘^login$‘,views.Login.as_view(),name=‘login‘),    # 登入頁面驗證碼圖片請求    url(r‘^get_valid_img‘,views.GetValidImg.as_view(),name=‘get_valid_img‘),]

views.py

class Main(View):    def get(self,request):        return render(request,‘main.html‘)class Login(View):    def get(self,request):        return render(request, ‘login.html‘)    def post(self,request):        username = request.POST.get(‘username‘)        password = request.POST.get(‘password‘)        valid_code = request.POST.get(‘valid_code‘)        # print(valid_code)        # print(request.session.get(‘valid_code‘))        if valid_code.upper() != request.session.get(‘valid_code‘).upper():            return JsonResponse({‘state‘:False,‘msg‘:‘驗證碼錯誤‘})        user = auth.authenticate(request,username=username,password=password)        if user:            # 登入成功，通過auth的login方法將使用者寫到session中            auth.login(request,user)            # 提交表單登入成功後跳轉到使用者自己的部落格首頁            redirect_url = ‘/{}‘.format(user.username)            return JsonResponse({‘state‘:True,‘msg‘:‘登入成功！‘,‘url‘:redirect_url})        else:            return JsonResponse({‘state‘:False,‘msg‘:‘使用者名稱或密碼錯誤！‘})class GetValidImg(View):    def get(self,request):        obj = ValidCodeImg()        img_data,valid_code = obj.getValidCodeImg()        request.session[‘valid_code‘] = valid_code        return HttpResponse(img_data)

　　

python模組之PIL模組（產生隨機驗證碼圖片）