diff options
Diffstat (limited to 'pandora/client.py')
-rw-r--r-- | pandora/client.py | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/pandora/client.py b/pandora/client.py index a98056c..e39b2cc 100644 --- a/pandora/client.py +++ b/pandora/client.py | |||
@@ -218,7 +218,11 @@ class APIClient(BaseAPIClient): | |||
218 | ) | 218 | ) |
219 | 219 | ||
220 | def create_station( | 220 | def create_station( |
221 | self, search_token=None, artist_token=None, track_token=None | 221 | self, |
222 | search_token=None, | ||
223 | artist_token=None, | ||
224 | track_token=None, | ||
225 | song_token=None, | ||
222 | ): | 226 | ): |
223 | from .models.station import Station | 227 | from .models.station import Station |
224 | 228 | ||
@@ -227,7 +231,9 @@ class APIClient(BaseAPIClient): | |||
227 | if search_token: | 231 | if search_token: |
228 | kwargs = {"musicToken": search_token} | 232 | kwargs = {"musicToken": search_token} |
229 | elif artist_token: | 233 | elif artist_token: |
230 | kwargs = {"trackToken": artist_token, "musicType": "artist"} | 234 | kwargs = {"musicToken": artist_token, "musicType": "artist"} |
235 | elif song_token: | ||
236 | kwargs = {"musicToken": song_token, "musicType": "song"} | ||
231 | elif track_token: | 237 | elif track_token: |
232 | kwargs = {"trackToken": track_token, "musicType": "song"} | 238 | kwargs = {"trackToken": track_token, "musicType": "song"} |
233 | else: | 239 | else: |