from marshmallow_sqlalchemy import SQLAlchemyAutoSchema
from config import db
from models import User
from marshmallow import fields

#*______________ Base Schema ______________
class BaseScheme(SQLAlchemyAutoSchema):
    def __str__(self):
        return str(self.__class__) + ": " + str(self.__dict__)
    class Meta:
        ordered = True
        sqla_session = db.session
        include_fk = True
        load_instance = True

#*______________ User Schemes ______________
class UserSchema(BaseScheme):
    class Meta(BaseScheme.Meta):
        model = User
    id = fields.Int()
    name = fields.Str()
class UserInsertSchema(UserSchema):
    user_id = fields.Int()
class UserResponseSchema(UserSchema):
    name = fields.Str()