#!/usr/bin/python

import codecs
import re

from mutagen import flac

FILENAME='tracks'

CONFIRM=True
DEBUG=False
SWAPNAME=True
SAVE=True
SPLIT=False

if CONFIRM or DEBUG:
    import pprint
    # From
    # http://stackoverflow.com/questions/10883399/unable-to-encode-decode-pprint-output
    class PPUTF8(pprint.PrettyPrinter):
        def format(self, object, context, maxlevels, level):
            if isinstance(object, unicode):
                return (object.encode('utf8'), True, False)
            return pprint.PrettyPrinter.format(self, object, context,
                        maxlevels, level)


if __name__ == '__main__':
    if CONFIRM or DEBUG:
        pp = PPUTF8(indent=4)

    with codecs.open(FILENAME, encoding='utf-8', mode='r') as f:
        lines = f.readlines()
    headers = lines[0][:-1].split('\t')

    if SWAPNAME:
        first_last = re.compile('^(?P<last>[^,]+),\s+(?P<first>.+)$')

    files = {}
    for i, file in enumerate(lines[1:]):
        # i starts counting at zero while we need it to start counting at 1
        # since 0 is the header line
        i += 1
        line = lines[i].replace('\n', '').split('\t')
        file = line[0]
        files[file] = {}
        for j, tag in enumerate(line[1:]):
            # j starts counting at 0, should be 1
            j += 1
            header = headers[j]
            files[file][header] = tag.strip()
            if header == 'ARTIST' or header == 'LYRICIST' or \
                    header == 'ARRANGER':
                if SPLIT:
                    files[file][header] = [name.strip() for name in
                                                        tag.split(';')]
                else:
                    files[file][header] = tag.strip()
                if SWAPNAME:
                    for n, name in enumerate(files[file][header]):
                        res = first_last.match(name)
                        if res:
                            files[file][header][n] = res.group('first') + \
                                    ' ' + res.group('last')


    
    if CONFIRM:
        print "Loaded data:\n"
        pp.pprint(files)

    if CONFIRM:
        if raw_input("\nSave files? [Y/n]").lower()in ['', 'y']:
            SAVE = True
        else:
            SAVE = False

    file = flac.FLAC()
    for name in files.keys():
        file.load(name)
        # Delete tags
        file.delete()
        for key, val in files[name].iteritems():
            file[key] = val
            if SAVE:
                file.save()

        if DEBUG:
            print "\n\nData in flac file:", name
            pp.pprint(file)