Catch parse tweet bugs

Is there not a library for this?
StephanAkkerman · Nov 16, 2023 · a6fcfd5 · a6fcfd5
1 parent c5d5d42
commit a6fcfd5
Showing 1 changed file with 12 additions and 10 deletions.
diff --git a/src/util/parse_tweet.py b/src/util/parse_tweet.py
@@ -113,16 +113,18 @@ def parse_tweet(tweet: dict, update_tweet_id: bool = False):
     # Media
     media = []
     media_types = []
-    if "extended_entities" in tweet["legacy"].keys():
-        if "media" in tweet["legacy"]["extended_entities"].keys():
-            media = [
-                image["media_url_https"]
-                for image in tweet["legacy"]["extended_entities"]["media"]
-            ]
-            # photo, video
-            media_types = [
-                image["type"] for image in tweet["legacy"]["extended_entities"]["media"]
-            ]
+    if "legacy" in tweet.keys():
+        if "extended_entities" in tweet["legacy"].keys():
+            if "media" in tweet["legacy"]["extended_entities"].keys():
+                media = [
+                    image["media_url_https"]
+                    for image in tweet["legacy"]["extended_entities"]["media"]
+                ]
+                # photo, video
+                media_types = [
+                    image["type"]
+                    for image in tweet["legacy"]["extended_entities"]["media"]
+                ]
 
     # Remove t.co url from text
     text = remove_twitter_url_at_end(tweet["legacy"]["full_text"])