TelegramBotAdmin/wbb/utils/functions.py

from PIL import Image, ImageDraw, ImageFont, ImageFilter
from random import randint


def generate_captcha():
    # Generate one letter
    def gen_letter():
        return chr(randint(65, 90))
    
    def rndColor():
        return (randint(64, 255), randint(64, 255), randint(64, 255))
    def rndColor2():
        return (randint(32, 127), randint(32, 127), randint(32, 127))
    
    # Generate a 4 letter word
    def gen_wrong_answer():
        word = ""
        for _ in range(4):
            word += gen_letter()
        return word 

    # Generate 7 wrong captcha answers
    wrong_answers = []
    for _ in range(7):
        wrong_answers.append(gen_wrong_answer())
    
    width = 120 * 4
    height = 120
    correct_answer = ""
    font = ImageFont.truetype("assets/arial.ttf", 90)
    file = "assets/captcha.jpg"
    image = Image.new('RGB', (width, height), (255, 255, 255))
    draw = ImageDraw.Draw(image)
    
    # Draw random points on image
    for x in range(width):
        for y in range(height):
            draw.point((x, y), fill=rndColor())

    for t in range(4):
        letter = gen_letter()
        correct_answer += letter
        draw.text(
                (120 * t + 32, 3),
                letter,
                font=font,
                fill=rndColor2()
                )
    image = image.filter(ImageFilter.BLUR)
    image.save(file, 'jpeg')
    return [file, correct_answer, wrong_answers]