from PIL import Image, ImageDraw, ImageFont,ImageFilter
import random
import string
import io
from functools import wraps
from quart import session, redirect, url_for

def generate_captcha():
    characters = string.ascii_uppercase + string.digits
    captcha_text = ''.join(random.choices(characters, k=6))

    font = ImageFont.truetype("arial.ttf", 36)
    image = Image.new('RGB', (200, 60), color=(255, 255, 255))
    draw = ImageDraw.Draw(image)

    for i in range(6):
        draw.text((10 + i * 30, 10), captcha_text[i], font=font,
                  fill=(random.randint(0, 255), random.randint(0, 255), random.randint(0, 255)))
    # 模糊化处理
    image = image.filter(ImageFilter.BLUR)

    # 将图片保存到BytesIO流中
    buffer = io.BytesIO()
    image.save(buffer, 'jpeg')
    buffer.seek(0)

    # captcha_path = os.path.join('static', 'captcha', f'{captcha_text}.png')
    # image.save(captcha_path)
    # return captcha_text, f'static/captcha/{captcha_text}.png'
    return captcha_text, buffer


def verify_captcha(user_input, actual_captcha):
    return user_input == actual_captcha


def login_required(f):
    @wraps(f)
    async def decorated_function(*args, **kwargs):
        if 'user' not in session:
            return redirect(url_for('main.index',error='未登录，请重新登录'))
        return await f(*args, **kwargs)
    return decorated_function