小明：最近我在研究一个系统，需要实现统一身份认证，你觉得怎么做比较好？

小李：统一身份认证通常可以用OAuth2.0或者JWT来实现。你可以用Python的Flask框架来做，这样比较灵活。

小明：那如果还想结合“免费代理价”呢？是不是要和用户权限挂钩？

小李：对，你可以在用户登录后根据其身份判断是否允许使用免费代理价。比如设置一个权限字段，登录成功后返回这个字段。

小明：能给我看看代码示例吗？

小李：当然可以。下面是一个简单的JWT认证示例，同时结合了代理价逻辑：

from flask import Flask, request, jsonify
import jwt
import datetime

app = Flask(__name__)

SECRET_KEY = 'your-secret-key'

def generate_token(user_id):
    payload = {
        'user_id': user_id,
        'exp': datetime.datetime.utcnow() + datetime.timedelta(hours=1)
    }
    return jwt.encode(payload, SECRET_KEY, algorithm='HS256')

@app.route('/login', methods=['POST'])
def login():
    data = request.json
    username = data.get('username')
    password = data.get('password')

    # 假设这里做用户名密码校验
    if username == 'admin' and password == '123456':
        token = generate_token(1)
        return jsonify({'token': token, 'is_free_proxy': True})
    else:
        return jsonify({'error': 'Invalid credentials'}), 401

@app.route('/proxy', methods=['GET'])
def get_proxy():
    token = request.headers.get('Authorization')
    if not token:
        return jsonify({'error': 'Missing token'}), 401

    try:
        payload = jwt.decode(token, SECRET_KEY, algorithms=['HS256'])
        user_id = payload['user_id']
        is_free_proxy = payload.get('is_free_proxy', False)

        if is_free_proxy:
            return jsonify({'proxy': 'free.proxy.com'})
        else:
            return jsonify({'proxy': 'paid.proxy.com'})

    except jwt.ExpiredSignatureError:
        return jsonify({'error': 'Token expired'}), 401
    except jwt.InvalidTokenError:
        return jsonify({'error': 'Invalid token'}), 401

if __name__ == '__main__':
    app.run(debug=True)
    

登录统一用户基础平台

小明：这代码挺直观的，我明白了！不过还要注意安全性问题。

小李：没错，实际项目中还需要考虑更多细节，比如密钥管理、防止重放攻击等。

小明：谢谢你的帮助，我现在有方向了。

小李：不客气，有问题随时问我。