def validate_for_api(self, ):
valid = super(BaseForm, self).validate()
# 如果校验失败，返回自定义函数
if not valid:
# form errors
log.error(self.errors)
raise ParameterException(msg=self.errors)
return self
def deleteNode(form):
try:
model = Node.query.filter_by(node_id=form.node_id.data).first()
if model is None:
return ajaxResponse.fail(message='node_id 不存在，请更换')
db.session.delete(model)
db.session.commit()
return ajaxResponse.success(message="删除成功")
except Exception as e:
log.error("错误 - {}".format(e))
db.session.rollback()
raise ServerError()
def getNode(form):
try:
model = Node.query.filter_by(node_id=form.node_id.data).first()
if model is None:
return ajaxResponse.fail('node不存在')
data = {
'node_id': model.node_id,
'node_name': model.node_name,
'ip': model.ip,
'port': model.port,
'note': model.note,
'status': model.status
}
return ajaxResponse.success(message="保存成功", data=data)
except Exception as e:
log.error("错误 - {}".format(e))
raise ServerError()
def getToken(form):
try:
log.info("成功 - Parameter:{}".format(form))
user = User.objects(secret=form.secret.data).first()
if user:
expiration = current_app.config['TOKEN_EXPIRATION']
role = {1: "UserScope", 2: "AdminScope"}
token = generate_auth_token(secret=form.secret.data,
auth=role[user.role],
expiration=expiration)
model = {'token': token.decode('ascii')}
return ajaxResponse.success(data=model, message="返回token成功")
else:
return ajaxResponse.fail(message="secret无效")
except Exception as e:
log.error("错误 - {}".format(e))
raise ServerError()
def getNodesList():
try:
models = Node.query.all()
datas = []
for model in models:
data = {
'node_id': model.node_id,
'node_name': model.node_name,
'ip': model.ip,
'port': model.port,
'note': model.note,
'status': model.status
}
datas.append(data)
return ajaxResponse.success(message="保存成功", data=datas)
except Exception as e:
log.error("错误 - {}".format(e))
raise ServerError()
def editNode(form):
try:
model = Node.query.filter_by(node_id=form.node_id.data).first()
if model is None:
return ajaxResponse.fail(message='node不存在')
model.node_name = form.node_name.data
model.ip = form.ip.data
model.port = form.port.data
model.note = form.note.data
try:
db.session.commit()
except Exception:
db.session.rollback()
return ajaxResponse.fail('更新失败,可能部分字段已存在')
return ajaxResponse.success(message="保存成功")
except Exception as e:
log.error("错误 - {}".format(e))
raise ServerError()
def framework_error(e):
"""
AOP,全局异常处理
:param e:
:return:
"""
log.error(e.description)
if isinstance(e, APIException):
return e
if isinstance(e, HTTPException):
code = e.code
msg = e.description
error_code = 1007
return APIException(msg, code, error_code)
else:
# 调试模式
# log
if not app.config['DEBUG']:
return ServerError()
else:
raise e
def addNode(form: AddNodeForm):
try:
hasName = Node.query.filter_by(node_name=form.node_name.data).first()
if hasName:
return ajaxResponse.fail(message='node_name 已存在，请更换')
hasName = Node.query.filter_by(ip=form.ip.data,
port=form.port.data).first()
if hasName:
return ajaxResponse.fail(message='ip:port 已存在，请更换')
model = Node()
model.node_name = form.node_name.data
model.ip = form.ip.data
model.port = form.port.data
model.note = form.note.data
db.session.add(model)
db.session.commit()
return ajaxResponse.success(message="新增成功")
except Exception as e:
log.error("错误 - {}".format(e))
db.session.rollback()
raise ServerError()
def getProjectsList(form):
try:
return ajaxResponse.success(message="保存成功")
except Exception as e:
log.error("错误 - {}".format(e))
raise ServerError()