mappings: Reddit capture NoneType for media_type

Sieboldianus · May 16, 2023 · b1412d5 · b1412d5
1 parent 2410fae
commit b1412d5
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/resources/mappings/field_mapping_reddit.py b/resources/mappings/field_mapping_reddit.py
@@ -142,16 +142,17 @@ def value_count(x):
         if is_text:
             post_record.post_type = lbsn.Post.TEXT
         elif is_media:
-            if is_media.get("type") == "imgur.com":
+            media_type = is_media.get("type")
+            if media_type and media_type == "imgur.com":
                 post_record.post_type = lbsn.Post.IMAGE
-            elif is_media.get("type").startswith("youtube"):
+            elif media_type and media_type.startswith("youtube"):
                 post_record.post_type = lbsn.Post.VIDEO
             else:
                 # dig deeper
                 media_type = is_media.get("oembed").get("type")
-                if media_type == "video":
+                if media_type and media_type == "video":
                     post_record.post_type = lbsn.Post.VIDEO
-                elif media_type == "rich":
+                elif media_type and media_type == "rich":
                     post_record.post_type = lbsn.Post.OTHER
         elif is_url:
             if is_url.endswith((".jpg", ".webp", ".jpeg", ".png", ".gif")):