initial commit

[pwp.git] / app / models.py

from config import ADMINS
from enum import IntEnum
import DB
#from flask_login import mixins

class Format(IntEnum):
    jpg = 1
    raw = 2
    tiff = 3

class Role(IntEnum):
    read = 1
    write = 2
    admin = 3

class User(DB.baseorm.BaseOrm):
    __tablename__ = 'user'
    
    @property
    def name(self):
        return self.__name

    @name.setter
    def name(self, name):
        self.__name = name
        self.notify(self)

    @property
    def email(self):
        return self.__email

    @email.setter
    def email(self, email):
        self.__email = email
        self.notify(self)

    @property
    def username(self):
        return self.__username

    @username.setter
    def username(self, username):
        self.__username = username
        self.notify(self)

    @property
    def password(self):
        return self.__password

    @password.setter
    def password(self, password):
        self.__password = password
        self.notify(self)

    def __init__(self, name, email, username, password):
        super().__init__()
        self.email = email
        self.id = -1
        self.name = name
        self.username = username
        self.password = password
        
    def is_authenticated(self):
        return True

    def is_active(self):
        return True

    def is_anonymous(self):
        return False
    
    def is_admin(self):
        return self.email in ADMINS

    def __repr__(self):
        return '<User ID: %s Name: %s Email: %s username: %s>' % (self.id, self.name,  self.email,  self.username)
        
class Portfolio(DB.baseorm.BaseOrm):
    __tablename__ = 'portfolio'
    
    @property
    def name(self):
        return self.__name

    @name.setter
    def name(self, name):
        self.__name = name

    @property
    def public(self):
        return self.__public
        
    @public.setter
    def public(self, public):
        self.__public = public
        
    @property
    def visible(self):
        return self.__visible
        
    @visible.setter
    def visible(self, visible):
        self.__visible = visible
        
    @property
    def user(self):
        return self.__user
        
    @user.setter
    def user(self, user):
        self.__user = user
        
    def __init__(self, name, user):
        super().__init__()
        self.id = -1
        self.name = name
        self.public = False
        self.visible = False
        if not isinstance(user, User):
            raise Exception("{0}: Not a 'User'".format(user))
        self.user = user
        
    def set_user_count(self, usercount):
        self.usercount = usercount

    def __repr__(self):
        return '<Portfolio Name: %s Owner: %s>' % (self.name,  self.user)

class Album(DB.baseorm.BaseOrm):
    __tablename__ = 'album'

    @property
    def name(self):
        return self.__name

    @name.setter
    def name(self, name):
        self.__name = name

    @property
    def public(self):
        return self.__public
        
    @public.setter
    def public(self, public):
        self.__public = public
        
    @property
    def visible(self):
        return self.__visible
        
    @visible.setter
    def visible(self, visible):
        self.__visible = visible
        
    @property
    def portfolio(self):
        return self.__portfolio
        
    @portfolio.setter
    def portfolio(self, portfolio):
        self.__portfolio = portfolio
        
    def __init__(self, name, portfolio):
        super().__init__()
        self.id = -1
        self.name = name
        self.public = False
        self.visible = False
        if not isinstance(portfolio, Portfolio):
            raise Exception("{0}: Not a 'Portfolio'".format(portfolio))
        self.portfolio = portfolio
        
    def set_user_count(self, usercount):
        self.usercount = usercount

    def __repr__(self):
        return '<Album Name: %r Portfolio: %r>' % (self.name,  self.portfolio)

class Photo(DB.baseorm.BaseOrm):
    __tablename__ = 'photo'

    @property
    def name(self):
        return self.__name

    @name.setter
    def name(self, name):
        self.__name = name

    @property
    def public(self):
        return self.__public
        
    @public.setter
    def public(self, public):
        self.__public = public
        
    @property
    def visible(self):
        return self.__visible
        
    @visible.setter
    def visible(self, visible):
        self.__visible = visible
        
    @property
    def album(self):
        return self.__album
        
    @album.setter
    def album(self, album):
        self.__album = album
        
    @property
    def file(self):
        return self.__file

    @file.setter
    def file(self, file):
        self.__file = file

    @property
    def exif(self):
        return self.__exif

    @exif.setter
    def exif(self, exif):
        self.__exif = exif

    @property
    def format(self):
        return self.__format
        
    @format.setter
    def format(self, format):
        self.__format = format
        
    def __init__(self, name, album):
        super().__init__()
        self.id = -1
        self.name = name
        self.public = False
        self.visible = False
        if not isinstance(album, Album):
            raise Exception("{0}: Not a 'Album'".format(album))
        self.album = album
        self.file = None
        self.exif = None
        self.format = Format.raw.value
        
    def __repr__(self):
        return '<Photo Name: %r File: %r Format: %r Album: %r>' % (self.name,  self.file,  self.format,  self.album)

class AccessRight(DB.baseorm.BaseOrm):
    __tablename__ = 'accessright'

    @property
    def perm(self):
        return self.__perm

    @perm.setter
    def perm(self, perm):
        self.__perm = perm

    @property
    def user(self):
        return self.__user
        
    @user.setter
    def user(self, user):
        self.__user = user

    def __init__(self, user):
        super().__init__()
        self.id = -1
        self.perm = Role.read.value
        if not isinstance(user, User):
            raise Exception("{0}: Not a 'User'".format(user))
        self.user = user

    def __repr__(self):
        return '<AccessRight Right: %r User: %r>' % (self.perm,  self.user)

#class MyAnonymous(mixins.AnonymousUserMixin):
#    id = -1
#    name = 'Anonymous User'
#    email = 'no@email.com'
#    username = 'None'
#    
#    @property
#    def is_authenticated(self):
#        return False
#
#    @property
#    def is_admin(self):
#        return False
#
#    def get_id(self):
#        return self.id
#        
#    def __repr__(self):
#        return '<User Name: %r Email: %r username: %r>' % (self.name,  self.email,  self.username)