from datetime import datetime
from rustgods import db, login_manager
from flask_login import UserMixin


@login_manager.user_loader
def load_user(user_id):
    return User.query.get(int(user_id))


class User(db.Model, UserMixin):
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(20), unique=True, nullable=False)
    image_file = db.Column(db.String(20), nullable=False, default='default.jpg')
    password = db.Column(db.String(60), nullable=False)
    rank = db.Column(db.String(60), nullable=False, default='User')
    patient = db.Column(db.Integer, db.ForeignKey('patient.id'), nullable=True)
    patient = db.Column(db.Integer, db.ForeignKey('doctor.id'), nullable=True)
    posts = db.relationship('Post', backref='author', lazy=True)

    def __repr__(self):
        return f"user('{self.username}','{self.image_file}')"


class Patient(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
    firstname = db.Column(db.String(20), unique=False, nullable=False)
    lastname = db.Column(db.String(20), unique=False, nullable=False)
    dateofbirth = db.Column(db.DateTime, nullable=False, default=datetime.utcnow)
    doctor_id = db.Column(db.Integer, db.ForeignKey('doctor.id'), nullable=False)
    wachtlijst_id = db.Column(db.Integer, db.ForeignKey('doctor.id'), nullable=False)

    def __repr__(self):
        return f"patient('{self.first}','{self.user_id}')"


class Doctor(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
    firstname = db.Column(db.String(20), unique=False, nullable=False)
    lastname = db.Column(db.String(20), unique=False, nullable=False)
    dateofbirth = db.Column(db.DateTime, nullable=False, default=datetime.utcnow)
    afdeling_id = db.Column(db.Integer, db.ForeignKey('afdeling.id'), nullable=False)

    def __repr__(self):
        return f"doctor('{self.first}','{self.user_id}')"


class Afdeling(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(20), unique=True, nullable=False)
    doctor_id = db.Column(db.Integer, db.ForeignKey('doctor.id'), nullable=False)

    def __repr__(self):
        return f"afdeling('{self.first}','{self.name}')"


class Wachtlijst(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    afdeling_id = db.Column(db.Integer, db.ForeignKey('afdeling.id'), nullable=False)
    patient = db.Column(db.Integer, db.ForeignKey('patient.id'), nullable=False)
    melding = db.Column(db.String(20), unique=False, nullable=False)
    prio = db.Column(db.Integer, unique=False, nullable=False)
    wachttijd = db.Column(db.Integer, unique=False, nullable=False)

    def __repr__(self):
        return f"afdeling('{self.first}','{self.melding}')"