SCUEvals/scuevals-api

from datetime import datetime
from sqlalchemy import func

from . import db
from .permission import Permission
from .assoc import user_permission


class User(db.Model):
    Student = 's'
    Professor = 'p'
    Normal = 'u'

    __tablename__ = 'users'

    id = db.Column(db.Integer, primary_key=True)
    email = db.Column(db.Text, unique=True, nullable=False)
    first_name = db.Column(db.Text, nullable=False)
    last_name = db.Column(db.Text)
    picture = db.Column(db.Text)
    type = db.Column(db.String(1), nullable=False, server_default='u')
    signup_time = db.Column(db.DateTime(timezone=True), server_default=func.now(), nullable=False)
    suspended_until = db.Column(db.DateTime(timezone=True))

    university_id = db.Column(db.Integer, db.ForeignKey('universities.id'), nullable=False)

    university = db.relationship('University', back_populates='users')
    permissions = db.relationship('Permission', secondary=user_permission, back_populates='users', passive_deletes=True)
    flags = db.relationship('Flag', back_populates='user', passive_deletes=True)

    __mapper_args__ = {
        'polymorphic_identity': 'u',
        'polymorphic_on': type
    }

    __table_args__ = (db.CheckConstraint(type.in_(['s', 'p', 'u']), name='user_type'),)

    def _get_permissions(self):
        return [permission.id for permission in self.permissions]

    def _set_permissions(self, value):
        while self.permissions:
            del self.permissions[0]

        for permission_id in value:
            permission = Permission.query.get(permission_id)
            if permission is None:
                raise ValueError('permission does not exist: {}'.format(permission_id))

            self.permissions.append(permission)

    permissions_list = property(_get_permissions,
                                _set_permissions,
                                None,
                                'Property permissions_list is a simple wrapper for permissions relation')

    def to_dict(self):
        user = {
            'id': self.id,
            'university_id': self.university_id,
            'email': self.email,
            'first_name': self.first_name,
            'last_name': self.last_name,
            'picture': self.picture,
            'type': self.type,
            'permissions': self.permissions_list
        }

        return {k: v for k, v in user.items() if v is not None}

    def suspended(self):
        if self.suspended_until is None:
            return False

        if self.suspended_until > datetime.now(self.suspended_until.tzinfo):
            return True

        self.suspended_until = None
        return False