implemented downloading

src/music_kraken/pages/abstract.py

@@ -281,6 +281,8 @@ class Page:
 
     @classmethod
     def download(cls, music_object: Union[Song, Album, Artist, Label], download_features: bool = True):
+        print("downloading")
+        print(music_object)
         if type(music_object) is Song:
             return cls.download_song(music_object)
         if type(music_object) is Album:
@@ -292,6 +294,7 @@ class Page:
         
     @classmethod
     def download_label(cls, label: Label, download_features: bool = True, override_existing: bool = False):
+        cls.fetch_details(label)
         for artist in label.current_artist_collection:
             cls.download_artist(artist, download_features=download_features, override_existing=override_existing)
         
@@ -300,22 +303,35 @@ class Page:
 
     @classmethod
     def download_artist(cls, artist: Artist, download_features: bool = True, override_existing: bool = False):
+        cls.fetch_details(artist)
         for album in artist.main_album_collection:
             cls.download_album(album, override_existing=override_existing)
         
         if download_features:
-            for song in artist.feature_album:
+            for song in artist.feature_album.song_collection:
                 cls.download_song(song, override_existing=override_existing)
 
     @classmethod
     def download_album(cls, album: Album, override_existing: bool = False):
+        cls.fetch_details(album)
         for song in album.song_collection:
             cls.download_song(song, override_existing=override_existing)
 
     @classmethod
-    def download_song(cls, song: Song, override_existing: bool = False):
+    def download_song(cls, song: Song, override_existing: bool = False, create_target_on_demand: bool = True):
+        cls.fetch_details(song)
+        
         if song.target_collection.empty:
-            return
+            if create_target_on_demand and not song.main_artist_collection.empty and not song.album_collection.empty:
+                song.target_collection.append(
+                    Target(
+                            file=f"{song.title}.mp3",
+                            relative_to_music_dir=True,
+                            path=f"{song.main_artist_collection[0].name}/{song.album_collection[0].title}"
+                        )
+                )
+            else:
+                return
         
         target: Target
         if any(target.exists for target in song.target_collection) and not override_existing:
@@ -334,6 +350,8 @@ class Page:
         if len(sources) == 0:
             return
         
+        print("great")
+        
         temp_target: Target = Target(
             path=shared.TEMP_DIR,
             file=str(random.randint(0, 999999))

src/music_kraken/pages/download_center/download.py

@@ -42,4 +42,5 @@ class Download:
             return
         
         return source_page.fetch_object_from_source(source)
+        
     

src/music_kraken/pages/download_center/search.py

@@ -11,13 +11,21 @@ class MultiPageOptions:
     def __init__(
             self,
             max_displayed_options: int = 10,
-            option_digits: int = 3
+            option_digits: int = 3,
+            database_object: DatabaseObject = None,
+            page: Type[Page] = None
     ) -> None:
         self.max_displayed_options = max_displayed_options
         self.option_digits: int = option_digits
 
         self._length = 0
         self._current_option_dict: Dict[Type[Page], Options] = defaultdict(lambda: Options())
+        
+        self.database_object = database_object
+        self.page = page
+        
+        if database_object is not None and page is not None:
+            self[page] = database_object.options
 
     def __getitem__(self, key: Type[Page]):
         return self._current_option_dict[key]
@@ -95,6 +103,17 @@ class MultiPageOptions:
 
     def __repr__(self) -> str:
         return self.string_from_all_pages()
+    
+    def download(self) -> bool:
+        if self._length != 1:
+            return False
+        
+        if self.database_object is None or self.page is None:
+            return False
+        
+        self.page.download(self.database_object)
+        
+        return True
 
 
 class Search(Download):
@@ -132,6 +151,17 @@ class Search(Download):
         self._option_history.append(mpo)
         self._current_option = mpo
         return mpo
+    
+    def next_options_from_music_obj(self, database_obj: DatabaseObject, page: Type[Page]) -> MultiPageOptions:
+        mpo = MultiPageOptions(
+            max_displayed_options=self.max_displayed_options,
+            option_digits=self.option_digits,
+            database_object=database_obj,
+            page=page
+        )
+        self._option_history.append(mpo)
+        self._current_option = mpo
+        return mpo
 
     @property
     def _previous_options(self) -> MultiPageOptions:
@@ -172,8 +202,7 @@ class Search(Download):
         
         music_object = self.fetch_details(db_object)
         
-        mpo = self.next_options
-        mpo[page] = music_object.options
+        mpo = self.next_options_from_music_obj(music_object, page)
         
     def goto_previous(self):
         try:
@@ -195,3 +224,6 @@ class Search(Download):
         mpo[page] = new_object.options
         
         return True
+    
+    def download_chosen(self) -> bool:
+        return self._current_option.download()