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.")