ada/twitter_media_tool
Archived
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Small update to archival script to correct the archival of video files.

Browse source
This commit is contained in:
Alex Huddleston 2020-04-27 21:07:28 -05:00
parent ac2f4ebf95
commit 97a80494c8
1 changed files with 18 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
lib/archival.py
View file

@ -13,6 +13,23 @@ def archive_media(media_url):
return filename return filename
def filter_out_bad_content_type(video_variant_list):
filtered_variant_list = []
for video in video_variant_list:
if(video['content_type'] == 'video/mp4'):
filtered_variant_list.append(video)
return filtered_variant_list
def select_video_variant(video_variant_list):
filtered_variant_list = filter_out_bad_content_type(video_variant_list)
highest_bitrate_variant = filtered_variant_list[0]
for video in filtered_variant_list:
if(highest_bitrate_variant['bitrate'] < video['bitrate']):
highest_bitrate_variant = video
return highest_bitrate_variant['url']
def archive_media_status(status): def archive_media_status(status):
try: try:
if(not ('retweeted_status' in status._json.keys())): if(not ('retweeted_status' in status._json.keys())):
@ -25,7 +42,7 @@ def archive_media_status(status):
archive_media(media_dict['media_url'])) archive_media(media_dict['media_url']))
elif(media_dict['type'] == 'video'): elif(media_dict['type'] == 'video'):
archive_filenames.append(archive_media( archive_filenames.append(archive_media(
media_dict['video_info']['variants'][-1:][0]['url'])) select_video_variant(media_dict['video_info']['variants'])))
return archive_filenames return archive_filenames
except Exception as e: except Exception as e:
print(e) print(e)