136 lines
5.2 KiB
Python
136 lines
5.2 KiB
Python
#!/usr/bin/python
|
||
# -*- coding: utf-8 -*-
|
||
# @version : 1.0
|
||
# @Create Time : 2022/3/15 20:44
|
||
# @File : oauth.py
|
||
# @IDE : PyCharm
|
||
# @desc : 简要说明
|
||
|
||
import requests
|
||
from core.logger import logger
|
||
from utils.wx.wx_access_token import WxAccessToken
|
||
from redis.asyncio import Redis
|
||
|
||
|
||
class WXOAuth:
|
||
|
||
def __init__(self, rd: Redis, index: int = 0):
|
||
"""
|
||
初始化微信认证
|
||
:param index: 选择小程序，0：微信服务端
|
||
"""
|
||
# 重试次数
|
||
self.retry_count = 5
|
||
self.appid = None
|
||
self.secret = None
|
||
self.rd = rd
|
||
self.tab_name = None
|
||
if index == 0:
|
||
self.tab_name = "wx_server"
|
||
|
||
async def __get_settings(self, retry: int = 3):
|
||
"""
|
||
获取配置信息
|
||
"""
|
||
if not self.tab_name:
|
||
logger.error(f"请选择认证的微信平台")
|
||
wx_config = []
|
||
self.appid = wx_config.get("wx_server_app_id")
|
||
self.secret = wx_config.get("wx_server_app_secret")
|
||
|
||
async def get_code2session(self, code: str) -> dict:
|
||
"""
|
||
通过微信用户临时登录凭证 code 进行校验，获取用户openid，与 session 信息
|
||
|
||
官方文档：https://developers.weixin.qq.com/miniprogram/dev/api-backend/open-api/login/auth.code2Session.html
|
||
:param code: 登录时获取的 code
|
||
:return: 正确：{'session_key': 'F8/5LZrdtINYLPEdUJgXXQ==', 'openid': 'okLlC5Kcv7DH2J99dz-Z2FwJeEeU'}
|
||
:return: 报错：{'errcode': 40029, 'errmsg': 'invalid code, rid: 62308e5d-0b0b697e-1db652eb'}
|
||
"""
|
||
if not self.appid or not self.secret:
|
||
await self.__get_settings()
|
||
api = "https://api.weixin.qq.com/sns/jscode2session"
|
||
params = {
|
||
"appid": self.appid,
|
||
"secret": self.secret,
|
||
"js_code": code,
|
||
"grant_type": "authorization_code"
|
||
}
|
||
response = requests.get(url=api, params=params)
|
||
result = response.json()
|
||
if "openid" not in result:
|
||
logger.error(f"微信校验失败：{result}, code：{code}")
|
||
else:
|
||
logger.info(f"微信校验成功：{result}, code：{code}")
|
||
return result
|
||
|
||
async def get_phone_number(self, code: str):
|
||
"""
|
||
获取微信用户手机号
|
||
|
||
官方文档：https://developers.weixin.qq.com/miniprogram/dev/api-backend/open-api/phonenumber/phonenumber.getPhoneNumber.html
|
||
:param code: 动态令牌。可通过动态令牌换取用户手机号。
|
||
:return: 成功：{'errcode': 0, 'errmsg': 'ok', 'phone_info': {'phoneNumber': '15093430559'
|
||
, 'purePhoneNumber': '15093430559', 'countryCode': '86', 'watermark': {'timestamp': 1647355468, 'appid': 'wx069c452f9a733df1'}}}
|
||
失败：{'errcode': 40001, 'errmsg': 'invalid credential, access_token is invalid or not latest rid: 62690257-2894b530-58c6fcf3'}
|
||
"""
|
||
if not self.appid or not self.secret:
|
||
await self.__get_settings()
|
||
api = "https://api.weixin.qq.com/wxa/business/getuserphonenumber"
|
||
at = WxAccessToken(self.appid, self.secret, self.rd)
|
||
access_token = await at.get()
|
||
if not access_token.get("status", False):
|
||
result = {'errcode': 40001, 'errmsg': '获取微信令牌失败'}
|
||
# print(result)
|
||
logger.error(f"获取微信用户手机号失败：{result}")
|
||
return result
|
||
params = {
|
||
"access_token": access_token.get("token"),
|
||
}
|
||
data = {
|
||
"code": code,
|
||
}
|
||
response = requests.post(url=api, params=params, json=data)
|
||
result = response.json()
|
||
if result.get("errcode", 0) == 0:
|
||
# print("获取微信用户手机号成功", result)
|
||
logger.info(f"获取微信用户手机号成功：{result}, code：{code}")
|
||
else:
|
||
# print("获取微信用户手机号失败", result)
|
||
logger.error(f"获取微信用户手机号失败：{result}, code：{code}")
|
||
if result.get("errcode", 0) == 40001:
|
||
await at.update()
|
||
if self.retry_count > 0:
|
||
logger.error(f"重试获取微信手机号，重试剩余次数, {self.retry_count}")
|
||
self.retry_count -= 1
|
||
return await self.get_phone_number(code)
|
||
return result
|
||
|
||
async def parsing_phone_number(self, code: str):
|
||
"""
|
||
解析微信用户手机号
|
||
:param code: 动态令牌。可通过动态令牌换取用户手机号。
|
||
:return:
|
||
"""
|
||
result = await self.get_phone_number(code)
|
||
if result.get("errcode") == 0:
|
||
phone_info = result["phone_info"]
|
||
assert isinstance(phone_info, dict)
|
||
return phone_info["phoneNumber"]
|
||
return None
|
||
|
||
async def parsing_openid(self, code: str):
|
||
"""
|
||
解析openid
|
||
:param code: 动态令牌。可通过动态令牌换取用户手机号。
|
||
:return: openid | None
|
||
"""
|
||
result = await self.get_code2session(code)
|
||
if "openid" in result:
|
||
return result["openid"]
|
||
return None
|
||
|
||
|
||
# if __name__ == '__main__':
|
||
# WXOAuth().get_code2session("063mPDFa1v16PC0yqhGa1uQ86t4mPDFV")
|