Python Flask框架入门_2.通过token认证验证API的访问权限

增加 POST 方法 bili，获取传入的 ordered_string 参数，调用算法生成 sign 签名返回 ：

1. import hashlib
2. from flask import Flask, request, jsonify
3. app = Flask(__name__)
4. @app.route("/bili", methods=["POST"])
5. def bili():
6.     """
7.     请求的数据格式要求：{ "ordered_string":"......" }
8.     :return:
9.     """
10.     ordered_string = request.json.get("ordered_string")
11.     if not ordered_string:
12.         return jsonify({"status": False, "error": "参数错误"})
13.     # 调用核心算法，生成sign签名
14.     encrypt_string = ordered_string + "560c52ccd288fed045859ed18bffd973"
15.     obj = hashlib.md5(encrypt_string.encode('utf-8'))
16.     sign = obj.hexdigest()
17.     return jsonify({"status": True, "data": sign})
20. if __name__ == '__main__':
21.     app.run(host="127.0.0.1",port=5000)

复制代码

postman 中调用 bili 方法示例：

通过调用 API 时传入的 token 参数值与授权码对比，验证用户是否具有API的访问权限：

1. import hashlib
2. from flask import Flask, request, jsonify
3. app = Flask(__name__)
5. # 将授权码信息直接预存到字典中
6. USER_DICT = {"fec22f96-2783-414a-a6e3-991d357c96e6":"李杨",
7. "18bc44aa-76c9-43e7-a5c9-3aeb7884d6f1":"张三"}
9. # 从文件中读取授权码信息存入字典中并返回
10. def get_user_dict():
11.     info_dict = {}
12.     with open("db.txt",mode='r',encoding='utf-8') as f:
13.         for line in f :
14.             line = line.strip()
15.             print(line)
16.             token, name = line.split(",")
17.             print(token, name)
18.             info_dict[token] = name
19.     return  info_dict
21. @app.route("/bili", methods=["POST"])
22. def bili():
23.     """
24.     请求URL中携带/bili?token=...
25.     请求的数据格式要求：{ "ordered_string":"......" }
26.     :return:
27.     """
28.     token = request.args.get("token")
29.     if not token:
30.         return jsonify({"status":False , 'error':"认证失败"})
32.     # 从文件中读取授权码的情况
33.     user_dict = get_user_dict()
34.     if token not in user_dict:
35.         return jsonify({"status":False , 'error':"认证失败"})
37.     # 从字典中读取授权码的情况
38.     if token not in USER_DICT:
39.         return jsonify({"status":False , 'error':"认证失败"})
41.     ordered_string = request.json.get("ordered_string")
42.     if not ordered_string:
43.         return jsonify({"status": False, "error": "参数错误"})
45.     # 调用核心算法，生成sign签名
46.     encrypt_string = ordered_string + "560c52ccd288fed045859ed18bffd973"
47.     obj = hashlib.md5(encrypt_string.encode('utf-8'))
48.     sign = obj.hexdigest()
49.     # 返回签名
50.     return jsonify({"status": True, "data": sign})
53. if __name__ == '__main__':
54.     app.run(host="127.0.0.1",port=5000)

复制代码

授权文件名 db.txt ，授权文件内容如下：

1. fec22f96-2783-414a-a6e3-991d357c96e6,李杨
2. 18bc44aa-76c9-43e7-a5c9-3aeb7884d6f1,张三

复制代码

 postman 中调用带 token 认证的 bili 方法示例：

 

来源：豆瓜网用户自行投稿发布，如果侵权，请联系站长删除