diff --git a/lib/archival.py b/lib/archival.py index b979844..802a4c9 100644 --- a/lib/archival.py +++ b/lib/archival.py @@ -70,7 +70,19 @@ def archive_media_status(status): """ try: if(not ('retweeted_status' in status._json.keys())): - if('extended_entities' in status._json.keys()): + if('extended_tweet' in status._json.keys()): + if('extended_entities' in status.extended_tweet.keys()): + if('media' in status.extended_tweet['extended_entities'].keys()): + archive_filenames = [] + for media_dict in status.extended_tweet['extended_entities']['media']: + if(media_dict['type'] == 'photo'): + archive_filenames.append( + archive_media(media_dict['media_url'])) + elif(media_dict['type'] == 'video'): + archive_filenames.append( + archive_media(select_video_variant(media_dict['video_info']['variants']))) + return archive_filenames + elif('extended_entities' in status._json.keys()): if('media' in status.extended_entities.keys()): archive_filenames = [] for media_dict in status.extended_entities['media']: