from __future__ import absolute_import
from dostadmin import db
from dostadmin.mixins import TimestampMixin


class SchoolLeader(TimestampMixin, db.Model):
    __tablename__ = "schoolleader"

    id = db.Column(db.Integer, primary_key=True)
    school_id = db.Column(db.Integer, db.ForeignKey("partner.id"))
    user_id = db.Column(db.Integer, db.ForeignKey("users.id"))

    @classmethod
    def add_school_leader(cls, user_id, school_id):
        school_leader = SchoolLeader(user_id=user_id, school_id=school_id)
        db.session.add(school_leader)
        db.session.commit()
        return school_leader