mirror of
https://gitlab.com/crafty-controller/crafty-4.git
synced 2024-08-30 18:23:09 +00:00
2a512d7273
Mostly just breaking up strings and comments into new lines Some strings dont require 'f' but keeping in for readability with the rest of the concatinated string
89 lines
2.4 KiB
Python
89 lines
2.4 KiB
Python
import logging
|
|
import datetime
|
|
|
|
from app.classes.shared.helpers import helper
|
|
|
|
try:
|
|
from peewee import (
|
|
SqliteDatabase,
|
|
Model,
|
|
CharField,
|
|
DoesNotExist,
|
|
AutoField,
|
|
DateTimeField,
|
|
)
|
|
from playhouse.shortcuts import model_to_dict
|
|
|
|
except ModuleNotFoundError as e:
|
|
helper.auto_installer_fix(e)
|
|
|
|
logger = logging.getLogger(__name__)
|
|
peewee_logger = logging.getLogger("peewee")
|
|
peewee_logger.setLevel(logging.INFO)
|
|
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()
|