diff --git a/spotify_model/__init__.py b/spotify_model/__init__.py index 12e9131..3eae198 100644 --- a/spotify_model/__init__.py +++ b/spotify_model/__init__.py @@ -6,11 +6,12 @@ from .album import SearchAlbum, SimplifiedAlbum from .extra import ReleaseDatePrecision from .paging import Paging from .playlist import SimplifiedPlaylist -from .track import SimplifiedTrack +from .track import PlaylistTrack, SimplifiedTrack, Track from .user import PrivateUser, PublicUser __all__ = [ "Paging", + "PlaylistTrack", "PrivateUser", "PublicUser", "ReleaseDatePrecision", @@ -18,4 +19,5 @@ __all__ = [ "SimplifiedAlbum", "SimplifiedPlaylist", "SimplifiedTrack", + "Track", ] diff --git a/spotify_model/album.py b/spotify_model/album.py index 528faca..16360a4 100644 --- a/spotify_model/album.py +++ b/spotify_model/album.py @@ -30,7 +30,7 @@ class SearchAlbum(BaseModel): class SimplifiedAlbum(SearchAlbum): """ - Album as returned by the Album API + Album as returned by non-album APIs https://developer.spotify.com/documentation/web-api/reference/#object-simplifiedalbumobject """ diff --git a/spotify_model/track.py b/spotify_model/track.py index 1e44a70..8e8d99a 100644 --- a/spotify_model/track.py +++ b/spotify_model/track.py @@ -1,14 +1,18 @@ """ Classes designed to manage track-like objects """ +from datetime import datetime from typing import Dict, List, Optional, Union from pydantic import BaseModel, Field +from .album import SearchAlbum +from .user import PublicUser + class SimplifiedTrack(BaseModel): """ - Track as returned by the Tracks API + Track as returned by non-track APIs https://developer.spotify.com/documentation/web-api/reference/#object-simplifiedtrackobject """ @@ -28,5 +32,29 @@ class SimplifiedTrack(BaseModel): preview_url: Optional[str] restrictions: Optional[str] track_number: int - type_: str = Field(alias="type") + spotify_type: str = Field(alias="type") uri: str + + +class Track(SimplifiedTrack): + """ + Track as returned by the Tracks APIs + + https://developer.spotify.com/documentation/web-api/reference/#object-trackobject + """ + + album: SearchAlbum + external_ids: Dict[str, str] + popularity: int + + +class PlaylistTrack(BaseModel): + """ + Track as returned from a playlist + """ + + added_at: datetime + added_by: PublicUser + is_local: bool + primary_color: Optional[str] + track: Track diff --git a/spotify_model/user.py b/spotify_model/user.py index d71f233..8ae25e7 100644 --- a/spotify_model/user.py +++ b/spotify_model/user.py @@ -1,7 +1,7 @@ """ Classes for managing users """ -from typing import Any, Dict, List +from typing import Any, Dict, List, Optional from pydantic import BaseModel, Field @@ -13,7 +13,7 @@ class PublicUser(BaseModel): https://developer.spotify.com/documentation/web-api/reference/#object-publicuserobject """ - display_name: str + display_name: Optional[str] external_urls: Dict[str, str] href: str spotify_id: str = Field(alias="id")