Use new StarRecord class for StarStore objects

Author: C0rn3j

src/tauon/t_modules/t_db_migrate.py

@@ -9,7 +9,7 @@
 from tauon.t_modules.t_extra import RadioPlaylist, RadioStation, TauonPlaylist, TauonQueueItem
 
 if TYPE_CHECKING:
-	from tauon.t_modules.t_main import GuiVar, Prefs, StarStore, Tauon, TrackClass
+	from tauon.t_modules.t_main import GuiVar, Prefs, StarRecord, StarStore, Tauon, TrackClass
 
 
 def database_migrate(
@@ -18,7 +18,7 @@ def database_migrate(
 	db_version: float,
 	master_library: dict[int, TrackClass],
 	install_mode: bool,
-	multi_playlist: list | list[TauonPlaylist],
+	multi_playlist: list[str | int | bool] | list[TauonPlaylist],
 	star_store: StarStore,
 	a_cache_dir: str,
 	cache_directory: Path,
@@ -29,7 +29,7 @@ def database_migrate(
 	gen_codes: dict[int, str],
 	prefs: Prefs,
 	radio_playlists: list[dict[str, int | str | list[dict[str, str]]]] | list[RadioPlaylist],
-	p_force_queue: list | list[TauonQueueItem],
+	p_force_queue: list[int | bool] | list[TauonQueueItem],
 	theme: int,
 ) -> tuple[
 	dict[int, TrackClass],
@@ -545,7 +545,7 @@ def database_migrate(
 		p_force_queue = new_queue
 
 	if db_version <= 69:
-		logging.info("Updating database to version 69")
+		logging.info("Updating database to version 70")
 		new_radio_playlists: list[RadioPlaylist] = []
 		for playlist in radio_playlists:
 			stations: list[RadioStation] = []
@@ -567,5 +567,20 @@ def database_migrate(
 					stations=stations))
 		radio_playlists = new_radio_playlists
 
+	if db_version <= 70:
+		logging.info("Updating database to version 71")
+		new_starstore_db: dict[tuple[str, str, str], StarRecord] = {}
+		old_record: list[int | str] = [] # Here just for typing
+		for key, old_record in star_store.db.items():
+			new_record = star_store.new_object()
+			new_record.playtime = old_record[0]
+			new_record.loved = "L" in old_record[1]
+			new_record.rating = old_record[2]
+			# There was a bug where the fourth element was not set
+			if len(old_record) == 4:
+				new_record.loved_timestamp = old_record[3]
+			new_starstore_db[key] = new_record
+
+		star_store.db = new_starstore_db
 
 	return master_library, multi_playlist, star_store, p_force_queue, theme, prefs, gui, gen_codes, radio_playlists