微信登录需要域名

from flask import Flask, request, redirect
import requests

app = Flask(__name__)

@app.route('/wechat/callback')
def wechat_callback():
    code = request.args.get('code')
    if not code:
        return 'Code not found', 400

    # 使用code换取access_token和openid
    token_url = 'https://api.weixin.qq.com/sns/oauth2/access_token'
    params = {
        'appid': 'YOUR_APP_ID',
        'secret': 'YOUR_APP_SECRET',
        'code': code,
        'grant_type': 'authorization_code'
    }
    response = requests.get(token_url, params=params)
    if response.status_code != 200:
        return 'Failed to get access token', 500

    data = response.json()
    access_token = data.get('access_token')
    openid = data.get('openid')

    # 使用access_token和openid获取用户信息
    user_info_url = 'https://api.weixin.qq.com/sns/userinfo'
    params = {
        'access_token': access_token,
        'openid': openid
    }
    response = requests.get(user_info_url, params=params)
    if response.status_code != 200:
        return 'Failed to get user info', 500

    user_info = response.json()
    # 处理用户信息，如保存到数据库等

    return redirect('/success')  # 跳转到成功页面

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