def test_drop_old_users(self):
# First create a bunch of users
r = get_router_table()
router = Router(r, SinkMetrics())
# Purge any existing users from previous runs.
router.drop_old_users(0)
for _ in range(0, 53):
router.register_user(self._create_minimal_record())
results = router.drop_old_users(months_ago=0)
assert list(results) == [25, 25, 3]
def test_drop_old_users(self):
# First create a bunch of users
r = get_router_table()
router = Router(r, SinkMetrics())
for _ in range(0, 53):
router.register_user(self._create_minimal_record())
results = router.drop_old_users(months_ago=0)
eq_(list(results), [25, 25, 3])
def drop_users(router_table_name, months_ago, batch_size, pause_time):
router_table = get_router_table(router_table_name)
router = Router(router_table, SinkMetrics())
click.echo("Deleting users with a last_connect %s months ago." %
months_ago)
count = 0
for deletes in router.drop_old_users(months_ago):
click.echo("")
count += deletes
if count >= batch_size:
click.echo("Deleted %s user records, pausing for %s seconds." %
pause_time)
time.sleep(pause_time)
count = 0
click.echo("Finished old user purge.")
