69 lines
1.9 KiB
Python
Executable File
69 lines
1.9 KiB
Python
Executable File
#!/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()
|