cake-erp-challenge/api/models/trainer.py

from api.app import db, ma
from werkzeug.security import generate_password_hash

teams = (
 "Team Valor",
 "Team Instinct",
 "Team Mystic"
 )

class InvalidTeam(Exception):
    pass

# modelo do Trainer para o SQLAlchemy
class Trainer(db.Model):
    __tablename__ = "trainers"

    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    nickname = db.Column(db.String(20), unique=True, nullable=False, index=True)
    first_name = db.Column(db.String(30), nullable=False)
    last_name = db.Column(db.String(30), nullable=False)
    email = db.Column(db.String(60), unique=True, index=True, nullable=False)
    password = db.Column(db.String(200), nullable=False)
    team = db.Column(db.String(10), nullable=False)
    pokemons_owned = db.Column(db.Integer, default=0)
    pokemons_list = db.relationship("PokemonOwned", lazy="dynamic")

    def __init__(self, nickname, first_name, last_name, email, password, team):
        if team not in teams:
            raise InvalidTeam()
        self.nickname = nickname
        self.first_name = first_name
        self.last_name = last_name
        self.email = email
        self.password = generate_password_hash(password)
        self.team = team


# schema do Marshmallow
class TrainerSchema(ma.Schema):
    class Meta:
        fields = ('id', 'nickname', 'first_name', 'last_name', 'email', 'team', 'pokemons_owned')

trainer_schema = TrainerSchema()
trainer_schemas = TrainerSchema(many=True)