test_client.py

import sys
import time
import os

from utils import (
    hms_string,
)

from nbs_client import (
    get_artists,
)

DEFAULT_MAX_RELEASES = 5
DEFAULT_MAX_APPEARANCES = 5

user_token = os.environ.get('NEXT_BIG_SOUND_TOKEN')

def main(argv):

    if not user_token:
        print 'Missing NEXT_BIG_SOUND_TOKEN environment variable'
        sys.exit(2)

    max_releases = DEFAULT_MAX_RELEASES
    max_appearances = DEFAULT_MAX_APPEARANCES

    try:
        max_releases = int(argv[0])
        max_appearances = int(argv[1])
    except Exception:
        pass

    print('\nGetting {} releases with {} appearances\n'.format(max_releases, max_appearances))

    start_time = time.time()

    count_artist = 0

    for artist in get_artists(
        max_chart_releases=max_releases,
        max_chart_appearances=max_appearances,
    ):
        count_artist += 1
        print('Artist {0}: {1} \n'.format(count_artist, artist))

    elapsed_time = time.time() - start_time

    print('Elapsed time: {}'.format(hms_string(elapsed_time)))
    print('Artists Parsed: {}'.format(count_artist))

if __name__ == '__main__':
   main(sys.argv[1:])