#!/usr/bin/env python3

import json
import csv
import os.path
import requests

__author__ = 'Alexander Popov'
__version__ = '2.0.1'
__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)


def scrobbling_export(tracks, username, export_format='is as'):
    """ Save scrobbled track via various format """

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

    elif export_format == 'simple':
        _ = []

        for track in tracks:
            _.append([
                track['artist']['#text'], track['name'], track['date']['uts']
                ])

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

    elif export_format == 'csv':
        _ = []

        for track in tracks:
            _.append([
                track['artist']['#text'],
                track['name'],
                int(track['date']['uts'])
                ])

        with open('%s.csv' % (username), 'w', encoding='utf-8', newline='') as f:
            data = csv.writer(f, quoting=csv.QUOTE_NONNUMERIC, delimiter=',')
            for row in _:
                data.writerow(row)

    return(1)


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

    # if get_now_scrobbling(username, api_key):
    #     scrobbled.pop(0)

    if scrobbling_export(scrobbled, username, _['export_format']):
        print('\n{0} tracks saved!'.format(
            len(scrobbled), username))