this repo has no description
favorites --stop_after option, refs #20
+5
-4
+3
-2
twitter_to_sqlite/cli.py
+3
-2
twitter_to_sqlite/cli.py
···
146
146
)
147
147
@click.option("--user_id", help="Numeric user ID")
148
148
@click.option("--screen_name", help="Screen name")
149
-
def favorites(db_path, auth, user_id, screen_name):
149
+
@click.option("--stop_after", type=int, help="Stop after this many")
150
+
def favorites(db_path, auth, user_id, screen_name, stop_after):
150
151
"Save tweets favorited by specified user"
151
152
auth = json.load(open(auth))
152
153
session = utils.session_for_auth(auth)
153
154
profile = utils.get_profile(session, user_id, screen_name)
154
155
db = sqlite_utils.Database(db_path)
155
156
with click.progressbar(
156
-
utils.fetch_favorites(session, user_id, screen_name),
157
+
utils.fetch_favorites(session, user_id, screen_name, stop_after),
157
158
label="Importing favorites",
158
159
show_pos=True,
159
160
) as bar:
+2
-2
twitter_to_sqlite/utils.py
+2
-2
twitter_to_sqlite/utils.py
···
92
92
)
93
93
94
94
95
-
def fetch_favorites(session, user_id, screen_name):
95
+
def fetch_favorites(session, user_id, screen_name, stop_after=None):
96
96
args = user_args(user_id, screen_name)
97
97
# Rate limit 75/15 mins = 5/minute = every 12 seconds
98
98
sleep = 12
99
99
yield from fetch_timeline(
100
-
session, "https://api.twitter.com/1.1/favorites/list.json", args, sleep=sleep
100
+
session, "https://api.twitter.com/1.1/favorites/list.json", args, sleep=sleep, stop_after=stop_after
101
101
)
102
102
103
103