80 lines
2.5 KiB
Python
80 lines
2.5 KiB
Python
"""俱乐部微信配置与 openid 解析。"""
|
||
import logging
|
||
import requests
|
||
from django.conf import settings
|
||
|
||
from jituan.constants import CLUB_ID_DEFAULT
|
||
from jituan.models import Club
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
def resolve_club_by_appid(app_id):
|
||
if not app_id:
|
||
return None
|
||
try:
|
||
return Club.query.filter(wx_appid=app_id, status=1).first()
|
||
except Exception:
|
||
return Club.objects.filter(wx_appid=app_id, status=1).first()
|
||
|
||
|
||
def get_club_wx_credentials(club_id):
|
||
"""优先读 club 表;xq 缺省时回退 settings 全局配置(兼容现网)。"""
|
||
club_id = club_id or CLUB_ID_DEFAULT
|
||
try:
|
||
club = Club.query.get(club_id=club_id)
|
||
except Club.DoesNotExist:
|
||
club = None
|
||
|
||
if club and club.wx_appid and club.wx_secret:
|
||
return club.wx_appid, club.wx_secret, club
|
||
|
||
if club_id == CLUB_ID_DEFAULT:
|
||
return (
|
||
getattr(settings, 'WEIXIN_APPID', ''),
|
||
getattr(settings, 'WEIXIN_SECRET', ''),
|
||
club,
|
||
)
|
||
return '', '', club
|
||
|
||
|
||
def jscode2session(code, club_id=None, app_id=None):
|
||
"""
|
||
按俱乐部微信配置换取 openid。
|
||
club_id 与 app_id 二选一;app_id 用于从小程序 appId 反查 club。
|
||
"""
|
||
club = None
|
||
if app_id:
|
||
club = resolve_club_by_appid(app_id)
|
||
club_id = club.club_id if club else club_id
|
||
appid, secret, club = get_club_wx_credentials(club_id)
|
||
if not appid or not secret:
|
||
return {'errmsg': '俱乐部微信配置未设置'}
|
||
|
||
url = 'https://api.weixin.qq.com/sns/jscode2session'
|
||
params = {
|
||
'appid': appid,
|
||
'secret': secret,
|
||
'js_code': code,
|
||
'grant_type': 'authorization_code',
|
||
}
|
||
try:
|
||
response = requests.get(url, params=params, timeout=10)
|
||
result = response.json()
|
||
if 'openid' in result:
|
||
return {
|
||
'openid': result['openid'],
|
||
'session_key': result.get('session_key', ''),
|
||
'unionid': result.get('unionid', ''),
|
||
'club_id': club_id or CLUB_ID_DEFAULT,
|
||
'wx_appid': appid,
|
||
}
|
||
errcode = result.get('errcode', 'unknown')
|
||
errmsg = result.get('errmsg', '未知错误')
|
||
return {'errmsg': f'[{errcode}]{errmsg}'}
|
||
except requests.exceptions.Timeout:
|
||
return {'errmsg': '请求微信接口超时'}
|
||
except Exception as e:
|
||
logger.exception('jscode2session 异常 club_id=%s', club_id)
|
||
return {'errmsg': f'请求微信接口异常: {str(e)}'}
|