feat: 存储wxid与openId绑定

This commit is contained in:
hackrobot
2024-03-24 23:16:32 +08:00
parent e40688816d
commit e0e0e7eb57
5 changed files with 25 additions and 5 deletions

View File

@@ -25,17 +25,21 @@ from pydantic import BaseModel
# 保存好友信息
def saveWxidInfo(account_wxid, wxid):
def saveWxidInfo(account_wxid, wxid,openId=None):
allFriend = WXBotService.get_address_list(account_wxid)
for friendItem in allFriend['data']["contactList"]:
if friendItem['wxid'] == wxid:
# 存储小程序openid
if openId:
friendItem['openId'] = openId
friendItemObj = models.User(**friendItem)
db_user = crud.get_user_by_wxid(db=SessionLocal(), wxid=wxid)
if db_user:
print("wxid already registered")
# return db_user
else:
crud.create_user(db=SessionLocal(), user=friendItemObj)
new_user=crud.create_user(db=SessionLocal(), user=friendItemObj)
return new_user
# 发送小程序模版,建议远程更新