From 06f5411c472c1ea5021f86f4d481d4e2f2d99165 Mon Sep 17 00:00:00 2001 From: Hellow <74311245+HeIIow2@users.noreply.github.com> Date: Tue, 12 Sep 2023 18:21:24 +0200 Subject: [PATCH] feat: added contact collection in the artist --- src/music_kraken/objects/song.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/music_kraken/objects/song.py b/src/music_kraken/objects/song.py index 3f46d4b..b7d021e 100644 --- a/src/music_kraken/objects/song.py +++ b/src/music_kraken/objects/song.py @@ -8,6 +8,7 @@ from ..utils.enums.album import AlbumType, AlbumStatus from .collection import Collection from .formatted_text import FormattedText from .lyrics import Lyrics +from .contact import Contact from .metadata import ( Mapping as id3Mapping, ID3Timestamp, @@ -481,6 +482,7 @@ class Artist(MainObject): source_list: List[Source] = None, feature_song_list: List[Song] = None, main_album_list: List[Album] = None, + contact_list: List[Contact] = None, notes: FormattedText = None, lyrical_themes: List[str] = None, general_genre: str = "", @@ -516,6 +518,8 @@ class Artist(MainObject): self.main_album_collection: Collection[Album] = Collection(data=main_album_list, element_type=Album) self.label_collection: Collection[Label] = Collection(data=label_list, element_type=Label) + self.contact_collection: Collection[Label] = Collection(data=contact_list, element_type=Contact) + def _add_other_db_objects(self, object_type: Type["DatabaseObject"], object_list: List["DatabaseObject"]): if object_type is Song: # this doesn't really make sense @@ -628,7 +632,8 @@ class Artist(MainObject): return [ ('id', self.id), ('name', self.unified_name), - *[('url', source.url) for source in self.source_collection] + *[('url', source.url) for source in self.source_collection], + *[('contact', contact.value) for contact in self.contact_collection] ] @property