Check key "document" exists before parsing (#2)

Check key "document" exists before parsing
Mte90 · Aug 22, 2019 · 203d84f · 203d84f
2 parents ef320ba + 9691da7
commit 203d84f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/downloader.py b/downloader.py
@@ -36,7 +36,7 @@
 for message in data['messages']:
     if 'media' in message:
         for term in search:
-            if re.search(term, message['media']['document']['attributes'][0]['file_name'], re.IGNORECASE):
+            if 'document' in message['media'] and re.search(term, message['media']['document']['attributes'][0]['file_name'], re.IGNORECASE):
                 print(' Download ' + message['media']['document']['attributes'][0]['file_name'])
                 response = urlopen('https://tg.i-c-a.su/media/' + config.get('channel', 'name') + '/' + str(message['id']))
                 file = open(config.get('channel', 'download') + message['media']['document']['attributes'][0]['file_name'], 'wb')