56 lines
1.6 KiB
Python
56 lines
1.6 KiB
Python
|
import base64
|
||
|
|
import requests
|
||
|
|
|
||
|
|
|
||
|
|
class CredentialError(Exception):
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
class Spotify:
|
||
|
|
def __init__(self, client_id, client_secret):
|
||
|
|
self.client_id = client_id
|
||
|
|
self.client_secret = client_secret
|
||
|
|
self.access_token = ""
|
||
|
|
self.token_type = ""
|
||
|
|
self.expiration_time = 0
|
||
|
|
self.scope = ""
|
||
|
|
|
||
|
|
def request_token(self):
|
||
|
|
b64_encoded = base64.b64encode(
|
||
|
|
"{}:{}".format(self.client_id, self.client_secret).encode()
|
||
|
|
)
|
||
|
|
headers = {"Authorization": "Basic {}".format(b64_encoded.decode("ascii"))}
|
||
|
|
|
||
|
|
data = {"grant_type": "client_credentials"}
|
||
|
|
|
||
|
|
post = requests.post(
|
||
|
|
"https://accounts.spotify.com/api/token", headers=headers, data=data
|
||
|
|
)
|
||
|
|
status_code = post.status_code
|
||
|
|
|
||
|
|
if status_code == 401:
|
||
|
|
raise CredentialError("Provided credentials are not correct.")
|
||
|
|
|
||
|
|
if status_code == 200:
|
||
|
|
self.access_token = post.json()["access_token"]
|
||
|
|
self.token_type = post.json()["token_type"]
|
||
|
|
self.expiration_time = post.json()["expires_in"]
|
||
|
|
self.scope = post.json()["scope"]
|
||
|
|
|
||
|
|
def search(self, search_term, types):
|
||
|
|
if not self.access_token:
|
||
|
|
self.request_token()
|
||
|
|
|
||
|
|
encoded_search_term = urllib.parse.quote(search_term)
|
||
|
|
full_url = "https://api.spotify.com/v1/search?q={}&type={}".format(
|
||
|
|
encoded_search_term, ",".join(types)
|
||
|
|
)
|
||
|
|
|
||
|
|
get = requests.get(
|
||
|
|
full_url, headers={"Authorization": "Bearer " + self.access_token}
|
||
|
|
)
|
||
|
|
|
||
|
|
results = get.json()
|
||
|
|
|
||
|
|
return results
|