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 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)
