def wrapper(*args, **kwargs):
user_name = request.cookies.get("user_name")
print(user_name)
token = request.cookies.get("token")
print(token)
if user_name != None or token != None:
helper = user_controller.user_controller()
real_token = helper.get_user_token(user_name=user_name)
if real_token == token:
ret = func(*args, **kwargs) # 被装饰函数
return ret
else:
return jsonify(
{
"code": "20000799",
"message": "access failed , token or username is not right."
}
)
else:
return jsonify(
{
"code": "20000799",
"message": "access failed , no login in"
}
)
def all_users_delete():
if request.method == "GET":
help = user_controller.user_controller()
help.user_info_delete()
return jsonify(
{
"result": "ok"
}
)
def user_delete_by_name(user_name):
if request.method == "GET":
help = user_controller.user_controller()
result = help.user_info_delete()
return jsonify(
{
"code": result["result"]["resultCode"],
"message": result["result"]["resultMessage"]
}
)
def login():
username = request.form.get('username')
password = request.form.get('password')
user_c = user_controller.user_controller()
result = user_c.login(user_name=username, user_password=password)
final_result = {
"code": result["result"]["resultCode"],
"message": result["result"]["resultMessage"],
"result": {
"username": result["userInfo"]["username"],
"token": result["userInfo"]["token"]
}
}
return jsonify(final_result)
def register():
data = request.get_data()
json_data = json.loads(data)
username = json_data['username']
password = json_data['password']
user_email = json_data['email']
user_c = user_controller.user_controller()
result = user_c.register(user_name=username, user_password=password, user_email=user_email)
final_result = {
"code": result["result"]["resultCode"],
"message": result["result"]["resultMessage"]
}
return jsonify(final_result)
def get_cluster_users():
if request.method == "GET":
helper = user_controller.user_controller()
user_infos = helper.local_user_info()
result = []
count_result = len(user_infos)
if count_result == 0:
result = []
else:
for info in user_infos:
result.append({
"user_name": info.user_name,
"user_from": info.user_from,
"user_email": info.user_email,
"update_time": info.update_time
})
# return result
return jsonify(
{
"result": result,
"total": count_result
}
)
def get_user_info():
if request.method == "GET":
user_name = request.cookies.get("user_name")
helper = user_controller.user_controller()
user_infos = helper.get_user_info("admin")
return jsonify(user_infos)