#!/usr/bin/python

# ipod.py
# a little script for viewing and modifying some of the
# contents of an ipod.
#
# requires: python & libipod
#
# pls email me bugs and not bear porn
# by: ktf <esthetiq@gmail.com>
# last change: 7/25/2007

from ipodlib import *
import getopt, sys, os, os.path
import string

hdr = "\033[1;36mipod\033[0;0m"
verbose = 0
paths = IPod.Discover()
def usage():
	print "usage: %s [-h?vraqplim] [-f file]" % sys.argv[0]
	print "-h\tshow this help message screen"
	print "-?\tsame function as the previous flag"
	print "-v\tverbosely show informational messages"
	print "-r\tremove tracks from ipod (by trackid ONLY (see '-l'))"
	print "-a\tadd tracks to ipod"
	print "-q\tremove playlist from ipod"
	print "-p\tadd playlist to ipod"
	print "-l\tlist tracks on ipod"
	print "-i\tlist playlists on ipod"
	print "-m\tsignifies the argument passed to the '-f' flag"
	print "  \tis a filename containing a list of multiple strings"
	print "-f str\ta string being either a track id, name of a file,"
	print "      \tor the name of a track/playlist to be added/removed"
	print
def about():
	print "ipod.py by ktf <esthetiq@gmail.com>"
	print "last change: 7/25/2007"
	print
def ipod_add_tracks(lm, str):
	if len(paths):
		path = paths[0]
		ipod = IPod(path)
		args = str
		arry = []
		if lm:
			for item in str:
				temp = string.split(item, '\n')
				arry.append(temp[0])
			args = arry
		print "%s: opening ipod @ %s" % (hdr, path)
		print "%s: adding track(s) to ipod db" % hdr
		for arg in args:
			if os.path.exists(arg):
				(root, ext) = os.path.splitext(arg)
				if ext in ['.mp3', '.m4a', '.wav']:
					track = IPodTrack(ipod, arg)
					if track:
						print "%s: trackid %d: %s" % (hdr, track.GetAttribute(IPOD_TRACK_ID), track.GetText(IPOD_TITLE))
					else:
						print "%s: error adding track from file %s" % (hdr, arg)
				else:
					print "%s: skipping file" % hdr
		ipod.Flush()
	else:
		print "%s: no ipods connected... is it mounted?" % hdr
	print "%s: finished" % hdr
def ipod_add_pls(lm, file):
	if len(paths):
		path = paths[0]
		ipod = IPod(path)
		args = file
		arry = []
		if lm:
			for item in file:
				temp = string.split(item, '\n')
				arry.append(temp[0])
			args = arry
		print "%s: opening ipod @ %s" % (hdr, path)
		print "%s: adding playlist to ipod db" % hdr
		for arg in args:
			pls = IPodPlaylist(ipod)
			if pls:
				pls.SetText(IPOD_TITLE, arg)
				print "%s: playlistid %d: %s" % hdr, (pls.GetAttribute(IPOD_PLAYLIST_PLAYLIST_ID_LO), pls.GetText(IPOD_TITLE))
		ipod.Flush()
	else:
		print "%s: no ipods connected... is it mounted?" % hdr
def ipod_rm_pls(lm, name):
	if len(paths):
		path = paths[0]
		ipod = IPod(path)
		args = name
		arry = []
		if lm:
			for item in name:
				temp = string.split(item, '\n')
				arry.append(temp[0])
			args = arry
		print "%s: opening ipod @ %s" % (hdr, path)
		print "%s: removing playlist from ipod db" % hdr
		for arg in args:
			for i in xrange(ipod.PlaylistCount()):
				pls = ipod.PlaylistByIndex(i)
				if pls:
					if pls.GetText(IPOD_TITLE) == arg:
						pls.Remove()
						print "%s: removed playlist %s" % (hdr, arg)
						break
		ipod.Flush()
	else:
		print "%s: no ipods connected... is it mounted?" % hdr
def ipod_rm_tracks(lm, name):
	if len(paths):
		path = paths[0]
		ipod = IPod(path)
		args = name
		arry = []
		if lm:
			for item in name:
				temp = string.split(item, '\n')
				arry.append(temp[0])
			args = arry
		print "%s: opening ipod @ %s" % (hdr, path)
		print "%s: removing track from ipod db" % hdr
		for arg in args:
			for i in xrange(ipod.TrackCount()):
				track = ipod.TrackByIndex(i)
				if track:
					if track.GetAttribute(IPOD_TRACK_ID) == int(name):
						track.Remove()
						print "%s: removed trackid %d" % (hdr, int(name))
						break
		ipod.Flush()
	else:
		print "%s: no ipods connected... is it mounted?" % hdr
def ipod_ls_tracks():
	for path in paths:
		ipod = IPod(path)
		(total, free) = ipod.DiskUsage()
		print "%s: found ipod @ %s (%d tracks, %d playlists, %dKB total, %dKB free)" % (hdr, path, ipod.TrackCount(), ipod.PlaylistCount(), total, free)
		for tindex in xrange(ipod.TrackCount()):
			track = ipod.TrackByIndex(tindex)
			trackid = track.GetAttribute(IPOD_TRACK_ID)
			print "%s: index %d trackid %d: %s %s/%s" % (hdr, tindex, trackid, track.GetText(IPOD_TITLE), track.GetText(IPOD_ARTIST), track.GetText(IPOD_ALBUM))
		for pindex in xrange(ipod.PlaylistCount()):
			pls = ipod.PlaylistByIndex(pindex)
			print "%s: playlist %d: %s" % (hdr, pindex, pls.GetText(IPOD_TITLE))
			for titemindex in xrange(pls.TrackItemCount()):
				titem = pls.TrackItemByIndex(titemindex)
				trackid = titem.GetAttribute(IPOD_TRACK_ITEM_TRACK_ID)
				track = ipod.TrackByTrackID(trackid)
				print "%s: index %d trackid %d: %s" % (hdr, titemindex, trackid, track.GetText(IPOD_TITLE))
def main():
	about()
	action = 0
	listmode = 0
	file = ""
	dir = ""
	if len(sys.argv) < 2:
		usage()
	try:
		opts, args = getopt.getopt(sys.argv[1:], "?hvf:raqplim")
	except getopt.GetoptError:
		usage()
		sys.exit(1)
	for a, b in opts:
		if a in ("-?", "-h"):
			usage()
		if a == "-v":
			verbose = 1
		if a == "-f":
			file = b
		if a == "-r":
			action = 1
		if a == "-a":
			action = 2
		if a == "-q":
			action = 3
		if a == "-p":
			action = 4
		if a == "-l":
			action = 5
		if a == "-i":
			action = 6
		if a == "-m":
			listmode = 1
	
	if listmode:
		if verbose:
			print "%s: using listmode, assuming '-f' value is filename" % hdr
		f = open(file, 'r')
	if action is 1:
		if listmode:
			ipod_rm_tracks(1, f.readlines())
			f.close()
		else:
			ipod_rm_tracks(0, file)
	if action is 2:
		if listmode:
			ipod_add_tracks(1, f.readlines())
			f.close()
		else:
			ipod_add_tracks(0, file)
	if action is 3:
		if listmode:
			ipod_rm_pls(1, f.readlines())
			f.close()
		else:
			ipod_rm_pls(0, file)
	if action is 4:
		if listmode:
			ipod_add_pls(1, f.readlines())
			f.close()
		else:
			ipod_add_pls(0, file)
	if action is 5:
		ipod_ls_tracks()
	if action is 6:
		print "%s: option currently unavaliable" % hdr
		# ipod_ls_pls()
if __name__ == "__main__":
	main()