#!/usr/bin/env python3

import json
import requests
import os.path

__author__ = 'Alexander Popov'
__version__ = '2.0.0'
__license__ = 'Unlicense'


def get_pages(username, api_key, limit=200):
    """ Getting the number of pages with scrobbling data """

    response = requests.get(
           'https://ws.audioscrobbler.com/2.0/'
           '?method=user.getrecenttracks&user='
           '{0}&api_key={1}&format=json&limit={2}'
           .format(username, api_key, limit))

    data = json.loads(response.content.decode("utf8"))

    return(int(data['recenttracks']['@attr']['totalPages']))


def get_page(username, api_key, page, limit=200):
    """ Getting scrobbling data from a page """

    response = requests.get(
               'https://ws.audioscrobbler.com/2.0/'
               '?method=user.getrecenttracks&user={0}&api_key={1}&format=json'
               '&limit={2}&page={3}'.format(username, api_key, limit, page))

    data = json.loads(response.content.decode("utf8"))

    return(data['recenttracks']['track'])


def get_now_scrobbling(username, api_key):
    """ Getting now scrobbling track """

    response = requests.get(
               'https://ws.audioscrobbler.com/2.0/'
               '?method=user.getrecenttracks&user={0}'
               '&api_key={1}&format=json&limit=1'
               .format(username, api_key))

    data = json.loads(response.content.decode('utf-8'))

    if '@attr' in data['recenttracks']['track'][0]:
        return(True)
    else:
        return(False)


if __name__ == '__main__':
    _ = dict()

    if os.path.exists('./config.json'):
        with open('./config.json') as f:
            _ = json.loads(f.read())
    else:
        _['api_key'] = input('API Key: ')
        _['username'] = input('Username: ')
    api_key, username = _['api_key'], _['username']

    total_pages = get_pages(username, api_key) - 603
    current_page = 1
    scrobbled = []

    while current_page <= total_pages:
        print('\r{0:.2f}% [{1} of {2}]'
              .format((current_page * 100 / total_pages),
                      current_page, total_pages), end='')

        response = get_page(username, api_key, current_page)

        for track in response:
            scrobbled.append(track)

        current_page += 1

    with open('%s.json' % (username), 'w+', encoding='utf-8') as f:
        data = json.dumps(scrobbled, indent=4,
                          sort_keys=True, ensure_ascii=False)
        f.write(data)

    print('\n{0} tracks saved in {1}.json!'.format(
        len(scrobbled), username))