增加数据库读取群成员备注功能

Author: 0honus0

setup.py

@@ -24,6 +24,7 @@
     install_requires=[
         "requests",
         "pydantic",
+        "protobuf==3.20.1",
     ],
     long_description=long_description,
     long_description_content_type="text/markdown",

wechatrobot/Api.py

@@ -2,6 +2,8 @@
 import requests
 import json
 from .Modles import *
+import base64
+from wechatrobot import ChatRoomData_pb2 as ChatRoom
 
 class Api:
     port : int = 18888
@@ -151,6 +153,22 @@ def GetPictureBySql(self):
             self.db_handle = self.GetDBHandle()
         sql = "select usrName,bigHeadImgUrl from ContactHeadImgUrl" 
         return self.QueryDatabase(db_handle=self.db_handle,sql=sql)
+
+    def GetAllGroupMembersBySql(self) -> Dict:
+        group_data = {} #{"group_id" : { "wxID" : "displayName"}}
+        if not self.db_handle:
+            self.db_handle = self.GetDBHandle()
+        sql = "select ChatRoomName,RoomData from ChatRoom"
+        GroupMemberList = self.QueryDatabase(db_handle = self.db_handle, sql = sql)['data']
+        chatroom = ChatRoom.ChatRoomData()
+        for index in range(1 , len(GroupMemberList)):
+            group_member = {}
+            chatroom.ParseFromString(bytes(base64.b64decode(GroupMemberList[index][1])))
+            for k in chatroom.members:
+                if k.displayName != "":
+                    group_member[k.wxID] = k.displayName
+            group_data[GroupMemberList[index][0]] = group_member
+        return group_data
     #自定义]
 
     def post(self , type : int, params : Body) -> Dict:

wechatrobot/ChatRoomData.proto

@@ -0,0 +1,21 @@
+syntax = "proto3";
+package RS_WXBOT_COM.protobuf;
+ 
+message ChatRoomData {
+
+  message ChatRoomMember {
+      string wxID = 1;
+      string displayName = 2;
+      int32 state = 3;
+  }
+
+  repeated ChatRoomMember members = 1;
+
+  int32 field_2 = 2;
+  int32 field_3 = 3;
+  int32 field_4 = 4;
+  int32 room_capacity = 5;
+  int32 field_6 = 6;
+  int64 field_7 = 7;
+  int64 field_8 = 8;
+}