def simple_import(path_to_module): err_prefix = "Importing '%s' failed: " % path_to_module if not os.path.exists(path_to_module): raise DataError(err_prefix + 'File does not exist') try: return _import_module_by_path(path_to_module) except: raise DataError(err_prefix + get_error_message())
def delete(self, id): """ 删除host :param id: :return: """ try: hosts = model.Hosts result = hosts.get(hosts.id == id).delete_instance() return result except Exception: model.database.rollback() raise error.Database( error.data_delete_error, error.get_error_message(error.data_delete_error))
def simple_import(path_to_module): err_prefix = "Importing '%s' failed: " % path_to_module if not os.path.exists(path_to_module): raise DataError(err_prefix + 'File does not exist') moddir, modname = _split_path_to_module(path_to_module) try: try: module = __import__(modname) if normpath(moddir) != normpath(os.path.dirname(module.__file__)): del sys.modules[modname] module = __import__(modname) except: raise DataError(err_prefix + get_error_message()) finally: sys.path.pop(0) return module
def update(self, id, **kwargs): """ 修改host信息 :param id: :param kwargs: :return: """ kwargs["update_time"] = common.get_time_now() try: hosts = model.Hosts hosts_update = hosts.update(**kwargs).where( (hosts.id == id)).execute() return hosts_update except Exception as e: model.database.rollback() raise error.Database( error.data_save_error, error.get_error_message(error.data_save_error))
def update_result(self, case_id, case_result): """ 更新case的成功返回的result :param case_id: :param case_result: :return: """ try: cases = model.Cases cases_update = cases.update({ "result": case_result }).where((cases.id == case_id)).execute() return cases_update except Exception: model.database.rollback() raise error.Database( error.data_delete_error, error.get_error_message(error.data_delete_error))
def update_password(self, **kwargs): """ 修改用户密码 :param id: :param kwargs: :return: """ kwargs["update_time"] = common.get_time_now() kwargs["password"] = encode_password(kwargs["password"]) email = kwargs["email"] del kwargs["email"] try: users = model.Users result = users.update(**kwargs).where( (users.email == email)).execute() return result except Exception: model.database.rollback() raise error.Database( error.data_save_error, error.get_error_message(error.data_save_error))
def get(self, email): """ 验证用户密码 :param param: :return: """ try: validate.check_password(self.arguments) password = self.service.check(**self.arguments) if password == sha1_password(self.arguments["password"]): self.echoJson(0, {"info": "success"}) else: self.echo_error(error.password_error, error.get_error_message(error.password_error)) except error.RequestData as e: self.echo_error(e.code, e.message) except exceptions.RequestData as e: self.echoJson(e.code, e.message) except peewee.DoesNotExist: self.echoJson() except Exception as e: self.echoJson(exceptions.exception, str(e)) self.close_connect_and_finish()
def __init__(self, code, message=""): self.code = code self.message = error.get_error_message(code)