flaschengeist/flaschengeist/system/models/user.py

from ..database import db
from sqlalchemy.orm.collections import attribute_mapped_collection
from flask import current_app
from werkzeug.local import LocalProxy

logger = LocalProxy(lambda: current_app.logger)

association_table = db.Table('user_group',
                             db.Column('user_id', db.Integer, db.ForeignKey('user.id')),
                             db.Column('group_id', db.Integer, db.ForeignKey('group.id'))
                             )


class User(db.Model):
    """ Database Object for User

        Table for all saved User

        Attributes:
            id: Id in Database as Primary Key.
            uid: User ID used by authentication provider
            display_name: Name to show
            firstname: Firstname of the User
            lastname: Lastname of the User
            mail: mail address of the User
    """
    __tablename__ = 'user'
    id = db.Column(db.Integer, primary_key=True)
    uid = db.Column(db.String(30))
    display_name = db.Column(db.String(30))
    firstname = db.Column(db.String(30))
    lastname = db.Column(db.String(30))
    mail = db.Column(db.String(30))
    groups = db.relationship("Group", secondary=association_table)
    sessions = db.relationship("AccessToken", back_populates="user")
    attributes = db.relationship("UserAttribute", collection_class=attribute_mapped_collection('name'),
                                 cascade="all, delete")

    def set_attribute(self, name, value):
        if name in self.attributes:
            self.attributes[name].value = value
        else:
            self.attributes[name] = UserAttribute(name=name, value=value)

    def add_group(self, name):
        r = Group.query.filter_by(name=name).first()
        if not r:
            r = Group(name=name)
        self.groups.append(r)

    def update_data(self, data):
        logger.debug("update data of user")
        if 'uid' in data:
            self.uid = data['uid']
        if 'firstname' in data:
            self.firstname = data['firstname']
        if 'lastname' in data:
            self.lastname = data['lastname']
        if 'mail' in data:
            self.mail = data['mail']
        if 'display_name' in data:
            self.display_name = data['display_name']

    def default(self):
        return {
            # TODO: username should be UID?
            "username": self.uid,
            "display_name": self.display_name,
            "firstname": self.firstname,
            "lastname": self.lastname,
            "mail": self.mail,
            "groups": ["user"] + [g.name for g in self.groups]
        }


class UserAttribute(db.Model):
    __tablename__ = 'userAttribute'
    id = db.Column(db.Integer, primary_key=True)
    user = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
    name = db.Column(db.String(30))
    value = db.Column(db.String(192))


group_permission_association_table = db.Table('group_permission',
                                              db.Column('group_id', db.Integer, db.ForeignKey('group.id')),
                                              db.Column('permission_id', db.Integer, db.ForeignKey('permission.id'))
                                              )


class Group(db.Model):
    __tablename__ = 'group'
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(30))
    permissions = db.relationship("Permission", secondary=group_permission_association_table)

    def toJSON(self):
        return {
            'name': self.name
        }


class Permission(db.Model):
    __tablename__ = 'permission'
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(30))