class Register():
__option_entity = None
__user_entity = None
__helpers = None
__logger = None
def __init__(self):
self.__option_entity = Option_Entity()
self.__user_entity = User_Entity()
self.__helpers = Helpers()
self.__logger = self.__helpers.get_logger(__name__)
def username_used(self, username):
return False if self.__user_entity.get_one_by_username(
username) == False else True
def email_used(self, email):
return False if self.__user_entity.get_one_by_email(
email) == False else True
def create_user(self, user_data):
status = True
status &= (self.__user_entity.insert_one(
{
"username": user_data["username"],
"email": user_data["email"],
"password": user_data["password"],
"first_name": user_data["first_name"],
"last_name": user_data["last_name"],
"is_superuser": False,
"is_active": True,
"is_staff": False
}) != False)
return status
class Install():
__options = [
{"key": "app_installed", "value": "true", "autoload": True},
{"key": "app_description", "value": "", "autoload": False},
{"key": "google_analytics_account", "value": "", "autoload": True},
{"key": "reset_mails_messages_count", "value": "5", "autoload": False},
{"key": "reset_mails_expire_after", "value": "24", "autoload": False},
{"key": "access_tokens_expire_after", "value": "48", "autoload": False},
{"key": "prometheus_token", "value": "", "autoload": False}
]
__admin = {
"username": "",
"email": "",
"password": "",
"is_superuser": True,
"is_active": True,
"is_staff": False
}
__option_entity = None
__user_entity = None
__helpers = None
__logger = None
__acl = None
def __init__(self):
self.__option_entity = Option_Entity()
self.__user_entity = User_Entity()
self.__helpers = Helpers()
self.__acl = ACL()
self.__logger = self.__helpers.get_logger(__name__)
def is_installed(self):
return False if self.__option_entity.get_one_by_key("app_installed") is False else True
def set_app_data(self, name, email, url):
self.__options.append({"key": "app_name", "value": name, "autoload": True})
self.__options.append({"key": "app_email", "value": email, "autoload": True})
self.__options.append({"key": "app_url", "value": url, "autoload": True})
def set_admin_data(self, username, email, password):
self.__admin["username"] = username
self.__admin["email"] = email
self.__admin["password"] = password
def init_base_acl(self, user_id):
self.__acl.truncate_default_permissions()
self.__acl.new_role("super_admin")
self.__acl.new_role("normal_user")
self.__acl.new_permission("Manage Settings", self.__acl.get_content_type_id("auth", "user"), "manage_settings")
self.__acl.add_permission_to_role("super_admin", "manage_settings")
self.__acl.add_role_to_user("super_admin", user_id)
return True
def install(self):
try:
execute_from_command_line(["manage.py", "migrate"])
except Exception as e:
self.__logger.error("Error While Running Migrations: %s" % e)
return False
status = True
status &= self.__option_entity.insert_many(self.__options)
user = self.__user_entity.insert_one(self.__admin)
status &= (user is not False)
if user is not False:
status &= self.init_base_acl(user.id)
return status
class Register():
__notification_entity = None
__option_entity = None
__user_entity = None
__helpers = None
__logger = None
__acl = None
def __init__(self):
self.__acl = ACL()
self.__option_entity = Option_Entity()
self.__user_entity = User_Entity()
self.__helpers = Helpers()
self.__notification_entity = Notification_Entity()
self.__logger = self.__helpers.get_logger(__name__)
def username_used(self, username):
return False if self.__user_entity.get_one_by_username(
username) is False else True
def email_used(self, email):
return False if self.__user_entity.get_one_by_email(
email) is False else True
def create_user(self, user_data):
status = True
user = self.__user_entity.insert_one({
"username":
user_data["username"],
"email":
user_data["email"],
"password":
user_data["password"],
"first_name":
user_data["first_name"],
"last_name":
user_data["last_name"],
"is_superuser":
False,
"is_active":
True,
"is_staff":
False
})
if user is not False:
self.__acl.add_role_to_user("normal_user", user.id)
self.__notification_entity.insert_one({
"highlight":
"$APP_NAME",
"notification":
_("Welcome, Hurry up and create your first host!"),
"url":
reverse('app.web.admin.hosts.list'),
"type":
"message",
"delivered":
False,
"user_id":
user.id,
"task_id":
None,
"host_id":
None
})
status &= (user is not False)
return status
class Install():
__option_entity = None
__user_entity = None
__options = [{
"key": "app_installed",
"value": "true",
"autoload": True
}, {
"key": "app_description",
"value": "",
"autoload": False
}, {
"key": "google_analytics_account",
"value": "",
"autoload": True
}, {
"key": "reset_mails_messages_count",
"value": "5",
"autoload": False
}, {
"key": "reset_mails_expire_after",
"value": "24",
"autoload": False
}, {
"key": "access_tokens_expire_after",
"value": "48",
"autoload": False
}]
__admin = {
"username": "",
"email": "",
"password": "",
"is_superuser": True,
"is_active": True,
"is_staff": False
}
__helpers = None
__logger = None
def __init__(self):
self.__option_entity = Option_Entity()
self.__user_entity = User_Entity()
self.__helpers = Helpers()
self.__logger = self.__helpers.get_logger(__name__)
def is_installed(self):
return False if self.__option_entity.get_one_by_key(
"app_installed") == False else True
def set_app_data(self, name, email, url):
self.__options.append({
"key": "app_name",
"value": name,
"autoload": True
})
self.__options.append({
"key": "app_email",
"value": email,
"autoload": True
})
self.__options.append({
"key": "app_url",
"value": url,
"autoload": True
})
def set_admin_data(self, username, email, password):
self.__admin["username"] = username
self.__admin["email"] = email
self.__admin["password"] = password
def install(self):
try:
execute_from_command_line(["manage.py", "migrate"])
except Exception as e:
self.__logger.error("Error While Running Migrations: %s" % e)
return False
status = True
status &= self.__option_entity.insert_many(self.__options)
status &= (self.__user_entity.insert_one(self.__admin) != False)
return status
