Added the ability to show meda as evidence

Author: LuisMayo

ace-attorney-reddit-bot

@@ -1,10 +1,20 @@
+import requests
+
 class Comment:
   def __init__(self, tweet):
     self.author = Author(tweet.user.name)
     self.body = tweet.full_text
     if (len(self.body) == 0):
         self.body = '...'
     self.score = 0
+    if (hasattr(tweet,'extended_entities') and tweet.extended_entities is not None
+    and 'media' in tweet.extended_entities and len(tweet.extended_entities['media']) > 0):
+        url = tweet.extended_entities['media'][0]['media_url_https'] + '?format=png&name=small'
+        name = tweet.extended_entities['media'][0]['media_url_https'].split('/')[-1] + '.png'
+        response = requests.get(url)
+        with open(name, 'wb') as file:
+            file.write(response.content)
+        self.evidence = name
 
 class Author:
     def __init__(self, name):

comment_list_brige.py

@@ -131,16 +131,30 @@ def process_tweets():
                         except Exception as second_error:
                             print(second_error)
                         print(e)
-                    os.remove(output_filename)
+                    clean(thread, output_filename)
                 else:
                     try:
                         api.update_status('@' + tweet.author.screen_name + " There should be at least two people in the conversation", in_reply_to_status_id=tweet.id_str)
                     except Exception as e:
                         print(e)
             time.sleep(1)
         except Exception as e:
+            clean(thread, output_filename)
             print(e)
 
+
+def clean(thread, output_filename):
+    try:
+        os.remove(output_filename)
+    except Exception as second_e:
+        print(second_e)
+    try:
+        for comment in thread:
+            if (hasattr(comment, 'evidence')):
+                os.remove(comment.evidence)
+    except Exception as second_e:
+        print(second_e)
+
 def restore_account():
     global mention_queue
     global next_account