Enhancements with ChatPhoto.save(...) method

Author: MarcoBuster

botogram/objects/media.py

@@ -55,6 +55,15 @@ class ChatPhoto(BaseObject, mixins.FileMixin):
     }
     _check_equality_ = "small_file_id"
 
+    def save(self, *args, small=False, **kwargs):
+        """Workaround for dealing with big and small chat photos"""
+        if small:
+            self.file_id = self.small
+        else:
+            self.file_id = self.big
+        super(ChatPhoto, self).save(*args, **kwargs)
+        del self.file_id
+
 
 class Photo(mixins.FileMixin):
     """Custom representation of a photo

botogram/objects/mixins.py

@@ -581,16 +581,9 @@ class FileMixin:
     """Add some methods for files"""
 
     @_require_api
-    def save(self, path, big=True):
+    def save(self, path):
         """Save the file to a particular path"""
-        if isinstance(self, ChatPhoto):
-            if big:
-                _file_id = self.big
-            else:
-                _file_id = self.small
-        else:
-            _file_id = self.file_id
-        response = self._api.call("getFile", {"file_id": _file_id})
+        response = self._api.call("getFile", {"file_id": self.file_id})
 
         # Save the file to the wanted path
         downloaded = self._api.file_content(response["result"]["file_path"])

docs/api/telegram.rst

@@ -2456,14 +2456,14 @@ about its business.
 
       The string ID of the 160x160 version of the chat photo
 
-   .. py:method:: save(path [, big=True])
+   .. py:method:: save(path [, small=False])
 
       Save the image represented to a file located by *path*. Be aware that
       Telegram does not provide the name of the original file sent by its
       sender. This should be generated as part of the path.
 
       :param str path: The file name path locating where the image should be saved.
-      :param bool big: Whether it should save the big or the small version of the chat photo
+      :param bool small: Whether it should save the big or the small version of the chat photo
 
    .. versionadded:: 0.7