最近在学习python,并对开发框架的登录模块进行完善,总结一份记录。
首先,需要安装PyJWT模块,这是用于实现JWT校验的常用模块之一。可以使用pip命令来安装该模块,例如:
pip install PyJWT
然后,需要编写登录接口的代码,其中可以使用requests模块来向远程的SSO服务器发起get请求,获取到服务器返回的token并将其存储在header中。示例代码如下:
import requests
from flask import Flask, request
app = Flask(__name__)
@app.route('/login', methods=['GET'])
def login():
ticket = request.args.get('ticket', 'QWERTYUIOPCAS')
service = request.args.get('service', 'http://127.0.0.1:5000')
sso_url = 'http://sso.service.com?ticket={}&service={}'.format(ticket, service)
response = requests.get(sso_url)
token = response.json()['token']
headers = {'Authorization': 'Bearer {}'.format(token)}
# 将token存储在header中,以便后续接