nambrogi
/
pomice
mirror of https://github.com/cloudwithax/pomice.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

using orjson

This commit is contained in:
Crussader 2022-03-04 18:12:23 +04:00 committed by GitHub
parent fa2c6fb90a
commit 76a6604ba8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
pomice/spotify/client.py
View File

@ -1,5 +1,6 @@
import re import re
import time import time
import orjson
from base64 import b64encode from base64 import b64encode
import aiohttp import aiohttp
@ -15,6 +16,8 @@ SPOTIFY_URL_REGEX = re.compile(
r"https?://open.spotify.com/(?P<type>album|playlist|track)/(?P<id>[a-zA-Z0-9]+)" r"https?://open.spotify.com/(?P<type>album|playlist|track)/(?P<id>[a-zA-Z0-9]+)"
) )
JSON_ENCODER = orjson.dumps
JSON_DECODER = orjson.loads
class Client: class Client:
"""The base client for the Spotify module of Pomice. """The base client for the Spotify module of Pomice.
@ -26,7 +29,7 @@ class Client:
self._client_id = client_id self._client_id = client_id
self._client_secret = client_secret self._client_secret = client_secret
self.session = aiohttp.ClientSession() self.session = aiohttp.ClientSession(json_serialize=JSON_ENCODER)
self._bearer_token: str = None self._bearer_token: str = None
self._expiry = 0 self._expiry = 0
@ -68,7 +71,7 @@ class Client:
f"Error while fetching results: {resp.status} {resp.reason}" f"Error while fetching results: {resp.status} {resp.reason}"
) )
data: dict = await resp.json() data: dict = await resp.json(loads=JSON_DECODER)
if spotify_type == "track": if spotify_type == "track":
return Track(data) return Track(data)
@ -93,7 +96,7 @@ class Client:
f"Error while fetching results: {resp.status} {resp.reason}" f"Error while fetching results: {resp.status} {resp.reason}"
) )
next_data: dict = await resp.json() next_data: dict = await resp.json(JSON_DECODER)
tracks += [ tracks += [
Track(track["track"]) Track(track["track"])