splitted zspotify.py file into mutiple files

src/zspotify.py

@@ -0,0 +1,93 @@
+#! /usr/bin/env python3
+
+"""
+ZSpotify
+It's like youtube-dl, but for Spotify.
+
+(Made by Deathmonger/Footsiefat - @doomslayer117:matrix.org)
+"""
+import json
+import os
+import os.path
+from getpass import getpass
+from typing import Any
+
+import requests
+from librespot.audio.decoders import VorbisOnlyAudioQuality
+from librespot.core import Session
+
+from const import CREDENTIALS_JSON, TYPE, \
+    PREMIUM, USER_READ_EMAIL, AUTHORIZATION, OFFSET, LIMIT, CONFIG_FILE_PATH, FORCE_PREMIUM, RAW_AUDIO_AS_IS
+from utils import MusicFormat
+
+
+class ZSpotify:
+    SESSION: Session = None
+    DOWNLOAD_QUALITY = None
+    CONFIG = {}
+
+    def __init__(self):
+        ZSpotify.load_config()
+        ZSpotify.check_raw()
+        ZSpotify.login()
+
+    @classmethod
+    def login(cls):
+        """ Authenticates with Spotify and saves credentials to a file """
+
+        if os.path.isfile(CREDENTIALS_JSON):
+            try:
+                cls.SESSION = Session.Builder().stored_file().create()
+                return
+            except RuntimeError:
+                pass
+        while True:
+            user_name = input('Username: ')
+            password = getpass()
+            try:
+                cls.SESSION = Session.Builder().user_pass(user_name, password).create()
+                return
+            except RuntimeError:
+                pass
+
+    @classmethod
+    def load_config(cls) -> None:
+        with open(CONFIG_FILE_PATH, encoding='utf-8') as config_file:
+            cls.CONFIG = json.load(config_file)
+
+    @classmethod
+    def get_config(cls, key) -> Any:
+        return cls.CONFIG.get(key)
+
+    @classmethod
+    def check_raw(cls) -> None:
+        if cls.get_config(RAW_AUDIO_AS_IS):
+            cls.DOWNLOAD_FORMAT = MusicFormat.WAV
+
+    @classmethod
+    def get_content_stream(cls, content_id, quality):
+        return cls.SESSION.content_feeder().load(content_id, VorbisOnlyAudioQuality(quality), False, None)
+
+    @classmethod
+    def get_auth_header(cls):
+        return {AUTHORIZATION: f'Bearer {cls.SESSION.tokens().get(USER_READ_EMAIL)}'}
+
+    @classmethod
+    def get_auth_header_and_params(cls, limit, offset):
+        return {AUTHORIZATION: f'Bearer {cls.SESSION.tokens().get(USER_READ_EMAIL)}'}, {LIMIT: limit, OFFSET: offset}
+
+    @classmethod
+    def invoke_url_with_params(cls, url, limit, offset, **kwargs):
+        headers, params = cls.get_auth_header_and_params(limit=limit, offset=offset)
+        params.update(kwargs)
+        return requests.get(url, headers=headers, params=params).json()
+
+    @classmethod
+    def invoke_url(cls, url):
+        headers = cls.get_auth_header()
+        return requests.get(url, headers=headers).json()
+
+    @classmethod
+    def check_premium(cls) -> bool:
+        """ If user has spotify premium return true """
+        return (cls.SESSION.get_user_attribute(TYPE) == PREMIUM) or cls.get_config(FORCE_PREMIUM)