from collections.abc import Sequence from dataclasses import dataclass from typing import ClassVar from anydi import singleton from fansifter_common.auth.requests import AuthAccountFilterRequest from fansifter_common.auth.services import AuthService from fansifter_common.auth.types import Permission from dmp.adapters.db import ReportingDB from dmp.artists.dtos import Artist, ArtistCriteria from dmp.artists.repositories import ArtistRepository @dataclass(kw_only=True) class GetArtistsRequest(AuthAccountFilterRequest): DEFAULT_LIMIT: ClassVar[int] = 50 DEFAULT_OFFSET: ClassVar[int] = 0 search: str | None = None limit: int offset: int @dataclass(frozen=True) class GetArtistsResponse: total: int items: Sequence[Artist] @singleton class GetArtistsHandler: permission = Permission("roster", "view") def __init__( self, db: ReportingDB, auth_service: AuthService, artist_repository: ArtistRepository, ) -> None: self.db = db self.auth_service = auth_service self.artist_repository = artist_repository def handle(self, request: GetArtistsRequest) -> GetArtistsResponse: account_access = self.auth_service.authorize_for_permission( request.identity_id, permission=self.permission ) vendor_ids = account_access.filter_vendor_ids(request.vendor_id) subaccount_ids = account_access.filter_subaccount_ids(request.subaccount_id) criteria = ArtistCriteria( vendor_ids=vendor_ids, subaccount_ids=subaccount_ids, search=request.search, ) total = self.artist_repository.count_by_criteria(criteria) if total == 0: return GetArtistsResponse(total=total, items=[]) artists = self.artist_repository.find_by_criteria( criteria, limit=request.limit, offset=request.offset, ) return GetArtistsResponse(total=total, items=artists)