browser-go-api/models/User.py

from ..app import db, ma, bcrypt
import enum

class Ranks(enum.Enum): # with minimal Elo rating
    D7 = "Seven Dan" # Elo 2700+
    D6 = "Six Dan"
    D5 = "Five Dan" # Elo 2500
    D4 = "Four Dan"
    D3 = "Three Dan"
    D2 = "Two Dan"
    D1 = "One Dan"
    K1 = "One Kyu" # Elo 2000
    K2 = "Two Kyu"
    K3 = "Three Kyu"
    K4 = "Four Kyu"
    K5 = "Five Kyu"
    K6 = "Six Kyu" # Elo 1500
    K7 = "Seven Kyu"
    K8 = "Eight Kyu"
    K9 = "Nine Kyu"
    K10 = "Ten Kyu"
    K11 = "Eleven Kyu" # ELo 1000
    K12 = "Twelve Kyu"
    K13 = "Thirteen Kyu"
    K14 = "Fourteen Kyu"
    K15 = "Fifteen Kyu"
    K16 = "Sixteen Kyu" # Elo 500
    K17 = "Seventeen Kyu"
    K18 = "Eighteen Kyu"
    K19 = "Nineteen Kyu"
    K20 = "Twenty Kyu"
    K21 = "Twenty-One Kyu" # Elo 0
    K22 = "Twenty-Two Kyu"
    K23 = "Twenty-Three Kyu"
    K24 = "Twenty-Four Kyu"
    K25 = "Twenty-Five Kyu"
    K26 = "Twenty-Six Kyu" # Elo -500
    K27 = "Twenty-Seven Kyu"
    K28 = "Twenty-Eight Kyu"
    K29 = "Twenty-Nine Kyu"
    K30 = "Thirty Kyu" # Elo -900

class User(db.Model):
    __tablename__ = "users"

    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    email = db.Column(db.String(255), unique=True, nullable=False)
    password = db.Column(db.DateTime, nullable=False)
    registered_on = db.Column(db.DateTime, nullable=False)
    admin = db.Column(db.Boolean, nullable=False, default=False)
    rank = db.Column(db.Enum(Ranks))
    elo = db.Column(db.Integer)
    rank_certainty = db.Column(db.Boolean, nullable=False, default=False)

    def __init__(self):
        self.password = bcrypt.generate_password_hash(
            password, app.config.get('BCRYPT_LOG_ROUNDS')
        ).decode()