#!/usr/bin/python3
# ex: set filetype=python

import argparse
import os
from shutil importy copyfile
import xml.etree.cElementTree as xml

class Track():
    """
    Stores all the information of a track inside an object.
    Can be used to easily access the tracks stored in the database.
    """

    def __init__(self, xml_doc):
        it = iter(xml_doc)
        for k, v in zip(it, it):
            setattr(self, k.text.replace(' ', ''), v.text)

def parseXml(xml_file):
    """
    Parses an xml file and returns the DOM tree.
    """
    tree = xml.parse(xml_file)
    root = tree.getroot()
    tracks = []
    for t in root.findall('.//dict/dict/dict'):
        tracks.append(Track(t))
    return tracks

def copy_file(track, targetFolder):
    """
    Copies a track from its source to the destination folder
    """
    path = track.Location[7:]
    dst = os.path.join(targetFolder, os.path.basename(path))
    if (os.path.exists(path)):
        print('Copying file {0}', path)
        copyfile(path, dst)
    else:
        print('File {0} not found.')
    return dst


def main():
    """
    Setup our main program
    """
    parser = argparse.ArgumentParser()
    parser.add_argument('-t', '--target', type=str, help='The target path to store the scraped \
                                                          files into', default='~/itunesscrape')
    parser.add_argument('-l', '--logfile', type=str, help='The path to the logfile to use.',
                        default='~/itunesscrape.log')
    parser.add_argument('libraryFile', help='The iTunes library file in XML format.')
    args = parser.parse_args()

    print('Parsing library file...')
    tracks = parseXml(args.libraryFile)
    print('{0} tracks found.'.format(len(tracks)))

    if not os.path.isdir(args.target):
        os.mkdir(args.target)

    for t in tracks:
        new_file = copy_file(t, args.target)

if __name__ == "__main__":
    main()