from random import randint, sample
import string
from PIL import Image, ImageFont, ImageDraw


# 隨機(jī)生成畫板顏色
bg_color = randint(0, 255), randint(0, 255), randint(0, 255)
# 定義畫板的寬和高
width, height = 200, 80
# 創(chuàng)建畫板對(duì)象
im = Image.new("RGB", (width, height), bg_color)
# 創(chuàng)建畫筆對(duì)象，接收畫板對(duì)象
# 這樣一來(lái)，畫筆所畫的內(nèi)容都會(huì)顯示在畫板上
draw = ImageDraw.Draw(im)
# 繪制噪點(diǎn)，噪點(diǎn)的數(shù)量一般為 width * height * 0.1
for _ in range(int(width * height * 0.1)):
    # 噪點(diǎn)的橫縱坐標(biāo)
    point_pos = randint(0, width), randint(0, height)
    # 噪點(diǎn)的顏色，盡量也是隨機(jī)的
    point_color = randint(0, 255), randint(0, 255), randint(0, 255)
    # 繪制
    draw.point(point_pos, point_color)

# 查看繪制的圖片
im.show()

# 直線的長(zhǎng)度要從畫板的左邊到畫板的右邊
# 因此左端點(diǎn)要在畫板左側(cè)上下變化，右端點(diǎn)要在畫板右側(cè)上下變化
for _ in range(5):
    left_pos = 0, randint(0, height)
    right_pos = width, randint(0, height)
    line_color = randint(0, 255), randint(0, 255), randint(0, 255)
    # 繪制直線
    draw.line([left_pos, right_pos], line_color)

# 繪制曲線，這里繪制的是一個(gè)超出畫板的大圓
# 這樣在畫板上顯示的部分只是大圓的一條弧，看起來(lái)就像是一條曲線
for _ in range(5):
    left_pos = (-100, -100)
    right_pos = (width * 5, randint(0, height))
    arc_color = randint(0, 255), randint(0, 255), randint(0, 255)
    draw.arc([left_pos, right_pos], 0, 360, arc_color)

# 查看一下，繪制的圖形長(zhǎng)什么樣子
im.show()

# 驗(yàn)證碼是由文字和數(shù)字組成，先來(lái)獲取所有的數(shù)字和字母
alpha_digit = string.ascii_letters + string.digits
# 驗(yàn)證碼一般是四個(gè)字符，從里面隨機(jī)選取4個(gè)
verify_code = sample(alpha_digit, 4)
# 生成字體對(duì)象
font = ImageFont.truetype("/System/Library/Fonts/Courier.ttc", 40)
# 為四個(gè)字符創(chuàng)建四種顏色
text_color = [(randint(0, 255), randint(0, 255), randint(0, 255))
              for _ in range(4)]
# 繪制文字
# 注意：坐標(biāo)加上字體的寬度不要超出畫板，否則顯示不全
draw.text((10, 10), verify_code[0], fill=text_color[0], font=font)
draw.text((60, 25), verify_code[1], fill=text_color[1], font=font)
draw.text((110, 15), verify_code[2], fill=text_color[2], font=font)
draw.text((150, 25), verify_code[3], fill=text_color[3], font=font)

# 繪制完成，最后再查看一下
im.show()

# 可以輸入一個(gè)路徑，然后保存成指定的文件
# 不過(guò)更常見的做法是拿到圖片的字節(jié)流，然后直接對(duì)字節(jié)流進(jìn)行渲染
from io import BytesIO
buf = BytesIO()
im.save(buf, "png")
# 此時(shí)圖片內(nèi)容就保存在了 buf 中
print(buf.getvalue()[: 6] == b"\x89PNG\r\n")  # True

import ddddocr

with open("code.png", "rb") as f:
    data = f.read()

# show_ad 默認(rèn)為 True，執(zhí)行時(shí)會(huì)輸出一些廣告，我們不讓它輸出
ocr = ddddocr.DdddOcr(show_ad=False)
code = ocr.classification(data)
print(code)  # 7abf