From 402e47e5d06ca1ff83e4b7d0f9e866b89fb0a249 Mon Sep 17 00:00:00 2001
From: hackrobot <xiaoshuang.eric@gmail.com>
Date: Wed, 10 Apr 2024 20:28:15 +0800
Subject: [PATCH] update

---
 models.py | 1 +
 utils.py  | 4 +++-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/models.py b/models.py
index 84d7c5e..2833177 100644
--- a/models.py
+++ b/models.py
@@ -10,6 +10,7 @@ class User(Base):
     id = Column(Integer, primary_key=True)
     wxid = Column(String) 
     openId = Column(String)
+    h5openid = Column(String)
     sex = Column(String) #性别 男/女
     city = Column(String) #城市
     country = Column(String) #"CN"
diff --git a/utils.py b/utils.py
index c93b30d..58b990c 100644
--- a/utils.py
+++ b/utils.py
@@ -25,13 +25,15 @@ from pydantic import BaseModel
 
 
 # 保存好友信息
-def saveWxidInfo(account_wxid, wxid,openId=None):
+def saveWxidInfo(account_wxid, wxid,openId=None,h5openid=None):
     allFriend = WXBotService.get_address_list(account_wxid)
     for friendItem in allFriend['data']["contactList"]:
         if friendItem['wxid'] == wxid:
             # 存储小程序openid
             if openId:
                 friendItem['openId'] = openId
+            if h5openid:
+                friendItem['h5openid'] = openId
             friendItemObj = models.User(**friendItem)
             db_user = crud.get_user_by_wxid(db=SessionLocal(), wxid=wxid)
             if db_user: