1 | #!/usr/bin/pythonw |
---|
2 | # -*- coding: utf-8 -*- |
---|
3 | |
---|
4 | from appscript import app, k, its, CommandError |
---|
5 | from AppKit import NSApplication, NSBeep, NSSystemDefined, NSURL, NSWorkspace |
---|
6 | from Foundation import NSDistributedNotificationCenter |
---|
7 | from PyObjCTools import AppHelper |
---|
8 | from Carbon.CarbonEvt import RegisterEventHotKey, GetApplicationEventTarget |
---|
9 | from Carbon.Events import cmdKey, shiftKey, controlKey |
---|
10 | import struct |
---|
11 | import scrape |
---|
12 | import HotKey |
---|
13 | |
---|
14 | GROWL_APP_NAME = 'StreamVision' |
---|
15 | NOTIFICATION_TRACK_INFO = 'iTunes Track Info' |
---|
16 | NOTIFICATIONS_ALL = [NOTIFICATION_TRACK_INFO] |
---|
17 | |
---|
18 | kEventHotKeyPressedSubtype = 6 |
---|
19 | kEventHotKeyReleasedSubtype = 9 |
---|
20 | |
---|
21 | growl = app('GrowlHelperApp') |
---|
22 | |
---|
23 | growl.register( |
---|
24 | as_application=GROWL_APP_NAME, |
---|
25 | all_notifications=NOTIFICATIONS_ALL, |
---|
26 | default_notifications=NOTIFICATIONS_ALL, |
---|
27 | icon_of_application='iTunes.app') |
---|
28 | # if we leave off the .app, we can get Classic iTunes's icon |
---|
29 | |
---|
30 | def growlNotify(title, description, **kw): |
---|
31 | growl.notify( |
---|
32 | with_name=NOTIFICATION_TRACK_INFO, |
---|
33 | title=title, |
---|
34 | description=description, |
---|
35 | application_name=GROWL_APP_NAME, |
---|
36 | **kw) |
---|
37 | |
---|
38 | def radioParadiseURL(): |
---|
39 | session = scrape.Session() |
---|
40 | session.go('http://www2.radioparadise.com/nowplay_b.php') |
---|
41 | return session.region.firsttag('a')['href'] |
---|
42 | |
---|
43 | def cleanStreamTitle(title): |
---|
44 | if title == k.MissingValue: |
---|
45 | return '' |
---|
46 | title = title.split(' [')[0] # XXX move to description |
---|
47 | title = title.replace('`', u'’') |
---|
48 | return title |
---|
49 | |
---|
50 | def cleanStreamTrackName(name): |
---|
51 | name = name.split('. ')[0] |
---|
52 | name = name.split(': ')[0] |
---|
53 | name = name.split(' - ') |
---|
54 | if len(name) > 1: |
---|
55 | name = ' - '.join(name[:-1]) |
---|
56 | else: |
---|
57 | name = name[0] |
---|
58 | return name |
---|
59 | |
---|
60 | def iTunesApp(): return app(id='com.apple.iTunes') |
---|
61 | def XTensionApp(): return app(creator='SHEx') |
---|
62 | |
---|
63 | HAVE_XTENSION = False |
---|
64 | try: |
---|
65 | XTensionApp() |
---|
66 | HAVE_XTENSION = True |
---|
67 | except: |
---|
68 | pass |
---|
69 | |
---|
70 | class StreamVision(NSApplication): |
---|
71 | |
---|
72 | hotKeyActions = {} |
---|
73 | hotKeys = [] |
---|
74 | |
---|
75 | def displayTrackInfo(self): |
---|
76 | iTunes = iTunesApp() |
---|
77 | |
---|
78 | trackClass = iTunes.current_track.class_.get() |
---|
79 | trackName = '' |
---|
80 | if trackClass != k.Property: |
---|
81 | trackName = iTunes.current_track.name.get() |
---|
82 | |
---|
83 | if iTunes.player_state.get() != k.playing: |
---|
84 | growlNotify('iTunes is not playing.', trackName) |
---|
85 | return |
---|
86 | if trackClass == k.URL_track: |
---|
87 | growlNotify(cleanStreamTitle(iTunes.current_stream_title.get()), |
---|
88 | cleanStreamTrackName(trackName)) |
---|
89 | return |
---|
90 | if trackClass == k.Property: |
---|
91 | growlNotify('iTunes is playing.', '') |
---|
92 | return |
---|
93 | kw = {} |
---|
94 | # XXX iTunes doesn't let you get artwork for shared tracks |
---|
95 | if trackClass != k.shared_track: |
---|
96 | artwork = iTunes.current_track.artworks.get() |
---|
97 | if artwork: |
---|
98 | kw['pictImage'] = artwork[0].data.get() |
---|
99 | growlNotify(trackName + ' ' + |
---|
100 | '★' * (iTunes.current_track.rating.get() / 20), |
---|
101 | iTunes.current_track.album.get() + "\n" + |
---|
102 | iTunes.current_track.artist.get(), |
---|
103 | **kw) |
---|
104 | |
---|
105 | def goToSite(self): |
---|
106 | iTunes = iTunesApp() |
---|
107 | if iTunes.player_state.get() == k.playing: |
---|
108 | url = iTunes.current_stream_URL.get() |
---|
109 | if url: |
---|
110 | if 'radioparadise.com' in url and 'review' not in url: |
---|
111 | url = radioParadiseURL() |
---|
112 | NSWorkspace.sharedWorkspace().openURL_(NSURL.URLWithString_(url)) |
---|
113 | return |
---|
114 | NSBeep() |
---|
115 | |
---|
116 | def registerHotKey(self, func, keyCode, mods=0): |
---|
117 | hotKeyRef = RegisterEventHotKey(keyCode, mods, (0, 0), |
---|
118 | GetApplicationEventTarget(), 0) |
---|
119 | self.hotKeys.append(hotKeyRef) |
---|
120 | self.hotKeyActions[HotKey.HotKeyAddress(hotKeyRef)] = func |
---|
121 | return hotKeyRef |
---|
122 | |
---|
123 | def unregisterHotKey(self, hotKeyRef): |
---|
124 | self.hotKeys.remove(hotKeyRef) |
---|
125 | del self.hotKeyActions[HotKey.HotKeyAddress(hotKeyRef)] |
---|
126 | hotKeyRef.UnregisterEventHotKey() |
---|
127 | |
---|
128 | def incrementRatingBy(self, increment): |
---|
129 | iTunes = iTunesApp() |
---|
130 | rating = iTunes.current_track.rating.get() |
---|
131 | rating += increment |
---|
132 | if rating < 0: |
---|
133 | rating = 0 |
---|
134 | NSBeep() |
---|
135 | elif rating > 100: |
---|
136 | rating = 100 |
---|
137 | NSBeep() |
---|
138 | iTunes.current_track.rating.set(rating) |
---|
139 | |
---|
140 | def playPause(self): |
---|
141 | iTunes = iTunesApp() |
---|
142 | was_playing = (iTunes.player_state.get() == k.playing) |
---|
143 | iTunes.playpause() |
---|
144 | if not was_playing and iTunes.player_state.get() == k.stopped: |
---|
145 | # most likely, we're focused on the iPod, so playing does nothing |
---|
146 | iTunes.browser_windows[1].view.set(iTunes.user_playlists.filter(its.name=='Stations')[1].get()) |
---|
147 | iTunes.play() |
---|
148 | if HAVE_XTENSION: |
---|
149 | if iTunes.player_state.get() == k.playing: |
---|
150 | XTensionApp().turnon('Stereo') |
---|
151 | else: |
---|
152 | XTensionApp().turnoff('Stereo') |
---|
153 | |
---|
154 | def registerZoomWindowHotKey(self): |
---|
155 | self.zoomWindowHotKey = self.registerHotKey(self.zoomWindow, 42, cmdKey | controlKey) # cmd-ctrl-\ |
---|
156 | |
---|
157 | def unregisterZoomWindowHotKey(self): |
---|
158 | self.unregisterHotKey(self.zoomWindowHotKey) |
---|
159 | self.zoomWindowHotKey = None |
---|
160 | |
---|
161 | def zoomWindow(self): |
---|
162 | systemEvents = app(id='com.apple.systemEvents') |
---|
163 | frontName = systemEvents.processes.filter(its.frontmost)[1].name() |
---|
164 | if frontName == 'iTunes': |
---|
165 | systemEvents.processes['iTunes'].menu_bars[1]. \ |
---|
166 | menu_bar_items['Window'].menus.menu_items['Zoom'].click() |
---|
167 | return |
---|
168 | elif frontName in ('X11', 'Emacs'): # preserve C-M-\ |
---|
169 | self.unregisterZoomWindowHotKey() |
---|
170 | systemEvents.key_code(42, using=[k.command_down, k.control_down]) |
---|
171 | self.registerZoomWindowHotKey() |
---|
172 | return |
---|
173 | try: |
---|
174 | zoomed = app(frontName).windows[1].zoomed |
---|
175 | zoomed.set(not zoomed()) |
---|
176 | except (CommandError, RuntimeError): |
---|
177 | systemEvents.processes[frontName].windows. \ |
---|
178 | filter(its.subrole == 'AXStandardWindow').windows[1]. \ |
---|
179 | buttons.filter(its.subrole == 'AXZoomButton').buttons[1].click() |
---|
180 | |
---|
181 | def finishLaunching(self): |
---|
182 | super(StreamVision, self).finishLaunching() |
---|
183 | self.registerHotKey(self.displayTrackInfo, 100) # F8 |
---|
184 | self.registerHotKey(self.goToSite, 100, cmdKey) # cmd-F8 |
---|
185 | self.registerHotKey(self.playPause, 101) # F9 |
---|
186 | self.registerHotKey(lambda: iTunesApp().previous_track(), 109) # F10 |
---|
187 | self.registerHotKey(lambda: iTunesApp().next_track(), 103) # F11 |
---|
188 | self.registerHotKey(lambda: self.incrementRatingBy(-20), 109, shiftKey) # shift-F10 |
---|
189 | self.registerHotKey(lambda: self.incrementRatingBy(20), 103, shiftKey) # shift-F11 |
---|
190 | self.registerZoomWindowHotKey() |
---|
191 | NSDistributedNotificationCenter.defaultCenter().addObserver_selector_name_object_(self, self.displayTrackInfo, 'com.apple.iTunes.playerInfo', None) |
---|
192 | |
---|
193 | def sendEvent_(self, theEvent): |
---|
194 | if theEvent.type() == NSSystemDefined and \ |
---|
195 | theEvent.subtype() == kEventHotKeyPressedSubtype: |
---|
196 | self.hotKeyActions[theEvent.data1()]() |
---|
197 | super(StreamVision, self).sendEvent_(theEvent) |
---|
198 | |
---|
199 | if __name__ == "__main__": |
---|
200 | AppHelper.runEventLoop() |
---|