Update DirectMessage model
authorHarmon <Harmon758@gmail.com>
Fri, 21 Jun 2019 01:33:27 +0000 (20:33 -0500)
committerHarmon <Harmon758@gmail.com>
Fri, 21 Jun 2019 04:03:23 +0000 (23:03 -0500)
tweepy/models.py

index bd98ab95ada8e4e677bc877f3d6952b03fc45824..86155888db95248a9e665a73947551cfe1e37cc2 100644 (file)
@@ -217,15 +217,24 @@ class DirectMessage(Model):
     @classmethod
     def parse(cls, api, json):
         dm = cls(api)
+        if "event" in json:
+            json = json["event"]
         for k, v in json.items():
-            if k == 'sender' or k == 'recipient':
-                setattr(dm, k, User.parse(api, v))
-            elif k == 'created_at':
-                setattr(dm, k, parse_datetime(v))
-            else:
-                setattr(dm, k, v)
+            setattr(dm, k, v)
         return dm
 
+    @classmethod
+    def parse_list(cls, api, json_list):
+        if isinstance(json_list, list):
+            item_list = json_list
+        else:
+            item_list = json_list['events']
+
+        results = ResultSet()
+        for obj in item_list:
+            results.append(cls.parse(api, obj))
+        return results
+
     def destroy(self):
         return self._api.destroy_direct_message(self.id)