微信登录

Flask - 加上微信登陆 - 回调

Flask - 加上微信登陆 - 回调

login.html扫码操作后到/session路由

  1. #微信登陆+设置session
  2. @Wechat.route("/session")
  3. def update_session():
  4.     #1.拿code
  5.     code = request.args.get("code")
  6.     if not code:
  7.         return "ERR_INVALID_CODE", 400
  8.     next = request.args.get("next", "/")
  10.     #2.向微信服务器发送https请求，获取access_token
  11.     url1 = "https://api.weixin.qq.com/sns/oauth2/access_token?appid=¥s&secret=¥s&code=¥s&grant_type=authorization_code" ¥(app_id,app_secret,code)
  12.     #发送请求，只传url说明用post
  13.     response1 = urllib.request.urlopen(url1)
  15.     #获取响应数据，微信返回的json数据
  16.     json_str1 = response1.read()
  17.     respond_dict1 = json.loads(json_str1)
  19.     #json提取access_token
  20.     if "errcode" in respond_dict1:
  21.         return "获取access_token失败"
  22.     access_token = respond_dict1.get("access_token")#access_token
  23.     open_id = respond_dict1.get("openid")#用户所有微信开发平台编号
  26.     #3.向微信服务器发送https请求，获取用户资料
  27.     url2 = "https://api.weixin.qq.com/sns/userinfo?access_token=¥s&openid=¥s&lang=zh_CN"¥(access_token,open_id)
  28.     response2 = urllib.request.urlopen(url2)
  30.     #获取响应数据，微信返回的json数据
  31.     user_json_str = response2.read()
  32.     user_respond_dict = json.loads(user_json_str)
  35.     #json提取access_token
  36.     if "errcode" in user_respond_dict:
  37.         return "获取用户信息失败"
  38.     else:
  40.         #用户数据：
  41.         openid = user_respond_dict.get("openid")#用户单个微信开发平台编号
  42.         unionid = user_respond_dict.get("unionid")#用户所有微信开发平台编号
  43.         nickname = user_respond_dict.get("nickname")#用户名字
  44.         sex = user_respond_dict.get("sex")#用户性别1男2女
  45.         province = user_respond_dict.get("province")#用户省份
  46.         city = user_respond_dict.get("city")#用户城市
  47.         country = user_respond_dict.get("country")#用户国家
  48.         headimgurl = user_respond_dict.get("headimgurl")#用户头像
  50.         #设置session("unionid",unionid)
  51.         session["unionid"] = unionid
  52.         #设置session("openid",openid)
  53.         session["openid"] = openid
  54.         #设置session("nickname",nickname)
  55.         session["nickname"] = nickname
  56.         #设置session("headimgurl",headimgurl)
  57.         session["headimgurl"] = headimgurl
  58.         #从浏览器拿state
  59.         state = request.args.get("state")
  60.         #设置session("state",state)
  61.         session["state"] = state
  62.         #开启session时间限制
  63.         session.permanent = True
  65.         # print(openid,unionid,nickname,sex,province,city,country,headimgurl)
  66.         try:
  67.             #查询数据库的Wechatdatabase表，看有没有这个id（数据表的名字=这里的unionid）
  68.             wechatdata = Wechatdatabase.query.filter_by(unionid=unionid).first()
  70.             #如果有就过了
  71.             if wechatdata:
  72.                 wechatdata.nickname = nickname
  73.                 wechatdata.province = province
  74.                 wechatdata.city = city
  75.                 wechatdata.country =country
  76.                 # 提交操作到数据库
  77.                 db.session.commit()
  78.             #如果没有就新增
  79.             else:
  80.                 try:
  81.                     #新id进来 = 数据库名(数据库表列1名=啥名,数据库表列2名=啥东西)
  82.                     #new_wechat_id = Wechatdatabase(openid=openid,sex=sex,province=province,city=city,country=country,unionid=unionid,points=50)
  83.                     new_wechat_id = Wechatdatabase(openid=openid,nickname=nickname,sex=sex,province=province,city=city,country=country,unionid=unionid)
  84.                     db.session.add(new_wechat_id)
  85.                     db.session.commit()
  86.                 except Exception as e:
  87.                     print(e)
  88.                     flash("添加id失败!")
  89.                     db.session.rollback()
  90.         except:
  91.             pass
  92.     return redirect(url_for("A.index"))

最后一句return redirect(url_for(‘A.index’))
意思是返回主页的方法
A是Flask蓝图名字
index是def index():方法

¥要改回来百分号