controller: Handle the removal of the album art

diff --git a/src/qml/controller.py b/src/qml/controller.py
index b96e284..db4f296 100644 (file)
--- a/src/qml/controller.py
+++ b/src/qml/controller.py
@@ -98,11 +98,16 @@ class MussorgskyController (QtCore.QObject):
     def save_option_for (self, coversModel, index):
         print "Saving option", index
         coverItem = coversModel.getData (index)
-        print "Moving image:", coverItem.url, "to", coversModel.albumItem.get_aa ().get_media_art_path ()
-        os.rename (coverItem.url, coversModel.albumItem.get_aa().get_media_art_path ())
+        coverItem.save (coversModel.albumItem.get_aa ().get_media_art_path ())
         # Update the main model. Is this enough?
-        coversModel.albumItem.album_art = None
-        coversModel.albumItem.album_art = coversModel.albumItem.get_aa().get_media_art_path ()
+        if not coverItem.initialImage and not coverItem.deleteAction:
+            coversModel.albumItem.album_art = None
+            coversModel.albumItem.album_art = coversModel.albumItem.get_aa().get_media_art_path ()
+        elif coverItem.deleteAction:
+            coversModel.albumItem.resetAlbumArt ()
+
+        coversModel.cleanCache ()
+
 
     @QtCore.Slot ()
     def stop_pending_jobs (self):