Fix isAt compacity and reduce the size of pickle

dragonetail · Mar 23, 2017 · 7781490 · 7781490
1 parent 9d9b343
commit 7781490
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/itchat/storage/__init__.py b/itchat/storage/__init__.py
@@ -32,9 +32,9 @@ def dumps(self):
         return {
             'userName'          : self.userName,
             'nickName'          : self.nickName,
-            'memberList'        : self.memberList,
-            'mpList'            : self.mpList,
-            'chatroomList'      : self.chatroomList,
+            'memberList'        : [dict(member) for member in self.memberList],
+            'mpList'            : [dict(mp) for mp in self.mpList],
+            'chatroomList'      : [dict(chatroom) for chatroom in self.chatroomList],
             'lastInputUserName' : self.lastInputUserName, }
     def loads(self, j):
         self.userName = j.get('userName', None)

diff --git a/itchat/storage/messagequeue.py b/itchat/storage/messagequeue.py
@@ -5,6 +5,8 @@
 
 class Queue(queue.Queue):
     def put(self, message):
+        if 'IsAt' in message:
+            message['isAt'] = message['IsAt']
         queue.Queue.put(self, Message(message))
 
 class Message(dict):