e8e9147a |
1 | import socket, hmac, urllib, urllib2, pprint, md5, os, pickle |
2 | from operator import itemgetter |
3 | |
4 | THUMB_URL = 'http://beta.grooveshark.com/static/amazonart/' |
5 | THUMB_URL_DEFAULT = 'http://grooveshark.com/webincludes/logo/Grooveshark_Logo_No-Text.png' |
6 | SONG_LIMIT = 25 |
7 | ALBUM_LIMIT = 15 |
8 | ARTIST_LIMIT = 15 |
9 | |
10 | |
11 | class GrooveAPI: |
12 | |
13 | sessionID = '' |
14 | userID = 0 |
15 | host = 'api.grooveshark.com' |
16 | |
17 | # Constructor |
18 | def __init__(self, sessionID = ''): |
19 | import simplejson |
20 | self.simplejson = simplejson |
21 | socket.setdefaulttimeout(40) |
22 | if sessionID == '': |
23 | self.sessionID = self._getSessionID() |
24 | else: |
25 | self.sessionID = sessionID |
26 | |
27 | def _keySort(self, d): |
28 | return [(k,d[k]) for k in sorted(d.keys())] |
29 | |
30 | # Make a message sig |
31 | def _createMessageSig(self, method, params, secret): |
32 | args = self._keySort(params); |
33 | data = ''; |
34 | for arg in args: |
35 | data += str(arg[0]) |
36 | data += str(arg[1]) |
37 | data = method + data |
38 | |
39 | h = hmac.new(secret, data) |
40 | return h.hexdigest() |
41 | |
42 | # The actual call to the API |
43 | def _callRemote(self, method, params = {}): |
44 | url = 'http://%s/ws/2.1/?method=%s&%s&wsKey=wordpress&sig=%s&format=json' % (self.host, method, urllib.urlencode(params), self._createMessageSig(method, params, 'd6c59291620c6eaa5bf94da08fae0ecc')) |
45 | print url |
46 | req = urllib2.Request(url) |
47 | response = urllib2.urlopen(req) |
48 | result = response.read() |
49 | pprint.pprint(result) |
50 | response.close() |
51 | try: |
52 | result = self.simplejson.loads(result) |
53 | return result |
54 | except: |
55 | return [] |
56 | |
57 | def getUserID(self): |
58 | return self.userID |
59 | |
60 | def getSessionID(self): |
61 | return self.sessionID |
62 | |
63 | def _getSessionID(self): |
64 | params = {} |
65 | result = self._callRemote('startSession', params) |
66 | return result['result']['sessionID'] |
67 | |
68 | # Make user authentication token |
69 | def _getUserToken(self, username, password): |
70 | return md5.new(username.lower() + md5.new(password).hexdigest()).hexdigest() |
71 | |
72 | # Authenticates the user for current API session |
73 | def _authenticateUser(self, username, token): |
74 | params = {'sessionID': self.sessionID, 'username': username, 'token': token} |
75 | result = self._callRemote('authenticateUser', params) |
76 | return result['result']['UserID'] |
77 | |
78 | # Login |
79 | def login(self, username, password): |
80 | token = self._getUserToken(username, password) |
81 | self.userID = self._authenticateUser(username, token) |
82 | return self.userID |
83 | |
84 | # Logs the user out |
85 | def logout(self): |
86 | self._callRemote('logout', {'sessionID' : self.sessionID}) |
87 | |
88 | # Search for albums |
89 | def getArtistSearchResults(self, query, limit=ARTIST_LIMIT): |
90 | result = self._callRemote('getArtistSearchResults', {'query' : query,'limit' : limit}) |
91 | if 'result' in result: |
92 | return self._parseArtists(result) |
93 | else: |
e58dda71 |
94 | return [] |
e8e9147a |
95 | |
96 | # Search for albums |
97 | def getAlbumSearchResults(self, query, limit=ALBUM_LIMIT): |
98 | result = self._callRemote('getAlbumSearchResults', {'query' : query,'limit' : limit}) |
99 | if 'result' in result: |
100 | return self._parseAlbums(result) |
101 | else: |
e58dda71 |
102 | return [] |
e8e9147a |
103 | |
104 | # Search for songs |
105 | def getSongSearchResults(self, query, limit=SONG_LIMIT): |
106 | result = self._callRemote('getSongSearchResultsEx', {'query' : query, 'limit' : limit}) |
107 | if 'result' in result: |
108 | return self._parseSongs(result) |
109 | else: |
e58dda71 |
110 | return [] |
e8e9147a |
111 | |
112 | # Gets the popular songs |
113 | def getPopularSongsToday(self, limit=SONG_LIMIT): |
114 | result = self._callRemote('getPopularSongsToday', {'limit' : limit}) |
115 | if 'result' in result: |
116 | return self._parseSongs(result) |
117 | else: |
e58dda71 |
118 | return [] |
e8e9147a |
119 | |
120 | # Gets the favorite songs of the logged-in user |
121 | def getUserFavoriteSongs(self): |
122 | if (self.userID == 0): |
123 | return []; |
124 | result = self._callRemote('getUserFavoriteSongs', {'sessionID' : self.sessionID}) |
125 | if 'result' in result: |
126 | return self._parseSongs(result) |
127 | else: |
e58dda71 |
128 | return [] |
e8e9147a |
129 | |
130 | # Get the url to link to a song on Grooveshark |
131 | def getSongURLFromSongID(self, songID): |
132 | result = self._callRemote('getSongURLFromSongID', {'songID' : songID}) |
133 | if 'result' in result: |
134 | url = result['result']['url'] |
135 | return url.encode('ascii', 'ignore') |
136 | else: |
137 | return '' |
138 | |
e58dda71 |
139 | # Get the url to link to a song on Grooveshark |
140 | def getSongInfo(self, songID): |
141 | result = self._callRemote('getSongInfoEx', {'songID' : songID}) |
142 | if 'result' in result and 'SongID' in result['result']: |
143 | info = result['result'] |
144 | if 'CoverArtFilename' in info and info['CoverArtFilename'] != None: |
145 | info['CoverArtFilename'] = THUMB_URL+info['CoverArtFilename'].encode('ascii', 'ignore') |
146 | else: |
147 | info['CoverArtFilename'] = THUMB_URL_DEFAULT |
148 | return info |
149 | else: |
150 | return '' |
151 | |
e8e9147a |
152 | # Gets the playlists of the logged-in user |
153 | def getUserPlaylists(self): |
154 | if (self.userID == 0): |
155 | return []; |
156 | result = self._callRemote('getUserPlaylists', {'sessionID' : self.sessionID}) |
157 | if 'result' in result: |
e58dda71 |
158 | return self._parsePlaylists(result) |
e8e9147a |
159 | else: |
160 | return [] |
e58dda71 |
161 | |
e8e9147a |
162 | # Creates a playlist with songs |
163 | def createPlaylist(self, name, songIDs): |
164 | result = self._callRemote('createPlaylist', {'name' : name, 'songIDs' : songIDs, 'sessionID' : self.sessionID}) |
e58dda71 |
165 | if 'result' in result and result['result']['success'] == 'true': |
166 | return result['result']['PlaylistID'] |
e8e9147a |
167 | elif 'errors' in result: |
e8e9147a |
168 | for error in result['errors']: |
169 | if (error['code'] == 800): |
e58dda71 |
170 | return -1 |
e8e9147a |
171 | else: |
172 | return 0 |
173 | else: |
174 | return 0 |
175 | |
176 | # Sets the songs for a playlist |
177 | def setPlaylistSongs(self, playlistID, songIDs): |
178 | result = self._callRemote('setPlaylistSongs', {'playlistID' : playlistID, 'songIDs' : songIDs, 'sessionID' : self.sessionID}) |
e58dda71 |
179 | if 'result' in result and result['result']['success'] == 'true': |
e8e9147a |
180 | return True; |
181 | else: |
182 | return False; |
183 | |
184 | # Gets the songs of a playlist |
185 | def getPlaylistSongs(self, playlistID): |
186 | result = self._callRemote('getPlaylistSongs', {'playlistID' : playlistID}); |
187 | if 'result' in result: |
e58dda71 |
188 | return self._parseSongs(result) |
e8e9147a |
189 | else: |
e58dda71 |
190 | return [] |
e8e9147a |
191 | |
192 | # Extract song data |
193 | def _parseSongs(self, items): |
194 | if 'result' in items: |
195 | i = 0 |
196 | list = [] |
197 | if 'songs' in items['result']: |
198 | l = len(items['result']['songs']) |
199 | index = 'songs' |
200 | elif 'song' in items['result']: |
201 | l = 1 |
202 | index = 'song' |
203 | else: |
e58dda71 |
204 | l = len(items['result']) |
e8e9147a |
205 | index = '' |
206 | while(i < l): |
207 | if index == 'songs': |
208 | s = items['result'][index][i] |
e58dda71 |
209 | elif index == 'song': |
e8e9147a |
210 | s = items['result'][index] |
e58dda71 |
211 | else: |
212 | s = items['result'][i] |
213 | if 'CoverArtFilename' not in s: |
214 | info = self.getSongInfo(s['SongID']) |
215 | coverart = info['CoverArtFilename'] |
216 | elif s['CoverArtFilename'] != None: |
e8e9147a |
217 | coverart = THUMB_URL+s['CoverArtFilename'].encode('ascii', 'ignore') |
218 | else: |
219 | coverart = THUMB_URL_DEFAULT |
220 | list.append([s['SongName'].encode('ascii', 'ignore'),\ |
221 | s['SongID'],\ |
222 | s['AlbumName'].encode('ascii', 'ignore'),\ |
223 | s['AlbumID'],\ |
224 | s['ArtistName'].encode('ascii', 'ignore'),\ |
225 | s['ArtistID'],\ |
226 | coverart]) |
227 | i = i + 1 |
228 | return list |
229 | else: |
230 | return [] |
231 | |
232 | # Extract artist data |
233 | def _parseArtists(self, items): |
234 | if 'result' in items: |
235 | i = 0 |
236 | list = [] |
237 | artists = items['result']['artists'] |
238 | while(i < len(artists)): |
239 | s = artists[i] |
240 | list.append([s['ArtistName'].encode('ascii', 'ignore'),\ |
241 | s['ArtistID']]) |
242 | i = i + 1 |
243 | return list |
244 | else: |
245 | return [] |
246 | |
247 | # Extract album data |
248 | def _parseAlbums(self, items): |
249 | if 'result' in items: |
250 | i = 0 |
251 | list = [] |
252 | albums = items['result']['albums'] |
253 | while(i < len(albums)): |
254 | s = albums[i] |
e58dda71 |
255 | if 'CoverArtFilename' in s and s['CoverArtFilename'] != None: |
e8e9147a |
256 | coverart = THUMB_URL+s['CoverArtFilename'].encode('ascii', 'ignore') |
257 | else: |
258 | coverart = THUMB_URL_DEFAULT |
259 | list.append([s['ArtistName'].encode('ascii', 'ignore'),\ |
260 | s['ArtistID'],\ |
261 | s['AlbumName'].encode('ascii', 'ignore'),\ |
262 | s['AlbumID'],\ |
263 | coverart]) |
264 | i = i + 1 |
265 | return list |
266 | else: |
267 | return [] |
268 | |
269 | def _parsePlaylists(self, items): |
270 | if 'result' in items: |
271 | i = 0 |
272 | list = [] |
e58dda71 |
273 | playlists = items['result'] |
e8e9147a |
274 | while(i < len(playlists)): |
275 | s = playlists[i] |
276 | list.append([s['PlaylistID'],\ |
e58dda71 |
277 | s['Name'].encode('ascii', 'ignore')]) |
e8e9147a |
278 | i = i + 1 |
279 | return list |
280 | else: |
281 | return [] |
282 | |
283 | |
284 | groovesharkApi = GrooveAPI('c4285d93d556349d284704100eb557d7') |
285 | print "SessionID: " + groovesharkApi.getSessionID() |
286 | print "UserID: " + str(groovesharkApi.login('stephendenham', 'lond0n')) |
287 | |
288 | #res = groovesharkApi.getSongSearchResults('jimmy jazz', 3) |
289 | #res = groovesharkApi.getPopularSongsToday() |
290 | #res = groovesharkApi.getSongURLFromSongID('27425375') |
291 | #res = groovesharkApi.getAlbumSearchResults('london calling', 3) |
292 | #res = groovesharkApi.getArtistSearchResults('the clash', 3) |
e58dda71 |
293 | #res = groovesharkApi.getUserFavoriteSongs() |
294 | #res = groovesharkApi.getUserPlaylists() |
295 | #res = groovesharkApi.getSongInfo('27425375') |
e8e9147a |
296 | |
e58dda71 |
297 | res = groovesharkApi.getPlaylistSongs(40902662) |
e8e9147a |
298 | |
299 | pprint.pprint(res) |
300 | |
301 | |
302 | |