class Role(db.Model, RestrictionsMixin, PhotoricMixin):
__tablename__ = 'roles'
__restrictions__ = {}
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(100), nullable=False, unique=True)
_users = db.relationship('User', secondary=UserRole)
@hybrid_property
def users(self):
return self._users
@users.setter
def users(self, users):
for user in users:
existing_user = User.get_by_name(name=user["name"])
if existing_user:
self._users.append(existing_user)
else:
return {
"message":
"User is not exists. Role can be assigned to existing user(-s) only"
}, 404
def __repr__(self):
return '<Role %r>' % self.name
class Group(db.Model, AllowancesMixin, PhotoricMixin):
__tablename__ = 'groups'
__allowances__ = {}
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(100), nullable=False, unique=True)
_users = db.relationship('User', secondary=UserGroup)
@hybrid_property
def users(self):
return self._users
@users.setter
def users(self, users):
for user in users:
existing_user = User.get_by_name(name=user["name"])
if existing_user:
self._users.append(existing_user)
else:
return {
"message":
"User does not exist. Only existing user(-s) can be included in a group"
}, 404
def __repr__(self):
return '<Group %r>' % self.name
class Menu(db.Model):
__tablename__='menus'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(100), unique=True, nullable=False)
html_class = db.Column(db.String(255), nullable=False, default='')
html_style = db.Column(db.String(255), nullable=False, default='')
items = db.relationship('MenuItem', back_populates='menu')
class MenuItem(db.Model):
__tablename__='menu_items'
id = db.Column(db.Integer, primary_key=True)
menu_id = db.Column(db.Integer, db.ForeignKey('menus.id'))
parent_id = db.Column(db.Integer, db.ForeignKey('menu_items.id'))
item_type = db.Column(db.String(20), default='plain')
name = db.Column(db.String(100), nullable=False, unique=True)
desc = db.Column(db.String(255), nullable=False)
item_target = db.Column(db.String(255), nullable=True)
icon_type = db.Column(db.String(50), nullable=True)
icon_src = db.Column(db.String(255), nullable=True)
visible = db.Column(db.Boolean, nullable=False, default=True)
anonym_only = db.Column(db.Boolean, nullable=False, default=False)
auth_req = db.Column(db.Boolean, nullable=False, default=False)
group_req = db.Column(db.String(100), nullable=True)
role_req = db.Column(db.String(100), nullable=True)
menu = db.relationship('Menu', back_populates='items')
children = db.relationship('MenuItem')
class Album(db.Model, PermissionsMixin):
__tablename__ = "albums"
__user_model__ = User
__permissions__ = dict(
owner=['create', 'read', 'update', 'delete', 'revoke'],
group=['read', 'update', 'revoke'],
other=['read']
)
id = db.Column(db.Integer, primary_key=True)
parent_id = db.Column(db.Integer, db.ForeignKey('albums.id'))
name = db.Column(db.String(100), unique=True, nullable=False)
description = db.Column(db.String(500), nullable=True)
keywords = db.Column(db.String(255), nullable=True)
icon_id = db.Column(db.Integer, nullable=True)
created_on = db.Column(db.DateTime, nullable=False, default=datetime.now)
is_published = db.Column(db.Boolean(), nullable=False, default=False)
published_on = db.Column(db.DateTime, nullable=True, index=True)
children_images = db.relationship(
'Image',
secondary=AlbumImage,
back_populates='parent_albums'
)
children_albums = db.relationship(
'Album', backref=backref('parent',
remote_side=[id])
)
def publish(self):
# mark gallery item as published, i.e. accessible for both registered and anonymous users
self.is_published = True
self.published_on = datetime.utcnow
def unpublish(self):
# mark gallery item as not published, i.e. accessible for both registered and anonymous users
self.is_published = False
def __repr__(self):
return '<Album %r>' % self.name
class NavbarItem(db.Model):
__tablename__='navbar_items'
id = db.Column(db.Integer, primary_key=True)
navbar_id = db.Column(db.Integer, db.ForeignKey('navbars.id'))
name = db.Column(db.String(100), unique=True, nullable=False)
item_type = db.Column(db.String(100), nullable=False)
item_target = db.Column(db.String(255), nullable=True)
icon_type = db.Column(db.String(50), nullable=True)
icon_src = db.Column(db.String(255), nullable=True)
visible = db.Column(db.Boolean, nullable=False, default=True)
anonym_only = db.Column(db.Boolean, nullable=False, default=False)
auth_req = db.Column(db.Boolean, nullable=False, default=False)
group_req = db.Column(db.String(100), nullable=True)
role_req = db.Column(db.String(100), nullable=True)
item_src = db.column_property('/' + item_type + '/' + name + '.html')
navbar = db.relationship('Navbar', back_populates='items')
class User(PhotoricMixin, UserMixin, db.Model):
__tablename__ = "users"
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String, nullable=False, unique=True)
email = db.Column(db.String, )
_password = db.Column(db.String, nullable=False)
_created_on = db.Column(db.DateTime,
nullable=False,
default=datetime.utcnow())
_is_active = db.Column(db.Boolean, nullable=False, default=True)
_last_login = db.Column(db.DateTime, nullable=True)
_roles = db.relationship('Role', secondary=UserRole)
_groups = db.relationship('Group', secondary=UserGroup)
@hybrid_property
def created_on(self):
return self._created_on # .strftime("%c")
@hybrid_property
def is_active(self):
return self._is_active
@is_active.setter
def is_active(self, status=True):
self._is_active = status
@property
def password(self):
return self._password
@password.setter
def password(self, password):
self._password = generate_password_hash(password, method='sha256')
def check_password(self, password):
return check_password_hash(self._password, password)
@hybrid_property
def last_login(self):
return self._last_login # .strftime("%c")
@last_login.setter
def last_login(self, trigger):
if trigger == "set":
self._last_login = datetime.utcnow()
db.session.commit()
else:
pass
@hybrid_property
def roles(self):
return self._roles
@roles.setter
def roles(self, roles):
missed = self._build_relationship("_roles", roles, Role, True)
if missed != []:
msg = "The following roles do not exist and were not added: {}.".format(
missed)
return {"message": msg}, 404
@hybrid_property
def groups(self):
return self._groups
@groups.setter
def groups(self, groups):
missed = self._build_relationship("_groups", groups, Group, True)
if missed != []:
msg = "The following groups do not exist and were not added: {}.".format(
missed)
return {"message": msg}, 404
# Required for administrative interface
def __unicode__(self):
return self.name
def __repr__(self):
return '<User %r>' % self.name