extract feat from album titles

app/models/album.py

@@ -4,6 +4,9 @@ from dataclasses import dataclass
 from .track import Track
 from .artist import Artist
 from ..utils.hashing import create_hash
+from ..utils.parsers import parse_feat_from_title
+
+from app.settings import FromFlags
 
 
 @dataclass(slots=True)
@@ -16,13 +19,14 @@ class Album:
     title: str
     albumartists: list[Artist]
 
-    albumartisthash: str = ""
+    albumartists_hashes: str = ""
     image: str = ""
     count: int = 0
     duration: int = 0
     colors: list[str] = dataclasses.field(default_factory=list)
     date: str = ""
 
+    og_title: str = ""
     is_soundtrack: bool = False
     is_compilation: bool = False
     is_single: bool = False
@@ -32,8 +36,20 @@ class Album:
     genres: list[str] = dataclasses.field(default_factory=list)
 
     def __post_init__(self):
+        self.og_title = self.title
         self.image = self.albumhash + ".webp"
-        self.albumartisthash = "-".join(a.artisthash for a in self.albumartists)
+
+        if FromFlags.EXTRACT_FEAT:
+            featured, self.title = parse_feat_from_title(self.title)
+
+            if len(featured) > 0:
+                original_lower = "-".join([a.name.lower() for a in self.albumartists])
+                self.albumartists.extend([Artist(a) for a in featured if a.lower() not in original_lower])
+
+                from ..store.tracks import TrackStore
+                TrackStore.append_track_artists(self.albumhash, featured)
+
+        self.albumartists_hashes = "-".join(a.artisthash for a in self.albumartists)
 
     def set_colors(self, colors: list[str]):
         self.colors = colors

app/models/track.py

@@ -31,12 +31,13 @@ class Track:
 
     filetype: str = ""
     image: str = ""
-    artist_hashes: list[str] = dataclasses.field(default_factory=list)
+    artist_hashes: str = ""
     is_favorite: bool = False
     og_title: str = ""
 
     def __post_init__(self):
         self.og_title = self.title
+
         if self.artist is not None:
             artists = split_artists(self.artist)
             new_title = self.title
@@ -55,7 +56,7 @@ class Track:
 
             self.title = new_title
 
-            self.artist_hashes = [create_hash(a, decode=True) for a in artists]
+            self.artist_hashes = "-".join(create_hash(a, decode=True) for a in artists)
             self.artist = [ArtistMinimal(a) for a in artists]
 
             albumartists = split_artists(self.albumartist)
@@ -72,7 +73,21 @@ class Track:
         self.recreate_hash()
 
     def recreate_hash(self):
+        """
+        Recreates a track hash if the track title was altered
+        to prevent duplicate tracks having different hashes.
+        """
         if self.og_title == self.title:
             return
 
         self.trackhash = create_hash(", ".join([a.name for a in self.artist]), self.album, self.title)
+
+    def recreate_artists_hash(self):
+        self.artist_hashes = "-".join(a.artisthash for a in self.artist)
+
+    def add_artists(self, artists: list[str]):
+        for artist in artists:
+            if create_hash(artist) not in self.artist_hashes:
+                self.artist.append(ArtistMinimal(artist))
+
+        self.recreate_artists_hash()