me/offline-twitter
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update command line output when you get rate limited

Browse Source
This commit is contained in:
Alessio 2024-06-10 20:37:43 -07:00
parent a1faacaf6b
commit 80f8ed0a5d
2 changed files with 39 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
cmd/twitter/helpers.go
View File

@ -36,7 +36,12 @@ func die(text string, display_help bool, exit_code int) {
/** /**
* Print a happy exit message and exit * Print a happy exit message and exit
*/ */
func happy_exit(text string) { func happy_exit(text string, was_rate_limited bool) {
if was_rate_limited {
fmt.Printf(terminal_utils.COLOR_YELLOW + text + terminal_utils.COLOR_RESET + "\n")
fmt.Printf(terminal_utils.COLOR_YELLOW + "Exiting early (rate limited)." + terminal_utils.COLOR_RESET + "\n")
os.Exit(1)
}
fmt.Printf(terminal_utils.COLOR_GREEN + text + terminal_utils.COLOR_RESET + "\n") fmt.Printf(terminal_utils.COLOR_GREEN + text + terminal_utils.COLOR_RESET + "\n")
fmt.Printf(terminal_utils.COLOR_GREEN + "Exiting successfully." + terminal_utils.COLOR_RESET + "\n") fmt.Printf(terminal_utils.COLOR_GREEN + "Exiting successfully." + terminal_utils.COLOR_RESET + "\n")
os.Exit(0) os.Exit(0)

53
cmd/twitter/main.go
View File

@ -237,7 +237,7 @@ func login(username string, password string) {
} }
profile.SaveSession(api) profile.SaveSession(api)
happy_exit("Logged in as " + string(api.UserHandle)) happy_exit("Logged in as "+string(api.UserHandle), false)
} }
/** /**
@ -272,7 +272,7 @@ func fetch_user(handle scraper.UserHandle) {
} }
download_user_content(handle) download_user_content(handle)
happy_exit("Saved the user") happy_exit("Saved the user", false)
} }
/** /**
@ -293,11 +293,11 @@ func fetch_tweet_only(tweet_identifier string) {
} }
log.Debug(tweet) log.Debug(tweet)
err = profile.SaveTweet(tweet) err2 := profile.SaveTweet(tweet)
if err != nil { if err2 != nil {
die(fmt.Sprintf("Error saving tweet: %s", err.Error()), false, 4) die(fmt.Sprintf("Error saving tweet: %s", err2.Error()), false, 4)
} }
happy_exit("Saved the tweet") happy_exit("Saved the tweet", errors.Is(err, scraper.ErrRateLimited))
} }
/** /**
@ -318,7 +318,7 @@ func fetch_tweet_conversation(tweet_identifier string, how_many int) {
} }
profile.SaveTweetTrove(trove, true) profile.SaveTweetTrove(trove, true)
happy_exit(fmt.Sprintf("Saved %d tweets and %d users", len(trove.Tweets), len(trove.Users))) happy_exit(fmt.Sprintf("Saved %d tweets and %d users", len(trove.Tweets), len(trove.Users)), errors.Is(err, scraper.ErrRateLimited))
} }
/** /**
@ -339,7 +339,10 @@ func fetch_user_feed(handle string, how_many int) {
} }
profile.SaveTweetTrove(trove, true) profile.SaveTweetTrove(trove, true)
happy_exit(fmt.Sprintf("Saved %d tweets, %d retweets and %d users", len(trove.Tweets), len(trove.Retweets), len(trove.Users))) happy_exit(
fmt.Sprintf("Saved %d tweets, %d retweets and %d users", len(trove.Tweets), len(trove.Retweets), len(trove.Users)),
errors.Is(err, scraper.ErrRateLimited),
)
} }
func get_user_likes(handle string, how_many int) { func get_user_likes(handle string, how_many int) {
@ -354,7 +357,10 @@ func get_user_likes(handle string, how_many int) {
} }
profile.SaveTweetTrove(trove, true) profile.SaveTweetTrove(trove, true)
happy_exit(fmt.Sprintf("Saved %d tweets, %d retweets and %d users", len(trove.Tweets), len(trove.Retweets), len(trove.Users))) happy_exit(
fmt.Sprintf("Saved %d tweets, %d retweets and %d users", len(trove.Tweets), len(trove.Retweets), len(trove.Users)),
errors.Is(err, scraper.ErrRateLimited),
)
} }
func get_followees(handle string, how_many int) { func get_followees(handle string, how_many int) {
@ -370,7 +376,7 @@ func get_followees(handle string, how_many int) {
profile.SaveTweetTrove(trove, true) profile.SaveTweetTrove(trove, true)
profile.SaveAsFolloweesList(user.ID, trove) profile.SaveAsFolloweesList(user.ID, trove)
happy_exit(fmt.Sprintf("Saved %d followees", len(trove.Users))) happy_exit(fmt.Sprintf("Saved %d followees", len(trove.Users)), errors.Is(err, scraper.ErrRateLimited))
} }
func get_followers(handle string, how_many int) { func get_followers(handle string, how_many int) {
user, err := profile.GetUserByHandle(scraper.UserHandle(handle)) user, err := profile.GetUserByHandle(scraper.UserHandle(handle))
@ -384,7 +390,7 @@ func get_followers(handle string, how_many int) {
profile.SaveTweetTrove(trove, true) profile.SaveTweetTrove(trove, true)
profile.SaveAsFollowersList(user.ID, trove) profile.SaveAsFollowersList(user.ID, trove)
happy_exit(fmt.Sprintf("Saved %d followers", len(trove.Users))) happy_exit(fmt.Sprintf("Saved %d followers", len(trove.Users)), errors.Is(err, scraper.ErrRateLimited))
} }
func get_bookmarks(how_many int) { func get_bookmarks(how_many int) {
trove, err := scraper.GetBookmarks(how_many) trove, err := scraper.GetBookmarks(how_many)
@ -396,6 +402,7 @@ func get_bookmarks(how_many int) {
happy_exit(fmt.Sprintf( happy_exit(fmt.Sprintf(
"Saved %d tweets, %d retweets, %d users, and %d bookmarks", "Saved %d tweets, %d retweets, %d users, and %d bookmarks",
len(trove.Tweets), len(trove.Retweets), len(trove.Users), len(trove.Bookmarks)), len(trove.Tweets), len(trove.Retweets), len(trove.Users), len(trove.Bookmarks)),
errors.Is(err, scraper.ErrRateLimited),
) )
} }
func fetch_timeline(is_following_only bool) { func fetch_timeline(is_following_only bool) {
@ -405,7 +412,10 @@ func fetch_timeline(is_following_only bool) {
} }
profile.SaveTweetTrove(trove, true) profile.SaveTweetTrove(trove, true)
happy_exit(fmt.Sprintf("Saved %d tweets, %d retweets and %d users", len(trove.Tweets), len(trove.Retweets), len(trove.Users))) happy_exit(
fmt.Sprintf("Saved %d tweets, %d retweets and %d users", len(trove.Tweets), len(trove.Retweets), len(trove.Users)),
errors.Is(err, scraper.ErrRateLimited),
)
} }
func download_tweet_content(tweet_identifier string) { func download_tweet_content(tweet_identifier string) {
@ -442,7 +452,7 @@ func search(query string, how_many int) {
} }
profile.SaveTweetTrove(trove, true) profile.SaveTweetTrove(trove, true)
happy_exit(fmt.Sprintf("Saved %d tweets and %d users", len(trove.Tweets), len(trove.Users))) happy_exit(fmt.Sprintf("Saved %d tweets and %d users", len(trove.Tweets), len(trove.Users)), errors.Is(err, scraper.ErrRateLimited))
} }
func follow_user(handle string, is_followed bool) { func follow_user(handle string, is_followed bool) {
@ -453,9 +463,9 @@ func follow_user(handle string, is_followed bool) {
profile.SetUserFollowed(&user, is_followed) profile.SetUserFollowed(&user, is_followed)
if is_followed { if is_followed {
happy_exit("Followed user: " + handle) happy_exit("Followed user: "+handle, false)
} else { } else {
happy_exit("Unfollowed user: " + handle) happy_exit("Unfollowed user: "+handle, false)
} }
} }
@ -468,7 +478,7 @@ func unlike_tweet(tweet_identifier string) {
if err != nil { if err != nil {
die(err.Error(), false, -10) die(err.Error(), false, -10)
} }
happy_exit("Unliked the tweet.") happy_exit("Unliked the tweet.", false)
} }
func like_tweet(tweet_identifier string) { func like_tweet(tweet_identifier string) {
@ -484,7 +494,7 @@ func like_tweet(tweet_identifier string) {
if err != nil { if err != nil {
die(err.Error(), false, -1) die(err.Error(), false, -1)
} }
happy_exit("Liked the tweet.") happy_exit("Liked the tweet.", false)
} }
func list_followed() { func list_followed() {
@ -501,7 +511,7 @@ func start_webserver(addr string, should_auto_open bool) {
func fetch_inbox(how_many int) { func fetch_inbox(how_many int) {
trove, _ := scraper.GetInbox(how_many) trove, _ := scraper.GetInbox(how_many)
profile.SaveDMTrove(trove, true) profile.SaveDMTrove(trove, true)
happy_exit(fmt.Sprintf("Saved %d messages from %d chats", len(trove.Messages), len(trove.Rooms))) happy_exit(fmt.Sprintf("Saved %d messages from %d chats", len(trove.Messages), len(trove.Rooms)), false)
} }
func fetch_dm(id string, how_many int) { func fetch_dm(id string, how_many int) {
@ -512,7 +522,10 @@ func fetch_dm(id string, how_many int) {
max_id := scraper.DMMessageID(^uint(0) >> 1) max_id := scraper.DMMessageID(^uint(0) >> 1)
trove := scraper.GetConversation(room.ID, max_id, how_many) trove := scraper.GetConversation(room.ID, max_id, how_many)
profile.SaveDMTrove(trove, true) profile.SaveDMTrove(trove, true)
happy_exit(fmt.Sprintf("Saved %d messages from %d chats", len(trove.Messages), len(trove.Rooms))) happy_exit(
fmt.Sprintf("Saved %d messages from %d chats", len(trove.Messages), len(trove.Rooms)),
errors.Is(err, scraper.ErrRateLimited),
)
} }
func send_dm(room_id string, text string, in_reply_to_id int) { func send_dm(room_id string, text string, in_reply_to_id int) {
@ -523,5 +536,5 @@ func send_dm(room_id string, text string, in_reply_to_id int) {
trove := scraper.SendDMMessage(room.ID, text, scraper.DMMessageID(in_reply_to_id)) trove := scraper.SendDMMessage(room.ID, text, scraper.DMMessageID(in_reply_to_id))
profile.SaveDMTrove(trove, true) profile.SaveDMTrove(trove, true)
happy_exit(fmt.Sprintf("Saved %d messages from %d chats", len(trove.Messages), len(trove.Rooms))) happy_exit(fmt.Sprintf("Saved %d messages from %d chats", len(trove.Messages), len(trove.Rooms)), false)
} }