From c66da02f2b42e479190c912246913a8819f61bbf Mon Sep 17 00:00:00 2001 From: Alex Huddleston Date: Fri, 24 Apr 2020 17:56:54 -0500 Subject: [PATCH] Small changes. Will most likely move back to Tweepy. --- lib/archival.py | 12 +++++++++++- twitter_media_tool.py | 9 ++------- 2 files changed, 13 insertions(+), 8 deletions(-) diff --git a/lib/archival.py b/lib/archival.py index 73d7510..35a036a 100644 --- a/lib/archival.py +++ b/lib/archival.py @@ -1,6 +1,7 @@ #!/usr/bin/python3 from twitter.twitter_utils import parse_media_file +from twitter import Api def archive_media_status(media_object): try: @@ -11,4 +12,13 @@ def archive_media_status(media_object): temp_media_file.close() except Exception as e: print(e) - exit(1) \ No newline at end of file + exit(1) + +def twitter_archive_stream(twitter_api, authenticated_user): + tweet_stream = twitter_api.GetUserStream(replies='none', withuser='user') + # try: + # status_media_list = tweet_stream.status.media + # for media_file in status_media_list: + # archive_media_status(media_file) + # except Exception as e: + # print(e) \ No newline at end of file diff --git a/twitter_media_tool.py b/twitter_media_tool.py index 23faff7..7c652ab 100644 --- a/twitter_media_tool.py +++ b/twitter_media_tool.py @@ -3,18 +3,13 @@ from time import sleep from lib.setup import import_config_file from lib.authentication import authenticate_twitter -from lib.archival import archive_media_status +from lib.archival import twitter_archive_stream def main(): config = import_config_file() twitter_api = authenticate_twitter(config) authenticated_user = twitter_api.GetUser(screen_name=config['authenticated_user']) - try: - status_media_list = authenticated_user.status.media - for media_file in status_media_list: - archive_media_status(media_file) - except Exception as e: - print(e) + twitter_archive_stream(twitter_api, authenticated_user) if(__name__ == '__main__'):