1

messages/base.py

@@ -1,3 +1,4 @@
+import traceback
 from datetime import datetime
 from store.mongo import MongoStore
 
@@ -14,7 +15,10 @@ class Base:
         return dict()
         
     def user(self):
-        return self.instance.user
+        if(hasattr(self.instance, 'user')):
+            return self.instance.user
+        
+        return None
 
     def persists(self):
         if config()['mongo']['enabled'] != 'on':
@@ -22,41 +26,39 @@ class Base:
 
         try:
             store = MongoStore()
-            store.set_collection('user')
 
             user = self.user()
+            if user is not None:
+                store.set_collection('user')
 
-            store.replace_one({
-                "id": user.id
-            }, {
-                "id": user.id,
-                "shortId": user.shortId,
-                "nickname": user.nickname,
-                "gender": user.gender,
-                "avatar_thumb": user.avatarThumb.urlList[0],
-                "followInfo": {
-                    "followingCount": user.followInfo.followingCount,
-                    "followerCount": user.followInfo.followerCount
-                }
-            }, upsert=True)
+                if not store.exists({'id': user.id}):
+                    store.insert_one({
+                        'id': user.id,
+                        'shortId': user.shortId,
+                        'nickname': user.nickname,
+                        'gender': user.gender
+                    })
 
             store.set_collection(self.instance.common.method)
-
             msg = {
                 "msgId": self.instance.common.msgId,
                 "roomId": self.instance.common.roomId,
-                "userId": user.id,
                 'content': self.format_content(),
                 'created_at': datetime.today().replace(microsecond=0)
             }
 
+            if user is not None:
+                msg.update({
+                    'userId': user.id
+                })
+
             if len(self.extra_info()):
                 msg.update(self.extra_info())
 
             store.insert_one(msg)
         except Exception as e:
             print(self.instance.common.method + ' persists error')
-
+            print(traceback.format_exc())
 
     def __str__(self):
         pass

messages/roomuserseq.py

@@ -7,6 +7,11 @@ class RoomUserSeqMessage(Base):
     def __init__(self):
         self.instance = message_pb2.RoomUserSeqMessage()
 
+    def extra_info(self):
+        return {
+            'total': self.instance.total,
+        }
+
     def format_content(self):
         return self.instance.totalUserStr
 

messages/utils.py

@@ -74,9 +74,9 @@ def decodeMsg(messages):
             elif message.method == 'WebcastRoomUserSeqMessage':
                 room_user_seq_message = RoomUserSeqMessage() 
                 room_user_seq_message.set_payload(message.payload)
-                # room_user_seq_message.persists()
+                room_user_seq_message.persists()
 
                 print(f"\n{YELLOW}[+] {room_user_seq_message} {RESET}")
 
         except Exception as e:
-            print(e)
+            print(e)