def modify():
if isinstance(request.json, dict):
rule_id = request.json.get('rule_id', None)
path = request.json.get('path', None)
filename = request.json.get('filename', None)
write_log = request.json.get('write_log', None)
send_mail = request.json.get('send_mail', None)
comment = request.json.get('comment', None)
if isinstance(rule_id, int):
rule = Rule.query.filter_by(rule_id=rule_id).first()
if rule is not None:
if isinstance(path, str):
rule.path = path
if isinstance(filename, str):
rule.filename = filename
if isinstance(write_log, bool):
rule.write_log = write_log
if isinstance(send_mail, bool):
rule.send_mail = send_mail
if isinstance(comment, str):
rule.comment = comment
db.session.add(rule)
db.session.commit()
return jsonify(Response.success('修改成功'))
else:
return jsonify(Response.failed('规则不存在'))
else:
return jsonify(Response.invalid('参数无效'))
else:
return jsonify(Response.invalid('参数无效'))
def modify():
if isinstance(request.json, dict):
filename = request.json.get('filename', None)
new_filename = request.json.get('new_filename', None)
content = request.json.get('content', None)
if isinstance(filename, str):
filename = secure_filename(filename)
path = join(UPLOAD_PATH, filename)
if exists(path):
if isinstance(content, str):
file = open(path, 'w')
file.write(content)
file.close()
if isinstance(new_filename, str):
new_filename = secure_filename(new_filename)
rename(path, join(UPLOAD_PATH, new_filename))
return jsonify(Response.success('修改成功'))
else:
return jsonify(Response.failed('文件不存在'))
else:
return jsonify(Response.invalid('参数无效'))
else:
return jsonify(Response.invalid('参数无效'))
def add():
if isinstance(request.json, dict):
path = request.json.get('path', None)
filename = request.json.get('filename', None)
write_log = request.json.get('write_log', None)
send_mail = request.json.get('send_mail', None)
comment = request.json.get('comment', None)
if isinstance(path, str) and isinstance(filename, str) and isinstance(
write_log, bool) and isinstance(
send_mail, bool) and isinstance(comment, str):
rule = Rule.query.filter_by(path=path).first()
if rule is None:
rule = Rule(path=path,
filename=filename,
write_log=write_log,
send_mail=send_mail,
comment=comment)
db.session.add(rule)
db.session.commit()
return jsonify(Response.success('添加成功'))
else:
return jsonify(Response.failed('已经存在此规则'))
else:
return jsonify(Response.invalid('参数无效'))
else:
return jsonify(Response.invalid('参数无效'))
def modify():
if isinstance(request.json, dict):
TEMP_FILE_SAVE = request.json.get('TEMP_FILE_SAVE', None)
RECV_MAIL_ADDR = request.json.get('RECV_MAIL_ADDR', None)
ADMIN_PASSWORD = request.json.get('ADMIN_PASSWORD', None)
ORIGINAL_ADMIN_PASSWORD = request.json.get('ORIGINAL_ADMIN_PASSWORD',
None)
if isinstance(TEMP_FILE_SAVE, bool):
cached_config.TEMP_FILE_SAVE = TEMP_FILE_SAVE
if isinstance(RECV_MAIL_ADDR, str):
cached_config.RECV_MAIL_ADDR = RECV_MAIL_ADDR
if isinstance(ADMIN_PASSWORD, str) and isinstance(
ORIGINAL_ADMIN_PASSWORD, str):
if compare_digest(cached_config.ADMIN_PASSWORD,
passwd_hash(ORIGINAL_ADMIN_PASSWORD,
LOGIN_SALT)):
cached_config.ADMIN_PASSWORD = passwd_hash(
ADMIN_PASSWORD, LOGIN_SALT)
return jsonify(Response.success('修改成功'))
else:
return jsonify(Response.failed('原密码错误'))
return jsonify(Response.success('修改成功'))
else:
return jsonify(Response.invalid('参数无效'))
def preview():
if isinstance(request.json, dict):
filename = request.json.get('filename', None)
if isinstance(filename, str):
filename = secure_filename(filename)
path = join(TEMP_FILE_PATH, filename)
if exists(path) and getsize(path) < MAX_PREVIEW_SIZE:
file = open(path, 'rb')
content = file.read()
try:
content = content.decode('utf-8')
return jsonify(Response.success('', content))
except Exception:
return jsonify(Response.failed('文件过大或者文件不是纯文本, 无法预览'))
else:
return jsonify(Response.failed('文件不存在'))
else:
return jsonify(Response.invalid('参数无效'))
else:
return jsonify(Response.invalid('参数无效'))
def add():
file = request.files.get('file', None)
if file:
filename = secure_filename(file.filename)
path = join(UPLOAD_PATH, filename)
if not exists(path):
file.save(path)
return jsonify(Response.success('上传成功'))
else:
return jsonify(Response.failed('已存在同名文件'))
else:
return jsonify(Response.invalid('无效参数'))
def delete():
if isinstance(request.json, dict):
filename = request.json.get('filename', None)
if isinstance(filename, str):
filename = secure_filename(filename)
path = join(UPLOAD_PATH, filename)
if exists(path):
unlink(path)
return jsonify(Response.success('删除成功'))
else:
return jsonify(Response.failed('删除的文件不存在'))
else:
return jsonify(Response.invalid('参数无效'))
else:
return jsonify(Response.invalid('参数无效'))
def delete():
if isinstance(request.json, dict):
rule_id = request.json.get('rule_id', None)
if isinstance(rule_id, int):
rule = Rule.query.filter_by(rule_id=rule_id).first()
if rule is not None:
db.session.delete(rule)
db.session.commit()
return jsonify(Response.success('删除成功'))
else:
return jsonify(Response.failed('规则不存在'))
else:
return jsonify(Response.invalid('参数无效'))
else:
return jsonify(Response.invalid('参数无效'))
def download():
if isinstance(request.json, dict):
filename = request.json.get('filename', None)
if isinstance(filename, str):
filename = secure_filename(filename)
path = join(TEMP_FILE_PATH, filename)
if exists(path):
return send_file(path,
as_attachment=True,
attachment_filename=filename)
else:
return jsonify(Response.failed('文件不存在')), 400
else:
return jsonify(Response.invalid('参数无效')), 400
else:
return jsonify(Response.invalid('参数无效')), 400
def login():
if isinstance(request.json, dict):
password = request.json.get('password', None)
username = request.json.get('username', None)
if isinstance(username, str) and isinstance(
password, str) and compare_digest(
cached_config.ADMIN_PASSWORD,
passwd_hash(password, LOGIN_SALT)):
token = sign_token().decode()
system_log = SystemLog(
log_content=f'Admin login with username [{username}]')
db.session.add(system_log)
db.session.commit()
return jsonify(Response.success('登录成功', token))
else:
return jsonify(Response.failed('用户名或密码错误'))
else:
return jsonify(Response.invalid('无效请求'))
def status():
if jwt_auth:
return jsonify(Response.success())
else:
return jsonify(Response.failed())