<filename>api/api/models/user.py<gh_stars>1-10
from api.db import db

from api.models.committee import Committee
from api.utility.receptionmode import RECEPTION_MODE


class User(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    kth_id = db.Column(db.String, unique=True)
    email = db.Column(db.String)
    profile_picture = db.Column(
        db.String, default="/static/profiles/default.png")
    first_name = db.Column(db.String)
    last_name = db.Column(db.String)
    frack_name = db.Column(db.String, nullable=True)
    kth_year = db.Column(db.Integer)
    linkedin = db.Column(db.String, nullable=True)
    facebook = db.Column(db.String, nullable=True)
    alumni = db.Column(db.Boolean, default=False)
    post_terms = db.relationship("CommitteePostTerm", back_populates="user", lazy='dynamic')
    is_admin = db.Column(db.Boolean, default=False)

    def to_dict(self):
        terms = []

        for term in self.post_terms:
            terms.append({"post": term.post.to_dict_without_terms(),
                          "startDate": term.start_date,
                          "endDate": term.end_date})

        return {"id": self.id,
                "email": self.email,
                "kthId": self.kth_id,
                "profilePicture": self.profile_picture if not RECEPTION_MODE else "/static/profiles/default.png",
                "firstName": self.frack_name if (RECEPTION_MODE and self.frack_name is not None) else self.first_name,
                "lastName": "" if (RECEPTION_MODE and self.frack_name is not None) else self.last_name,
                "frackName": self.frack_name,
                "kthYear": self.kth_year,
                "linkedin": self.linkedin if not RECEPTION_MODE else None,
                "facebook": self.facebook if not RECEPTION_MODE else None,
                "committeePostTerms": terms,
                "alumni": self.alumni,
                "isAdmin": self.is_admin
                }
    
    def to_dict_without_terms(self):
        return {"id": self.id,
                "email": self.email,
                "kthId": self.kth_id,
                "profilePicture": self.profile_picture if not RECEPTION_MODE else "/static/profiles/default.png",
                "firstName": self.frack_name if (RECEPTION_MODE and self.frack_name is not None) else self.first_name,
                "lastName": "" if (RECEPTION_MODE and self.frack_name is not None) else self.last_name,
                "frackName": self.frack_name,
                "kthYear": self.kth_year,
                "linkedin": self.linkedin if not RECEPTION_MODE else None,
                "facebook": self.facebook if not RECEPTION_MODE else None,
                "alumni": self.alumni,
                "isAdmin": self.is_admin
                }
