captcha_generator.py

from __future__ import annotations

from asyncio import to_thread
from random import sample
from string import digits
from typing import TYPE_CHECKING

from captcha.image import ImageCaptcha


if TYPE_CHECKING:
    from io import BytesIO


async def generate_captcha(
    length: int = 6,
    chars: str = digits,
    width: int = 380,
    height: int = 180,
) -> tuple[bytes, str]:
    captcha_factory = ImageCaptcha(width=width, height=height)
    code = "".join(sample(population=chars, k=length))
    image: BytesIO = await to_thread(captcha_factory.generate, chars=code)
    return image.getvalue(), code