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 __init__(self): self.__option_entity = Option_Entity() self.__user_entity = User_Entity() self.__helpers = Helpers() self.__token = Token() self.__profile_entity = Profile_Entity() self.__logger = self.__helpers.get_logger(__name__)
def __init__(self): self.__option_entity = Option_Entity() self.__user_entity = User_Entity() self.__host_entity = Host_Entity() self.__task_entity = Task_Entity() self.__profile_entity = Profile_Entity() self.__helpers = Helpers() self.__logger = self.__helpers.get_logger(__name__) self.__app_name = self.__option_entity.get_value_by_key("app_name").lower()
class Reset_Password(): __reset_request_entity = None __helpers = None __user_entity = None def __init__(self): self.__helpers = Helpers() self.__user_entity = User_Entity() self.__reset_request_entity = Reset_Request_Entity() def check_token(self, token): request = self.__reset_request_entity.get_one_by_token(token) if request is not False and timezone.now() < request.expire_at: return True return False def reset_password(self, token, new_password): request = self.__reset_request_entity.get_one_by_token(token) if request is not False and timezone.now() < request.expire_at: return self.__user_entity.update_password_by_email(request.email, new_password) return False def delete_reset_request(self, token): return self.__reset_request_entity.delete_one_by_token(token)
def __init__(self): self.__reset_request_entity = Reset_Request_Entity() self.__option_entity = Option_Entity() self.__helpers = Helpers() self.__user_entity = User_Entity() self.__job_entity = Job_Entity() messages_count_option = self.__option_entity.get_one_by_key( "reset_mails_messages_count") reset_expire_option = self.__option_entity.get_one_by_key( "reset_mails_expire_after") if messages_count_option != False: self.__messages_count_option = int(messages_count_option.value) if reset_expire_option != False: self.__reset_expire_option = int(reset_expire_option.value)
def __init__(self): self.__option_entity = Option_Entity() self.__user_entity = User_Entity() self.__helpers = Helpers() self.__data["AUTHOR"] = AUTHOR self.__data["COPYRIGHT"] = COPYRIGHT self.__data["LICENSE"] = LICENSE self.__data["VERSION"] = VERSION self.__data["MAINTAINER"] = MAINTAINER self.__data["EMAIL"] = EMAIL self.__data["STATUS"] = STATUS self.__data["REPO_URL"] = REPO_URL self.__data["AUTHOR_URL"] = AUTHOR_URL self.__data["RELEASES"] = RELEASES self.__data["SUPPORT_URL"] = SUPPORT_URL self.__logger = self.__helpers.get_logger(__name__)
class Metric(): __option_entity = None __user_entity = None __host_entity = None __task_entity = None __profile_entity = None __helpers = None __logger = None __app_name = "" def __init__(self): self.__option_entity = Option_Entity() self.__user_entity = User_Entity() self.__host_entity = Host_Entity() self.__task_entity = Task_Entity() self.__profile_entity = Profile_Entity() self.__helpers = Helpers() self.__logger = self.__helpers.get_logger(__name__) self.__app_name = self.__option_entity.get_value_by_key("app_name").lower() def get_all_users(self): return { "type": "count", "record": "%s_all_users" % self.__app_name, "count": self.__user_entity.count_all_users(), "comment": "Current All Users on System" } def get_all_profiles(self): return { "type": "count", "record": "%s_all_profiles" % self.__app_name, "count": self.__profile_entity.count_all_profiles(), "comment": "Current All Profiles on System" } def get_all_tasks(self): return { "type": "count", "record": "%s_all_tasks" % self.__app_name, "count": self.__task_entity.count_all_tasks(), "comment": "Current All Tasks on System" } def get_all_hosts(self): return { "type": "count", "record": "%s_all_hosts" % self.__app_name, "count": self.__host_entity.count_all_hosts(), "comment": "Current All Hosts on System" }
class Login(): __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 is_authenticated(self, request): if request.user and request.user.is_authenticated: return True else: return False def authenticate(self, username_email, password, request=None, with_login=True): is_email = False try: is_email = True if validate_email( username_email) == None else False except Exception as e: is_email = False if is_email: user = self.__user_entity.get_one_by_email(username_email) if user != False and user.check_password(password) == True: if with_login: self.login(request, user) return True else: return False else: user = authenticate(request=request, username=username_email, password=password) if user is not None: if with_login: self.login(request, user) return True else: return False def login(self, request, user): return login(request, user)
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
def __init__(self): self.__helpers = Helpers() self.__user_entity = User_Entity() self.__reset_request_entity = Reset_Request_Entity()
class Forgot_Password(): __reset_request_entity = None __option_entity = None __user_entity = None __job_entity = None __helpers = None __reset_expire_option = 24 __messages_count_option = 5 def __init__(self): self.__reset_request_entity = Reset_Request_Entity() self.__option_entity = Option_Entity() self.__helpers = Helpers() self.__user_entity = User_Entity() self.__job_entity = Job_Entity() messages_count_option = self.__option_entity.get_one_by_key( "reset_mails_messages_count") reset_expire_option = self.__option_entity.get_one_by_key( "reset_mails_expire_after") if messages_count_option != False: self.__messages_count_option = int(messages_count_option.value) if reset_expire_option != False: self.__reset_expire_option = int(reset_expire_option.value) def check_email(self, email): return True if self.__user_entity.get_one_by_email( email) != False else False def reset_request_exists(self, email): return self.__reset_request_entity.get_one_by_email(email) def is_spam(self, request): if request.messages_count >= self.__messages_count_option and timezone.now( ) < request.expire_at: return True return False def update_request(self, request): # Delete Old Request self.__reset_request_entity.delete_one_by_id(request.id) # Create a Fresh Request if timezone.now() > request.expire_at: return self.create_request(request.email) # Create from the Old Request request = self.__reset_request_entity.insert_one({ "email": request.email, "expire_at": request.expire_at, "messages_count": request.messages_count + 1 }) return request.token if request != False else False def create_request(self, email): request = self.__reset_request_entity.insert_one({ "email": email, "expire_after": self.__reset_expire_option, "messages_count": 0 }) return request.token if request != False else False def send_message(self, email, token): return self.__job_entity.insert_one({ "name": "reset_password_msg_for_%s" % (email), "executor": "forgot_password_email.Forgot_Password_Email", "parameters": { "recipient_list": [email], "token": token }, "interval": { "type": Job_Entity.ONCE } })
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 Forgot_Password(): __reset_request_entity = None __option_entity = None __user_entity = None __task_core = None __helpers = None __reset_expire_option = 24 __messages_count_option = 5 def __init__(self): self.__reset_request_entity = Reset_Request_Entity() self.__option_entity = Option_Entity() self.__helpers = Helpers() self.__user_entity = User_Entity() self.__task_core = Task_Core() messages_count_option = self.__option_entity.get_one_by_key( "reset_mails_messages_count") reset_expire_option = self.__option_entity.get_one_by_key( "reset_mails_expire_after") if messages_count_option: self.__messages_count_option = int(messages_count_option.value) if reset_expire_option: self.__reset_expire_option = int(reset_expire_option.value) def check_email(self, email): return True if self.__user_entity.get_one_by_email( email) is not False else False def reset_request_exists(self, email): return self.__reset_request_entity.get_one_by_email(email) def is_spam(self, request): if request.messages_count >= self.__messages_count_option and timezone.now( ) < request.expire_at: return True return False def update_request(self, request): # Delete Old Request self.__reset_request_entity.delete_one_by_id(request.id) # Create a Fresh Request if timezone.now() > request.expire_at: return self.create_request(request.email) # Create from the Old Request request = self.__reset_request_entity.insert_one({ "email": request.email, "expire_at": request.expire_at, "messages_count": request.messages_count + 1 }) return request.token if request is not False else False def create_request(self, email): request = self.__reset_request_entity.insert_one({ "email": email, "expire_after": self.__reset_expire_option, "messages_count": 0 }) return request.token if request is not False else False def send_message(self, email, token): app_name = self.__option_entity.get_value_by_key("app_name") app_email = self.__option_entity.get_value_by_key("app_email") app_url = self.__option_entity.get_value_by_key("app_url") user = self.__user_entity.get_one_by_email(email) return self.__task_core.delay( "forgot_password_email", { "app_name": app_name, "app_email": app_email, "app_url": app_url, "recipient_list": [email], "token": token, "subject": _("%s Password Reset") % (app_name), "template": "mails/reset_password.html", "fail_silently": False }, user.id)
class Profile(): __option_entity = Option_Entity() __user_entity = User_Entity() __helpers = Helpers() __logger = None __token = Token() __profile_entity = Profile_Entity() def __init__(self): self.__logger = self.__helpers.get_logger(__name__) def get_profile(self, user_id): profile_data = { "first_name": "", "last_name": "", "username": "", "email": "", "job_title": "", "company": "", "address": "", "github_url": "", "twitter_url": "", "facebook_url": "", "access_token": "", "refresh_token": "", "avatar": "" } user = self.__user_entity.get_one_by_id(user_id) profile = self.__profile_entity.get_profile_by_user_id(user_id) if user != False: profile_data["first_name"] = user.first_name profile_data["last_name"] = user.last_name profile_data["username"] = user.username profile_data["email"] = user.email profile_data["avatar"] = Gravatar(user.email).get_image() if profile != False: profile_data["job_title"] = profile.job_title profile_data["company"] = profile.company profile_data["address"] = profile.address profile_data["github_url"] = profile.github_url profile_data["twitter_url"] = profile.twitter_url profile_data["facebook_url"] = profile.facebook_url profile_data["access_token"] = profile.access_token profile_data["refresh_token"] = profile.refresh_token return profile_data def update_profile(self, user_id, user_data): user_data["user"] = user_id if self.__profile_entity.profile_exists(user_data["user"]): status = self.__profile_entity.update_profile(user_data) status &= self.__user_entity.update_one_by_id( user_data["user"], user_data) return status else: status = (self.__profile_entity.create_profile(user_data) != False) status &= self.__user_entity.update_one_by_id( user_data["user"], user_data) return status def update_access_token(self, user_id): token = self.__token.generate_token() while self.__profile_entity.token_used(token) != False: token = self.__token.generate_token() return token if self.__profile_entity.update_access_token( user_id, token) else False def update_refresh_token(self, user_id): token = self.__token.generate_token() while self.__profile_entity.token_used(token) != False: token = self.__token.generate_token() return token if self.__profile_entity.update_refresh_token( user_id, token) else False def get_profile_by_access_token(self, access_token): return self.__profile_entity.get_profile_by_access_token(access_token) def change_password(self, user_id, password): return self.__user_entity.update_password_by_user_id(user_id, password) def restore_session(self, user_id, request): return update_session_auth_hash( request, self.__user_entity.get_one_by_id(user_id)) def validate_password(self, user_id, password): return self.__user_entity.validate_password_by_user_id( user_id, password) def update_user(self, user_id, user_data): return self.__user_entity.update_one_by_id(self, user_id, user_data) def username_used_elsewhere(self, user_id, username): user = self.__user_entity.get_one_by_username(username) return False if user == False or user.id == user_id else True def email_used_elsewhere(self, user_id, email): user = self.__user_entity.get_one_by_email(email) return False if user == False or user.id == user_id else True
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
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 Context(): __data = {} __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.__data["AUTHOR"] = AUTHOR self.__data["COPYRIGHT"] = COPYRIGHT self.__data["LICENSE"] = LICENSE self.__data["VERSION"] = VERSION self.__data["MAINTAINER"] = MAINTAINER self.__data["EMAIL"] = EMAIL self.__data["STATUS"] = STATUS self.__data["REPO_URL"] = REPO_URL self.__data["AUTHOR_URL"] = AUTHOR_URL self.__data["RELEASES"] = RELEASES self.__data["SUPPORT_URL"] = SUPPORT_URL self.__logger = self.__helpers.get_logger(__name__) def push(self, new_data): self.__data.update(new_data) def load_options(self, options): options_to_load = {} for key in options.keys(): options_to_load[key] = options[key] if key not in self.__data.keys(): self.__data[key] = options[key] if len(options_to_load.keys()) > 0: new_options = self.__option_entity.get_many_by_keys( options_to_load.keys()) for option in new_options: self.__data[option.key] = option.value def autoload_options(self): options = self.__option_entity.get_many_by_autoload(True) for option in options: self.__data[option.key] = option.value def autoload_user(self, user_id): user_data = { "user_first_name": "", "user_last_name": "", "user_username": "", "user_email": "", "user_avatar": "" } if user_id is not None: user = self.__user_entity.get_one_by_id(user_id) if user is not False: user_data["user_first_name"] = user.first_name user_data["user_last_name"] = user.last_name user_data["user_username"] = user.username user_data["user_email"] = user.email user_data["user_avatar"] = Gravatar(user.email).get_image() self.__data.update(user_data) def get(self, key=None, default=None): if key is not None: return self.__data[key] if key in self.__data else default return self.__data