X-Git-Url: https://code.delx.au/notipod/blobdiff_plain/a9de8db967ce3c8d7063301bf3a806a1b1597503..fd23beafb68339a9c4c5dd35e6c2d14fce0bb2ce:/libnotipod.py

diff --git a/libnotipod.py b/libnotipod.py
index 76b7fb0..2a996ec 100644
--- a/libnotipod.py
+++ b/libnotipod.py
@@ -2,6 +2,7 @@
 # Copyright 2009 James Bunton <jamesbunton@fastmail.fm>
 # Licensed for distribution under the GPL version 2, check COPYING for details
 
+import collections
 import logging
 import os
 import shutil
@@ -50,6 +51,8 @@ class ITunesLibrary(NSObject):
 		pl_tracks = plist["Tracks"]
 		pl_lookup = {}
 		self.playlists = []
+		self.track2playlist = collections.defaultdict(set)
+		self.track2filename = {}
 		for pl_playlist in plist["Playlists"]:
 			playlist = self.make_playlist(pl_playlist, pl_tracks, pl_lookup)
 			if not playlist:
@@ -95,17 +98,25 @@ class ITunesLibrary(NSObject):
 			parent = pl_lookup[parent_pid]
 		except KeyError:
 			pass
+
 		tracks = []
 		for item in pl_playlist.get("Playlist Items", []):
 			trackID = item["Track ID"]
-			filename = str(pl_tracks[str(trackID)]["Location"])
-			filename = self.loc2name(filename)
-			filename = filename.decode("utf-8")
-			if not filename.startswith(self.folder):
-				logging.warn("Skipping: " + filename)
-				continue
-			filename = strip_prefix(filename, self.folder)
-			tracks.append(filename)
+			item = pl_tracks[str(trackID)]
+			self.track2playlist[trackID].add(pid)
+			tracks.append(trackID)
+			if trackID not in self.track2filename:
+				if item["Track Type"] != "File":
+					continue
+				filename = str(item["Location"])
+				filename = self.loc2name(filename)
+				filename = filename.decode("utf-8")
+				if not filename.startswith(self.folder):
+					logging.warn("Skipping: " + filename)
+					continue
+				filename = strip_prefix(filename, self.folder)
+				self.track2filename[trackID] = filename
+
 		playlist = Playlist.alloc().init()
 		playlist.set(name, pid, ptype, tracks, parent)
 		return playlist
@@ -126,6 +137,12 @@ class ITunesLibrary(NSObject):
 			if playlist.pid == pid:
 				return playlist
 
+	def get_track_filename(self, trackID):
+		return self.track2filename.get(trackID, None)
+
+	def get_track_playlists(self, trackID):
+		return self.track2playlist.get(trackID, [])
+
 	def get_playlists(self):
 		return self.playlists
 
@@ -170,9 +187,14 @@ def mkdirhier(path):
 def export_m3u(dry_run, dest, path_prefix, playlist_name, files):
 	if dry_run:
 		return
+	dest = os.path.join(dest, "-Playlists-")
 	if not path_prefix:
-		path_prefix = "../"
-	playlist_file = os.path.join(dest, "-Playlists-", playlist_name) + ".m3u"
+		try:
+			f = open(os.path.join(dest, ".path_prefix"))
+			path_prefix = f.read().strip()
+		except:
+			path_prefix = "../"
+	playlist_file = os.path.join(dest, playlist_name) + ".m3u"
 	playlist_file = encode_filename(playlist_file)
 	mkdirhier(os.path.dirname(playlist_file))
 	logging.info("Writing: " + playlist_file)