Queer European MD passionate about IT
Просмотр исходного кода

Make `general_handler` handle photo updates

Davte 4 лет назад
Родитель
Сommit
92e39e7ed7
2 измененных файлов с 7 добавлено и 5 удалено
  1. 1 1
      davtelepot/__init__.py
  2. 6 4
      davtelepot/bot.py

+ 1 - 1
davtelepot/__init__.py

@@ -11,7 +11,7 @@ __author__ = "Davide Testa"
 __email__ = "davide@davte.it"
 __credits__ = ["Marco Origlia", "Nick Lee @Nickoala"]
 __license__ = "GNU General Public License v3.0"
-__version__ = "2.6.15"
+__version__ = "2.6.16"
 __maintainer__ = "Davide Testa"
 __contact__ = "t.me/davte"
 

+ 6 - 4
davtelepot/bot.py

@@ -934,10 +934,10 @@ class Bot(TelegramBot, ObjectWithDatabase, MultiLanguageObject):
 
     async def photo_message_handler(self, update, user_record, language=None):
         """Handle `photo` message update."""
-        logging.info(
-            "A photo message update was received, "
-            "but this handler does nothing yet."
-        )
+        return await self.general_handler(update=update,
+                                          user_record=user_record,
+                                          language=language,
+                                          update_type='photo')
 
     async def sticker_message_handler(self, update, user_record, language=None):
         """Handle `sticker` message update."""
@@ -1003,6 +1003,8 @@ class Bot(TelegramBot, ObjectWithDatabase, MultiLanguageObject):
         user_id = update['from']['id'] if 'from' in update else None
         if update_type not in self.individual_handlers:
             self.individual_handlers[update_type] = dict()
+        if update_type not in self.handlers:
+            self.handlers[update_type] = OrderedDict()
         if user_id in self.individual_handlers[update_type]:
             replier = self.individual_handlers[update_type][user_id]
             del self.individual_handlers[update_type][user_id]