Fixes bug when to_data thumb for document and video

leandrotoledo · leandrotoledo · commit 185a9e991868 · 2015-07-20T08:01:14.000-03:00
diff --git a/telegram/document.py b/telegram/document.py
@@ -32,8 +32,9 @@ def de_json(data):
                         file_size=data.get('file_size', None))
 
     def to_dict(self):
-        data = {'file_id': self.file_id,
-                'thumb': self.thumb.to_dict()}
+        data = {'file_id': self.file_id}
+        if self.thumb:
+            data['thumb'] = self.thumb.to_dict()
         if self.file_name:
             data['file_name'] = self.file_name
         if self.mime_type:
diff --git a/telegram/video.py b/telegram/video.py
@@ -44,8 +44,9 @@ def to_dict(self):
         data = {'file_id': self.file_id,
                 'width': self.width,
                 'height': self.height,
-                'duration': self.duration,
-                'thumb': self.thumb.to_dict()}
+                'duration': self.duration}
+        if self.thumb:
+            data['thumb'] = self.thumb.to_dict()
         if self.mime_type:
             data['mime_type'] = self.mime_type
         if self.file_size:
