schemas.py

from marshmallow import Schema, fields, validate


class PlainBookSchema(Schema):
    link_id = fields.Str(dump_only=True)
    title = fields.Str(required=True)
    added_at = fields.Str(dump_only=True)     
    release_year = fields.Int(validate=lambda x: x > 0)
    completed = fields.Int(validate=validate.OneOf([0,1]), load_default=False)
    bookmarked = fields.Int(validate=validate.OneOf([0,1]), load_default=False)
    active = fields.Int(validate=validate.OneOf([0,1]), load_default=False)
    file_url = fields.URL(dump_only=True)
    mime_type = fields.Str(dump_only=True)

class PlainTagSchema(Schema):
    link_id = fields.Str(dump_only=True)
    tag = fields.Str(required=True)

class TagSchema(PlainTagSchema):
    books = fields.List(fields.Nested(PlainBookSchema()), dump_only=True)

class TagUpdateSchema(Schema):
    tag = fields.Str()

class BookSchema(PlainBookSchema):
    tags = fields.List(fields.Nested(PlainTagSchema()), dump_only=True)

class BookUpdateSchema(Schema):
    title = fields.Str()
    release_year = fields.Int(validate=lambda x: x > 0)
    completed = fields.Int(validate=validate.OneOf([0,1]))
    bookmarked = fields.Int(validate=validate.OneOf([0,1]))
    active = fields.Int(validate=validate.OneOf([0,1]))

class PlainUserSchema(Schema):
    link_id = fields.Str(dump_only=True)
    user_name = fields.Str(required=True)
    email = fields.Email(required=True, validate=validate.Email())
    is_admin = fields.Int(validate=validate.OneOf([0,1]), load_default=False)

class UserRegisterSchema(PlainUserSchema):
    user_password = fields.Str(required=True, load_only=True) 
    #access_code = fields.Str(required=True, load_only=True)

class UserUpdateSchema(Schema):
    email = fields.Email(required=True, validate=validate.Email())
    user_password = fields.Str(required=True, load_only=True)              

class UserLoginSchema(Schema):
    user_name = fields.Str(required=True)
    user_password = fields.Str(required=True, load_only=True)               
    
class MultipartFileSchema(Schema):
    file = fields.Raw(required=True, metadata={"type": "file"})                    

class BookSearchQueryArgs(Schema):
    title = fields.Str(required=False)
    release_year = fields.Int(required=False)
    sort = fields.Str(required=False)
    order = fields.Str(required=False)

class TagSearchQueryArgs(Schema):
    name = fields.Str(required=False)

class UserSearchQueryArgs(Schema):
    name = fields.Str(required=False)