mirror of
https://gitlab.com/crafty-controller/crafty-4.git
synced 2024-08-30 18:23:09 +00:00
86 lines
2.6 KiB
Python
86 lines
2.6 KiB
Python
import os
|
|
import sys
|
|
import logging
|
|
import datetime
|
|
|
|
from app.classes.shared.helpers import helper
|
|
from app.classes.shared.console import console
|
|
|
|
logger = logging.getLogger(__name__)
|
|
peewee_logger = logging.getLogger('peewee')
|
|
peewee_logger.setLevel(logging.INFO)
|
|
|
|
try:
|
|
from peewee import *
|
|
from playhouse.shortcuts import model_to_dict
|
|
from enum import Enum
|
|
import yaml
|
|
|
|
except ModuleNotFoundError as e:
|
|
logger.critical("Import Error: Unable to load {} module".format(e.name), exc_info=True)
|
|
console.critical("Import Error: Unable to load {} module".format(e.name))
|
|
sys.exit(1)
|
|
|
|
database = SqliteDatabase(helper.db_path, pragmas={
|
|
'journal_mode': 'wal',
|
|
'cache_size': -1024 * 10})
|
|
|
|
#************************************************************************************************
|
|
# Roles Class
|
|
#************************************************************************************************
|
|
class Roles(Model):
|
|
role_id = AutoField()
|
|
created = DateTimeField(default=datetime.datetime.now)
|
|
last_update = DateTimeField(default=datetime.datetime.now)
|
|
role_name = CharField(default="", unique=True, index=True)
|
|
|
|
class Meta:
|
|
table_name = "roles"
|
|
database = database
|
|
|
|
#************************************************************************************************
|
|
# Roles Helpers
|
|
#************************************************************************************************
|
|
class helper_roles:
|
|
@staticmethod
|
|
def get_all_roles():
|
|
query = Roles.select()
|
|
return query
|
|
|
|
@staticmethod
|
|
def get_roleid_by_name(role_name):
|
|
try:
|
|
return (Roles.get(Roles.role_name == role_name)).role_id
|
|
except DoesNotExist:
|
|
return None
|
|
|
|
@staticmethod
|
|
def get_role(role_id):
|
|
return model_to_dict(Roles.get(Roles.role_id == role_id))
|
|
|
|
@staticmethod
|
|
def add_role(role_name):
|
|
role_id = Roles.insert({
|
|
Roles.role_name: role_name.lower(),
|
|
Roles.created: helper.get_time_as_string()
|
|
}).execute()
|
|
return role_id
|
|
|
|
@staticmethod
|
|
def update_role(role_id, up_data):
|
|
return Roles.update(up_data).where(Roles.role_id == role_id).execute()
|
|
|
|
@staticmethod
|
|
def remove_role(role_id):
|
|
with database.atomic():
|
|
role = Roles.get(Roles.role_id == role_id)
|
|
return role.delete_instance()
|
|
|
|
@staticmethod
|
|
def role_id_exists(role_id):
|
|
if not roles_helper.get_role(role_id):
|
|
return False
|
|
return True
|
|
|
|
roles_helper = helper_roles()
|