Safely fetch comment data, by checking availability before calling

Author: wkoot

instagram/models.py

@@ -94,10 +94,12 @@ def object_from_dictionary(cls, entry):
             for like in entry['likes']['data']:
                 new_media.likes.append(User.object_from_dictionary(like))
 
-        new_media.comment_count = entry['comments']['count']
-        new_media.comments = []
-        for comment in entry['comments']['data']:
-            new_media.comments.append(Comment.object_from_dictionary(comment))
+        new_media.comment_count = entry.get('comments', {}).get('count', 0)
+        if new_media.comment_count:
+            new_media.comments = []
+            if entry.get('comments', {}).get('data'):
+                for comment in entry['comments']['data']:
+                    new_media.comments.append(Comment.object_from_dictionary(comment))
 
         new_media.users_in_photo = []
         if entry.get('users_in_photo'):