initial commit for GitHub

server/app/model/albums.py

@@ -0,0 +1,94 @@
+from typing import Optional, Type
+
+from icecream import ic
+from pydantic import HttpUrl
+
+from app.constants import ALBUMS_TABLE, ARTISTS_TABLE
+from app.model.model_object import ModelObject
+from app.model.response_object import ResponseObject
+
+
+class Artist(ModelObject, ResponseObject):
+    artist_name: str
+    artist_url: Optional[HttpUrl] = None
+
+
+class Album(ModelObject, ResponseObject):
+    album_name: str
+    release_year: int
+    artist: Artist
+    front_artwork_url: HttpUrl
+    spotify_url: Optional[HttpUrl] = None
+    itunes_url: Optional[HttpUrl] = None
+    bandcamp_url: Optional[HttpUrl] = None
+    apple_music_url: Optional[HttpUrl] = None
+    rear_artwork_url: Optional[HttpUrl] = None
+    bandcamp_player: Optional[str] = None
+
+    @classmethod
+    def select_one(
+        cls, album_id: int, table_name: str = ALBUMS_TABLE
+    ) -> "Album | None":
+        cursor, conn = cls._get_cursor_and_conn()
+        cursor.execute(
+            f"""-- sql
+            SELECT
+                	al.id,
+                    al.album_name ,
+                    al.release_year,
+                    al.artist_id ,
+                    al.spotify_url ,
+                    al.itunes_url ,
+                    al.bandcamp_url ,
+                    al.apple_music_url ,
+                    al.front_artwork_url ,
+                    al.rear_artwork_url ,
+                    al.bandcamp_player ,
+                    ar.artist_name ,
+                    ar.artist_url
+            FROM {table_name} al
+            LEFT JOIN {ARTISTS_TABLE} ar
+            ON al.artist_id = ar.id
+            WHERE al.id = {album_id};
+            """
+        )
+        data: dict = cursor.fetchone()  # type: ignore
+        cls._close_cursor_and_conn(cursor, conn)
+        return cls._construct(cls, data) if data else None
+
+    @classmethod
+    def select_all(cls, table_name: str = ALBUMS_TABLE) -> list["Album"]:
+        cursor, conn = cls._get_cursor_and_conn()
+        cursor.execute(
+            f"""-- sql
+            SELECT
+                	al.id,
+                    al.album_name ,
+                    al.release_year,
+                    al.artist_id ,
+                    al.spotify_url ,
+                    al.itunes_url ,
+                    al.bandcamp_url ,
+                    al.apple_music_url ,
+                    al.front_artwork_url ,
+                    al.rear_artwork_url ,
+                    al.bandcamp_player ,
+                    ar.artist_name ,
+                    ar.artist_url
+            FROM {table_name} al
+            LEFT JOIN {ARTISTS_TABLE} ar
+            ON al.artist_id = ar.id;
+            """
+        )
+        data: dict = cursor.fetchall()  # type: ignore
+        cls._close_cursor_and_conn(cursor, conn)
+        return sorted(
+            [cls._construct(cls, row) for row in data],
+            key=lambda album: album.release_year,
+            reverse=True,
+        )
+
+    @classmethod
+    def _construct(cls, Obj: Type, data: dict) -> "Album":
+        data["artist"] = Artist(**data)
+        return Obj(**data)