app/schemas/auth.py

from marshmallow import Schema, fields, validate, validates, ValidationError, validates_schema
from flask import current_app


class SendCodeSchema(Schema):
    email = fields.Email(required=True, error_messages={
        "required": "Email is required",
        "invalid": "Invalid email format"
    })

    @validates("email")
    def validate_email_domain(self, value):
        allowed_domains = current_app.config.get('ALLOWED_EMAIL_DOMAINS', [])
        if allowed_domains:
            domain = value.split('@')[-1]
            if domain not in allowed_domains:
                raise ValidationError("Email domain not allowed")


class RegisterSchema(Schema):
    email = fields.Email(required=True, error_messages={
        "required": "邮箱不能为空",
        "invalid": "邮箱格式不正确"
    })
    password = fields.String(
        required=True,
        validate=validate.Length(min=6),
        error_messages={
            "required": "密码不能为空",
            "too_short": "密码长度至少6位"
        }
    )
    code = fields.String(required=True, error_messages={"required": "验证码不能为空"})
class LoginSchema(Schema):
    email = fields.Email(required=True, error_messages={
        "required": "邮箱不能为空",
        "invalid": "邮箱格式不正确"
    })
    password = fields.String(required=True, error_messages={
        "required": "密码不能为空"
    })

class FindSendSchema(Schema):
    email = fields.Email(required=True, error_messages={
        "required": "邮箱不能为空",
        "invalid": "邮箱格式不正确"
    })

class FindResetSchema(Schema):
    email = fields.Email(required=True)
    code = fields.String(required=True)
    password = fields.String(required=True, validate=lambda x: len(x) >= 6)
    password_confirmation = fields.String(required=True)

    @validates_schema
    def validate_passwords(self, data, **kwargs):
        if data['password'] != data['password_confirmation']:
            raise ValidationError("两次密码不一致", "password_confirmation")