from dataclasses import dataclass from anydi import singleton from fansifter_common.auth.requests import AuthRequest from fansifter_common.auth.services import AuthService from fansifter_common.auth.types import Permission from dmp.ad_reporting.exceptions import ( AdReportingCampaignNotFoundError, AdReportingReportLastCampaignDeleteError, ) from dmp.ad_reporting.services import AdReportingReportService from dmp.adapters.db import ReportingDB, transactional @dataclass(kw_only=True) class RemoveAdReportingReportCampaignRequest(AuthRequest): report_id: str campaign_id: str @singleton class RemoveAdReportingReportCampaignHandler: permission = Permission("ad_report", "edit") def __init__( self, db: ReportingDB, auth_service: AuthService, report_service: AdReportingReportService, ) -> None: self.db = db self.auth_service = auth_service self.report_service = report_service @transactional def handle(self, request: RemoveAdReportingReportCampaignRequest) -> None: report = self.report_service.get_report( report_id=request.report_id, identity_id=request.identity_id, ) self.auth_service.check_owned_resource( report.identity_id, permission=self.permission, resource_id=report.id, ) try: report.campaign_ids.remove(request.campaign_id) except ValueError as exc: raise AdReportingCampaignNotFoundError from exc if not report.campaign_ids: raise AdReportingReportLastCampaignDeleteError self.report_service.save_report(report) return None