code.delx.au - notipod/blob - NotiPod.py

   1 #!/usr/bin/env python
   2 # Copyright 2009 James Bunton <jamesbunton@fastmail.fm>
   3 # Licensed for distribution under the GPL version 2, check COPYING for details
   4
   5 import logging
   6
   7 import objc
   8 from Foundation import *
   9 from AppKit import *
  10 from PyObjCTools import AppHelper
  11
  12 import libsyncitunes
  13
  14
  15 class PlaylistModel(NSObject):
  16         outlineView = objc.IBOutlet()
  17
  18         def awakeFromNib(self):
  19                 self.root = []
  20                 self.playlists = {}
  21                 self.outlineView.setDataSource_(self)
  22
  23         def setPlaylists(self, playlists):
  24                 self.root = []
  25                 self.playlists = playlists
  26                 for playlist in self.playlists:
  27                         if playlist.parent is None:
  28                                 self.root.append(playlist)
  29                 self.outlineView.reloadData()
  30                 self.outlineView.expandItem_expandChildren_(None, True)
  31
  32         def outlineView_child_ofItem_(self, _, childIndex, playlist):
  33                 if playlist == None:
  34                         return self.root[childIndex]
  35                 else:
  36                         return playlist.children[childIndex]
  37
  38         def outlineView_isItemExpandable_(self, _, playlist):
  39                 if playlist == None:
  40                         return True
  41                 else:
  42                         return len(playlist.children) > 0
  43
  44         def outlineView_numberOfChildrenOfItem_(self, _, playlist):
  45                 if playlist == None:
  46                         return len(self.root)
  47                 else:
  48                         return len(playlist.children)
  49
  50         def outlineView_objectValueForTableColumn_byItem_(self, _, col, playlist):
  51                 col = col.identifier() if col else "playlist"
  52
  53                 if col == "selected":
  54                         selected = NSApp.delegate().playlists()
  55                         return playlist.pid in selected
  56                 if col == None or col == "playlist":
  57                         return playlist.name
  58
  59         def outlineView_setObjectValue_forTableColumn_byItem_(self, _, v, col, playlist):
  60                 col = col.identifier() if col else "playlist"
  61
  62                 if col != "selected":
  63                         return
  64                 NSApp.delegate().setPlaylist_selected_(playlist.pid, v)
  65
  66
  67 class FolderModel(NSObject):
  68         window = objc.IBOutlet()
  69         folderPopup = objc.IBOutlet()
  70
  71         def awakeFromNib(self):
  72                 self.folderPopup.addItemsWithTitles_(NSApp.delegate().folders())
  73                 self.folderPopup.selectItemAtIndex_(2)
  74                 self.lastIndex = 2
  75
  76         @objc.IBAction
  77         def doSelectFolder_(self, sender):
  78                 currentIndex = self.folderPopup.indexOfSelectedItem()
  79                 if currentIndex >= 2:
  80                         self.lastIndex = currentIndex
  81                         NSApp.delegate().addFolder_(self.folderPopup.titleOfSelectedItem())
  82                         return
  83                 panel = NSOpenPanel.openPanel()
  84                 panel.setCanChooseFiles_(False)
  85                 panel.setCanChooseDirectories_(True)
  86                 panel.setAllowsMultipleSelection_(False)
  87                 panel.beginSheetForDirectory_file_types_modalForWindow_modalDelegate_didEndSelector_contextInfo_(
  88                         None, None, [], self.window, self, self.selectFolderEnd_returnCode_contextInfo_, None)
  89
  90         @objc.signature("v@:@ii")
  91         def selectFolderEnd_returnCode_contextInfo_(self, panel, ret, _):
  92                 if ret == NSOKButton:
  93                         assert len(panel.filenames()) == 1
  94                         folder = panel.filenames()[0]
  95                         NSApp.delegate().addFolder_(folder)
  96                         self.folderPopup.insertItemWithTitle_atIndex_(folder, 2)
  97                         self.folderPopup.selectItemAtIndex_(2)
  98                 else:
  99                         self.folderPopup.selectItemAtIndex_(self.lastIndex)
 100
 101
 102 class NotiPodAppDelegate(NSObject):
 103         window = objc.IBOutlet()
 104         playlistModel = objc.IBOutlet()
 105         folderModel = objc.IBOutlet()
 106         loadingSheet = objc.IBOutlet()
 107         loadingLabel = objc.IBOutlet()
 108
 109         def awakeFromNib(self):
 110                 self.gen = None
 111
 112         # Delegate methods
 113         def applicationWillFinishLaunching_(self, _):
 114                 pass
 115
 116         def applicationDidFinishLaunching_(self, _):
 117                 self.library = libsyncitunes.ITunesLibrary.alloc().init()
 118                 def finish():
 119                         self.playlistModel.setPlaylists(self.library.get_playlists())
 120                 self.runGenerator(lambda: self.library.load_(None), finish)
 121
 122         def applicationWillTerminate_(self, _):
 123                 self.prefs().synchronize()
 124
 125         def applicationShouldTerminateAfterLastWindowClosed_(self, _):
 126                 return True
 127
 128
 129         # Utility methods
 130         def runGenerator(self, func, finish):
 131                 NSApp.beginSheet_modalForWindow_modalDelegate_didEndSelector_contextInfo_(self.loadingSheet, self.window, None, None, None)
 132                 self.gen = func()
 133                 self.finish = finish
 134                 self.runGeneratorNext()
 135
 136         def runGeneratorNext(self):
 137                 if self.gen is None or self.finish is None:
 138                         return
 139                 try:
 140                         msg = self.gen.next()
 141                         self.loadingLabel.setStringValue_(msg)
 142                         self.performSelector_withObject_afterDelay_(
 143                                 self.runGeneratorNext, None, 0)
 144                 except StopIteration:
 145                         self.stopGenerator()
 146
 147         def stopGenerator(self):
 148                 self.gen = None
 149                 NSApp.endSheet_(self.loadingSheet)
 150                 self.loadingSheet.orderOut_(self)
 151                 self.finish()
 152                 self.finish = None
 153
 154         @objc.IBAction
 155         def doCancel_(self, sender):
 156                 self.stopGenerator()
 157
 158         @objc.IBAction
 159         def doSync_(self, sender):
 160                 folder = self.folders()[0]
 161                 playlists = [self.library.get_playlist_pid(pid) for pid in self.playlists()]
 162
 163                 all_tracks = []
 164                 for playlist in playlists:
 165                         all_tracks.extend(playlist.tracks)
 166                         libsyncitunes.export_m3u(dry_run=False, dest=folder, path_prefix="",
 167                                         playlist_name=playlist.name, files=playlist.tracks)
 168
 169                 def finish():
 170                         NSRunAlertPanel("Complete!", "Synchronisation is complete", "Ok", None, None)
 171                 self.runGenerator(
 172                         lambda:
 173                                 libsyncitunes.sync(
 174                                         dry_run=False,
 175                                         source=self.library.folder,
 176                                         dest=folder,
 177                                         files_to_copy=all_tracks
 178                                 )
 179                         ,
 180                         finish
 181                 )
 182
 183
 184         # Public accessors
 185
 186         def prefs(self):
 187                 return NSUserDefaults.standardUserDefaults()
 188
 189         def _getArray(self, key):
 190                 res = self.prefs().stringArrayForKey_(key)
 191                 return list(res) if res else []
 192
 193         def _saveArray(self, key, array):
 194                 self.prefs().setObject_forKey_(array, key)
 195
 196         def playlists(self):
 197                 return self._getArray("playlists")
 198
 199         def folders(self):
 200                 return self._getArray("folders")
 201
 202         def addFolder_(self, folder):
 203                 folders = self.folders()
 204                 while folder in folders:
 205                         folders.remove(folder)
 206                 folders.insert(0, folder)
 207                 folders = folders[:10]
 208                 self._saveArray("folders", folders)
 209
 210         def setPlaylist_selected_(self, playlist, selected):
 211                 playlists = self.playlists()
 212                 if selected:
 213                         playlists.append(playlist)
 214                 else:
 215                         playlists.remove(playlist)
 216                 playlists = list(set(playlists))
 217                 self._saveArray("playlists", list(set(playlists)))
 218
 219
 220 def main():
 221 ###     logging.basicConfig(format="%(levelname)s: %(message)s")
 222 ###     logging.getLogger().setLevel(logging.DEBUG)
 223         AppHelper.runEventLoop()
 224
 225 if __name__ == "__main__":
 226         main()
 227