| 1 | #!/usr/bin/pythonw |
|---|
| 2 | # -*- coding: utf-8 -*- |
|---|
| 3 | |
|---|
| 4 | from appscript import app, k |
|---|
| 5 | from AppKit import NSApplication, NSBeep, NSSystemDefined, NSURL, NSWorkspace |
|---|
| 6 | from PyObjCTools import AppHelper |
|---|
| 7 | from Carbon.CarbonEvt import RegisterEventHotKey, GetApplicationEventTarget |
|---|
| 8 | from Carbon.Events import cmdKey |
|---|
| 9 | import struct |
|---|
| 10 | import scrape |
|---|
| 11 | import _StreamVision |
|---|
| 12 | |
|---|
| 13 | GROWL_APP_NAME = 'StreamVision' |
|---|
| 14 | NOTIFICATION_TRACK_INFO = 'iTunes Track Info' |
|---|
| 15 | NOTIFICATIONS_ALL = [NOTIFICATION_TRACK_INFO] |
|---|
| 16 | |
|---|
| 17 | kEventHotKeyPressedSubtype = 6 |
|---|
| 18 | kEventHotKeyReleasedSubtype = 9 |
|---|
| 19 | |
|---|
| 20 | growl = app('GrowlHelperApp') |
|---|
| 21 | |
|---|
| 22 | growl.register( |
|---|
| 23 | as_application=GROWL_APP_NAME, |
|---|
| 24 | all_notifications=NOTIFICATIONS_ALL, |
|---|
| 25 | default_notifications=NOTIFICATIONS_ALL, |
|---|
| 26 | icon_of_application='iTunes.app') |
|---|
| 27 | # if we leave off the .app, we can get Classic iTunes's icon |
|---|
| 28 | |
|---|
| 29 | def growlNotify(title, description, **kw): |
|---|
| 30 | growl.notify( |
|---|
| 31 | with_name=NOTIFICATION_TRACK_INFO, |
|---|
| 32 | title=title, |
|---|
| 33 | description=description, |
|---|
| 34 | application_name=GROWL_APP_NAME, |
|---|
| 35 | **kw) |
|---|
| 36 | |
|---|
| 37 | def radioParadiseURL(): |
|---|
| 38 | session = scrape.Session() |
|---|
| 39 | session.go('http://www2.radioparadise.com/nowplay_3.php') |
|---|
| 40 | return session.region.firsttag('a', class_='sky')['href'] |
|---|
| 41 | |
|---|
| 42 | def cleanStreamName(name): |
|---|
| 43 | name = name.split('. ')[0] |
|---|
| 44 | name = name.split(': ')[0] |
|---|
| 45 | name = name.split(' - ') |
|---|
| 46 | if len(name) > 1: |
|---|
| 47 | name = ' - '.join(name[:-1]) |
|---|
| 48 | else: |
|---|
| 49 | name = name[0] |
|---|
| 50 | return name |
|---|
| 51 | |
|---|
| 52 | def cleanStreamTitle(title): |
|---|
| 53 | if title == k.MissingValue: |
|---|
| 54 | return '' |
|---|
| 55 | title = title.replace('`', u'’') |
|---|
| 56 | return title |
|---|
| 57 | |
|---|
| 58 | def iTunesApp(): |
|---|
| 59 | return app(id='com.apple.iTunes') |
|---|
| 60 | |
|---|
| 61 | class StreamVision(NSApplication): |
|---|
| 62 | |
|---|
| 63 | hotKeyActions = {} |
|---|
| 64 | hotKeys = [] |
|---|
| 65 | |
|---|
| 66 | def displayTrackInfo(self): |
|---|
| 67 | iTunes = iTunesApp() |
|---|
| 68 | if iTunes.player_state.get() == k.playing: |
|---|
| 69 | if iTunes.current_track.class_.get() == k.URL_track: |
|---|
| 70 | growlNotify(cleanStreamTitle(iTunes.current_stream_title.get()), |
|---|
| 71 | cleanStreamName(iTunes.current_track.name.get())) |
|---|
| 72 | else: |
|---|
| 73 | kw = {} |
|---|
| 74 | artwork = iTunes.current_track.artworks.get() |
|---|
| 75 | if artwork: |
|---|
| 76 | kw['pictImage'] = artwork[0].data.get() |
|---|
| 77 | growlNotify(iTunes.current_track.name.get(), |
|---|
| 78 | iTunes.current_track.album.get() + "\n" + |
|---|
| 79 | iTunes.current_track.artist.get(), |
|---|
| 80 | **kw) |
|---|
| 81 | else: |
|---|
| 82 | trackName = '' |
|---|
| 83 | if iTunes.current_track.class_.get() != k.Property: |
|---|
| 84 | trackName = iTunes.current_track.name.get() |
|---|
| 85 | growlNotify("iTunes is not playing.", trackName) |
|---|
| 86 | |
|---|
| 87 | def goToSite(self): |
|---|
| 88 | iTunes = iTunesApp() |
|---|
| 89 | if iTunes.player_state.get() == k.playing: |
|---|
| 90 | url = iTunes.current_stream_URL.get() |
|---|
| 91 | if url: |
|---|
| 92 | if 'radioparadise.com' in url: |
|---|
| 93 | url = radioParadiseURL() |
|---|
| 94 | NSWorkspace.sharedWorkspace().openURL_(NSURL.URLWithString_(url)) |
|---|
| 95 | return |
|---|
| 96 | NSBeep() |
|---|
| 97 | |
|---|
| 98 | def registerHotKey(self, func, keyCode, mods=0): |
|---|
| 99 | hotKeyRef = RegisterEventHotKey(keyCode, mods, (0, 0), |
|---|
| 100 | GetApplicationEventTarget(), 0) |
|---|
| 101 | self.hotKeys.append(hotKeyRef) |
|---|
| 102 | self.hotKeyActions[_StreamVision.HotKeyAddress(hotKeyRef)] = func |
|---|
| 103 | |
|---|
| 104 | def finishLaunching(self): |
|---|
| 105 | super(StreamVision, self).finishLaunching() |
|---|
| 106 | self.registerHotKey(self.displayTrackInfo, 100) # F8 |
|---|
| 107 | self.registerHotKey(self.goToSite, 100, cmdKey) # cmd-F8 |
|---|
| 108 | self.registerHotKey(lambda: iTunesApp().playpause(), 101) # F9 |
|---|
| 109 | self.registerHotKey(lambda: iTunesApp().previous_track(), 109) # F10 |
|---|
| 110 | self.registerHotKey(lambda: iTunesApp().next_track(), 103) # F11 |
|---|
| 111 | |
|---|
| 112 | def sendEvent_(self, theEvent): |
|---|
| 113 | if theEvent.type() == NSSystemDefined and \ |
|---|
| 114 | theEvent.subtype() == kEventHotKeyPressedSubtype: |
|---|
| 115 | self.hotKeyActions[theEvent.data1()]() |
|---|
| 116 | super(StreamVision, self).sendEvent_(theEvent) |
|---|
| 117 | |
|---|
| 118 | if __name__ == "__main__": |
|---|
| 119 | AppHelper.runEventLoop() |
|---|
