47 lines
1.6 KiB
Python
47 lines
1.6 KiB
Python
import sys
|
|
|
|
|
|
class Progress:
|
|
def __init__(self, current, total) -> None:
|
|
self.current = current
|
|
self.total = total
|
|
pass
|
|
|
|
def print_progress(self, current, waiting, retry_cnt, no_tweets_limit) -> None:
|
|
self.current = current
|
|
progress = current / self.total
|
|
bar_length = 40
|
|
progress_bar = (
|
|
"["
|
|
+ "=" * int(bar_length * progress)
|
|
+ "-" * (bar_length - int(bar_length * progress))
|
|
+ "]"
|
|
)
|
|
if no_tweets_limit:
|
|
if waiting:
|
|
sys.stdout.write(
|
|
"\rTweets scraped : {} - Rate Limited: waiting to access older tweets. Attempt to proceed {} over 15 min".format(
|
|
current, retry_cnt
|
|
)
|
|
)
|
|
else:
|
|
sys.stdout.write(
|
|
"\rTweets scraped : {} ".format(
|
|
current
|
|
)
|
|
)
|
|
else:
|
|
if waiting:
|
|
sys.stdout.write(
|
|
"\rProgress: [{:<40}] {:.2%} {} of {} - waiting to access older tweets {} min on 15 min".format(
|
|
progress_bar, progress, current, self.total, retry_cnt
|
|
)
|
|
)
|
|
else:
|
|
sys.stdout.write(
|
|
"\rProgress: [{:<40}] {:.2%} {} of {} ".format(
|
|
progress_bar, progress, current, self.total
|
|
)
|
|
)
|
|
sys.stdout.flush()
|