From 287878212885a13dd57a6801ce5d42cc4a7613a2 Mon Sep 17 00:00:00 2001
From: H Vs <vson.iwork@outlook.com>
Date: Tue, 21 Jan 2025 10:45:16 +0800
Subject: [PATCH] =?UTF-8?q?=E8=B0=83=E6=95=B4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 docker/entrypoint.sh |  2 +-
 wechat/gewe_chat.py  | 53 ++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 54 insertions(+), 1 deletion(-)

diff --git a/docker/entrypoint.sh b/docker/entrypoint.sh
index 544c545..d607c00 100644
--- a/docker/entrypoint.sh
+++ b/docker/entrypoint.sh
@@ -42,5 +42,5 @@ if [ "$environment" == "default" ]; then
     $AI_OPS_WECHAT_EXEC
 else
     # 非默认环境，使用 Gunicorn 启动
-    gunicorn -w 1 -b 0.0.0.0:5000 app:app
+    gunicorn -w 1 -b 0.0.0.0:5000 --timeout 1200 app:app
 fi
diff --git a/wechat/gewe_chat.py b/wechat/gewe_chat.py
index 2f620f7..0c49f31 100644
--- a/wechat/gewe_chat.py
+++ b/wechat/gewe_chat.py
@@ -272,6 +272,24 @@ class GeWeChatCom:
         response_object = response.json()
         return response_object.get('ret',None),response_object.get('msg',None),response_object.get('data',None)
     
+    def set_friend_remark(self,token_id, app_id,friend_wxid,remark):    
+        '''
+        设置好友备注
+        '''
+        api_url = f"{self.base_url}/v2/api/contacts/setFriendRemark"
+        headers = {
+            'X-GEWE-TOKEN': token_id,
+            'Content-Type': 'application/json'
+        }
+        data = {
+            "appId": app_id,
+            "wxid":friend_wxid,
+            "remark":remark
+        }
+        response = requests.post(url=api_url, headers=headers, data=json.dumps(data))
+        response_object = response.json()
+        return response_object.get('ret',None),response_object.get('msg',None),response_object.get('data',None)
+
 ###############################  消息模块   ############################### 
     def post_text(self,token_id,app_id,to_wxid,content):
         api_url = f"{self.base_url}/v2/api/message/postText"
@@ -552,6 +570,41 @@ class GeWeChatCom:
         response_object = response.json()
         return response_object.get('ret',None),response_object.get('msg',None),response_object.get('data',None)
 
+    def send_image_sns(self, token_id, app_id,infos:list): 
+        '''
+        发送图片朋友圈
+        '''
+        api_url = f"{self.base_url}/v2/api/sns/sendImageSns"
+        headers = {
+            'X-GEWE-TOKEN': token_id,
+            'Content-Type': 'application/json'
+        }
+        data = {
+            "appId": app_id,
+            "imgInfos": infos # 通过上传朋友圈图片接口获取
+        }
+        response = requests.post(url=api_url, headers=headers, data=json.dumps(data))
+        response_object = response.json()
+        return response_object.get('ret',None),response_object.get('msg',None),response_object.get('data',None)
+    
+    def upload_sns_image(self, token_id, app_id,img_urls:list):
+        '''
+        上传朋友圈图片
+        '''
+        api_url = f"{self.base_url}/v2/api/sns/uploadSnsImage"
+        headers = {
+            'X-GEWE-TOKEN': token_id,
+            'Content-Type': 'application/json'
+        }
+        data = {
+            "appId": app_id,
+            "imgUrls": img_urls
+        }
+        response = requests.post(url=api_url, headers=headers, data=json.dumps(data))
+        response_object = response.json()
+        return response_object.get('ret',None),response_object.get('msg',None),response_object.get('data',None)
+
+    
 ###############################     其他   ###############################
     def save_session_messages_to_cache(self, hash_key,object:object)->list:
         '''