semi completed pull function for album

src/goof.py

@@ -60,5 +60,7 @@ song_output_list = cache.pull_songs(album_ref=album_input.reference)
 print(len(song_output_list), song_output_list)
 
 # getting album
-album_output_list = cache.pull_albums(album_ref=album_input.ref)
-print(album_output_list[0])
+album_output_list = cache.pull_albums(album_ref=album_input.reference)
+album_output = album_output_list[0]
+print(album_output)
+print(album_output.tracklist)

src/music_kraken/database/new_database.py

@@ -39,7 +39,7 @@ FROM Lyrics
 WHERE {where};
 """
 ALBUM_QUERY = """
-SELECT Album.id AS album_id, title, copyright, album_status, year, date, country, barcode
+SELECT Album.id AS album_id, title, copyright, album_status, language, year, date, country, barcode
 FROM Album
 WHERE {where};
 """
@@ -268,7 +268,7 @@ class Database:
             url=source_row['url']
         ) for source_row in source_rows]
 
-    def get_song_from_row(self, song_result) -> Song:
+    def get_song_from_row(self, song_result, exclude_independent_relations: bool = False) -> Song:
         song_id = song_result['song_id']
         
         return Song(
@@ -286,17 +286,18 @@ class Database:
             album_ref=Reference(song_result['album_id'])
         )
 
-    def pull_songs(self, song_ref: Reference = None, album_ref: Reference = None) -> List[Song]:
+    def pull_songs(self, song_ref: Reference = None, album_ref: Reference = None, exclude_independent_relations: bool = False) -> List[Song]:
         """
         This function is used to get one song (including its children like Sources etc)
         from one song id (a reference object)
         :param song_ref:
+        :param album_ref:
+        :param exclude_independent_relations:
+        This excludes all relations from being fetched like for example the Album of the Song.
+        This is necessary when adding the Song as subclass as e.g. an Album (as tracklist or whatever).
         :return requested_song:
         """
-        """
-        if song_ref is None:
-            raise ValueError("The Song ref doesn't point anywhere. Remember to use the debugger.")
-        """
+
         where = "1=1"
         if song_ref is not None:
             where = f"Song.id=\"{song_ref.id}\""
@@ -307,17 +308,13 @@ class Database:
         self.cursor.execute(query)
 
         song_rows = self.cursor.fetchall()
-        """
-        if len(song_rows) == 0:
-            logger.warning(f"No song found for the id {song_ref.id}")
-            return Song()
-        if len(song_rows) > 1:
-            logger.warning(f"Multiple Songs found for the id {song_ref.id}. Defaulting to the first one.")
-        """
 
-        return [self.get_song_from_row(song_result=song_result) for song_result in song_rows]       
+        return [self.get_song_from_row(
+            song_result=song_result,
+            exclude_independent_relations=exclude_independent_relations
+        ) for song_result in song_rows]
 
-    def get_album_from_row(self, album_result) -> Album:
+    def get_album_from_row(self, album_result, exclude_independent_relations: bool = False) -> Album:
         album_id = album_result['album_id']
 
         album_obj = Album(
@@ -332,14 +329,17 @@ class Database:
             barcode=album_result['barcode']
         )
 
-        # getting the tracklist
-        tracklist: List[Song] = self.pull_songs(album_ref=Reference(id_=album_id))
-        for track in tracklist:
-            album_obj.add_song(track.reference, name=track.title)
+        if not exclude_independent_relations:
+            # getting the tracklist
+            tracklist: List[Song] = self.pull_songs(
+                album_ref=Reference(id_=album_id),
+                exclude_independent_relations=True
+            )
+            album_obj.set_tracklist(tracklist=tracklist)
 
         return album_obj
 
-    def pull_albums(self, album_ref: Reference = None) -> List[Album]:
+    def pull_albums(self, album_ref: Reference = None, exclude_independent_relations: bool = False) -> List[Album]:
         """
         This function is used to get matching albums/releses 
         from one song id (a reference object)
@@ -355,5 +355,11 @@ class Database:
 
         album_rows = self.cursor.fetchall()
 
+        return [self.get_album_from_row(
+            album_result=album_row,
+            exclude_independent_relations=exclude_independent_relations
+        ) for album_row in album_rows]
+
+
 if __name__ == "__main__":
     cache = Database("")

src/music_kraken/database/objects/song.py

@@ -11,11 +11,11 @@ from .database_object import (
     Reference
 )
 
-
 """
 All Objects dependent 
 """
 
+
 class SongAttribute:
     def __init__(self, song_ref: Reference = None):
         # the reference to the song the lyrics belong to
@@ -190,6 +190,11 @@ class Song(DatabaseObject):
         self.album_ref = album_ref
         self.artist_refs = artist_refs
 
+    def __eq__(self, other):
+        if type(other) != type(self):
+            return False
+        return self.id == other.id
+
     def __str__(self) -> str:
         return f"\"{self.title}\" by {', '.join(self.artist_names)}"
 
@@ -227,6 +232,7 @@ class Song(DatabaseObject):
 All objects dependend on Album
 """
 
+
 class Album(DatabaseObject):
     """
     -------DB-FIELDS-------
@@ -240,18 +246,18 @@ class Album(DatabaseObject):
     barcode         TEXT,
     song_id         BIGINT,
     """
+
     def __init__(
-        self,
-        id_: str = None,
-        title: str = None,
-        copyright_: str = None,
-        album_status: str = None,
-        language: str = None,
-        year: str = None,
-        date: str = None,
-        country: str = None,
-        barcode: str = None,
-        song_ref_list: List[Reference] = []
+            self,
+            id_: str = None,
+            title: str = None,
+            copyright_: str = None,
+            album_status: str = None,
+            language: str = None,
+            year: str = None,
+            date: str = None,
+            country: str = None,
+            barcode: str = None,
     ) -> None:
         DatabaseObject.__init__(self, id_=id_)
         self.title: str = title
@@ -263,41 +269,13 @@ class Album(DatabaseObject):
         self.country: str = country
         self.barcode: str = barcode
 
-        self.song_ref_list: List[Reference] = song_ref_list
-        self.track_names = {}
+        self.tracklist: List[Song] = []
 
-    def add_song(self, song_ref: Reference, name: str = None):
-        if name is not None:
-            self.track_names[song_ref.id] = name
+    def set_tracklist(self, tracklist: List[Song]):
+        self.tracklist = tracklist
 
-        for existing_song_ref in self.song_ref_list:
-            if song_ref == existing_song_ref:
+    def add_song(self, song: Song):
+        for existing_song in self.tracklist:
+            if existing_song == song:
                 return
-        self.song_ref_list.append(song_ref)
-
-        if song_ref.id not in self.track_names:
-            self.track_names[song_ref.id] = None
-
-
-
-
-
-if __name__ == "__main__":
-    """
-    Example for creating a Song object
-    """
-
-    song = Song(
-        title="Vein Deep in the Solution",
-        release_name="One Final Action",
-        target=Target(file="~/Music/genre/artist/album/song.mp3", path="~/Music/genre/artist/album"),
-        metadata={
-            "album": "One Final Action"
-        },
-        lyrics=[
-            Lyrics(text="these are some depressive lyrics", language="en")
-        ],
-        sources=[
-            Source(src="youtube", url="https://youtu.be/dfnsdajlhkjhsd")
-        ]
-    )
+        self.tracklist.append(song)