diff --git a/spotify_scraper.py b/spotify_scraper.py index 25036af..1a5e0cf 100644 --- a/spotify_scraper.py +++ b/spotify_scraper.py @@ -6,6 +6,7 @@ from utils.spotify_album import SpotifyAlbum from utils.spotify_playlist import SpotifyPlaylist from utils.spotify_category import SpotifyCategory from spotify_client import SpotifyClient +from typing import List class SpotifyScraper: @@ -160,7 +161,7 @@ class SpotifyScraper: def get_categories(self, limit=50) -> str: return self.get(f'https://api.spotify.com/v1/browse/categories/?limit={limit}&country=IL').json() - def get_categories_full(self, limit=50, query:str='') -> list[SpotifyCategory]: + def get_categories_full(self, limit=50, query:str='') -> List[SpotifyCategory]: categories = self.get_categories() categories_data = [] os.makedirs(f'{settings.DEFAULT_DOWNLOAD_DIRECTORY}/{settings.CATEGORY_METADATA_SUB_DIR}/', exist_ok=True) diff --git a/utils/spotify_playlist.py b/utils/spotify_playlist.py index 11eb9ec..0b68ef2 100644 --- a/utils/spotify_playlist.py +++ b/utils/spotify_playlist.py @@ -4,6 +4,7 @@ import requests import os from config import settings from utils.spotify_track import SpotifyTrack +from typing import List class SpotifyPlaylist: @@ -13,7 +14,7 @@ class SpotifyPlaylist: title = '' description = '' - def __init__(self, spotify_id, tracks:list[SpotifyTrack], data): + def __init__(self, spotify_id, tracks:List[SpotifyTrack], data): self.spotify_id = spotify_id self.tracks = tracks self.title = data['name']