diff --git a/kontor-api/src/db/models/comic.py b/kontor-api/src/db/models/comic.py index 678e9c0..0e1dae6 100644 --- a/kontor-api/src/db/models/comic.py +++ b/kontor-api/src/db/models/comic.py @@ -125,6 +125,18 @@ class Issue(Base, BaseMixin): return works + def get_artists(self) -> Dict[Any, List[Any]]: + works: Dict[Any, List[Any]] = {} + for work in self.issue_works: + work_type = work.work_type + artist = work.artist + if work_type in works: + works[work_type].append(artist) + else: + works[work_type] = [artist] + return works + + class Artist(Base, BaseMixin): __tablename__ = "artist" name: Mapped[str] = mapped_column(nullable=False)