• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• tekore-master
  • .github
    • PULL_REQUEST_TEMPLATE.md
    • ISSUE_TEMPLATE
      • feature.md
      • bug.md
      • config.yml
    • workflows
      • environment.yml
      • scheduled.yml
  • LICENSE
  • readme_pypi.rst
  • readme.rst
  • setup.py
  • .travis.yml
  • tests
    • _util.py
    • config.py
    • client
      • show.py
      • full.py
      • paging.py
      • search.py
      • browse.py
      • personalisation.py
      • user.py
      • artist.py
      • player.py
      • playlist.py
      • album.py
      • follow.py
      • __init__.py
      • track.py
      • library.py
      • episode.py
      • _resources.py
      • base.py
    • convert.py
    • _start.py
    • model.py
    • __init__.py
    • auth
      • util.py
      • refreshing.py
      • scope.py
      • __init__.py
      • expiring.py
    • sender
      • retrying.py
      • caching.py
      • client.py
      • options.py
      • concrete.py
      • __init__.py
      • base.py
    • error.py
    • conftest.py
  • .gitignore
  • docs
    • src
      • release_notes.rst
      • reference.rst
      • examples
        • discord_bot.rst
        • scripts.rst
        • artist_follower.rst
        • auth_server.rst
        • scripts
          • albums_top_artist.rst
          • recommended_playlist.rst
          • tracks_new_release.rst
          • scrape_playlists.rst
          • follow_by_search.rst
          • play_saved_album.rst
          • analyse_from_playlist.rst
          • follow_category_playlist.rst
          • related_artists_top_artist.rst
        • async_server.rst
      • advanced_usage.rst
      • extras
        • performance
          • performance.rst
      • getting_started.rst
      • examples.rst
      • extras.rst
      • reference
        • client.rst
        • conversions.rst
        • errors.rst
        • config.rst
        • auth.rst
        • senders.rst
        • models.rst
      • 404.rst
      • index.rst
      • conf.py
    • Makefile
    • make.bat
    • requirements.txt
  • MANIFEST.in
  • tox.ini
  • tekore
    • _auth
      • util.py
      • refreshing.py
      • scope.py
      • __init__.py
      • expiring.py
    • _convert.py
    • _model
      • serialise.py
      • local.py
      • recommendations.py
      • show
        • full.py
        • __init__.py
        • base.py
      • paging.py
      • category.py
      • user.py
      • audio_analysis.py
      • artist.py
      • play_history.py
      • playlist.py
      • __init__.py
      • track.py
      • member.py
      • currently_playing.py
      • error.py
      • context.py
      • album
        • full.py
        • __init__.py
        • base.py
      • audio_features.py
      • episode.py
      • device.py
      • base.py
    • _start.py
    • _config.py
    • _error.py
    • model.py
    • _client
      • full.py
      • paging.py
      • api
        • show.py
        • browse
          • validate.py
          • __init__.py
        • search.py
        • personalisation.py
        • user.py
        • artist.py
        • album.py
        • follow.py
        • playlist
          • modify.py
          • items.py
          • __init__.py
          • view.py
        • __init__.py
        • track.py
        • player
          • modify.py
          • __init__.py
          • view.py
        • library.py
        • episode.py
      • decor
        • __init__.py
        • handle.py
      • chunked.py
      • __init__.py
      • process.py
      • base.py
    • VERSION
    • __init__.py
    • _sender
      • client.py
      • extending.py
      • concrete.py
      • __init__.py

import json

from typing import Optional
from requests import Request

from tekore._sender import Sender, Client

prefix = 'https://api.spotify.com/v1/'


def build_url(url: str) -> str:
    """Attach API address to endpoint."""
    if not url.startswith('http'):
        url = prefix + url
    return url


def parse_url_params(params: Optional[dict]) -> Optional[dict]:
    """Generate parameter dict and filter Nones."""
    params = params or {}
    return {k: v for k, v in params.items() if v is not None} or None


class SpotifyBase(Client):
    """Base client with options and utility functions."""

    def __init__(
            self,
            token=None,
            sender: Sender = None,
            asynchronous: bool = None,
            max_limits_on: bool = False,
            chunked_on: bool = False,
    ):
        # Docstring in the main client
        super().__init__(sender, asynchronous)
        self.token = token
        self.max_limits_on = max_limits_on
        self.chunked_on = chunked_on

    def _create_headers(self, content_type: str = 'application/json'):
        return {
            'Authorization': f'Bearer {str(self.token)}',
            'Content-Type': content_type
        }

    def _request(
            self,
            method: str,
            url: str,
            payload=None,
            params: dict = None
    ):
        return Request(
            method=method,
            url=build_url(url),
            headers=self._create_headers(),
            params=parse_url_params(params),
            data=json.dumps(payload) if payload is not None else None
        )

    def _get(self, url: str, payload=None, **params):
        return self._request('GET', url, payload=payload, params=params)

    def _post(self, url: str, payload=None, **params):
        return self._request('POST', url, payload=payload, params=params)

    def _delete(self, url: str, payload=None, **params):
        return self._request('DELETE', url, payload=payload, params=params)

    def _put(self, url: str, payload=None, **params):
        return self._request('PUT', url, payload=payload, params=params)