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 ows_text_campaigns.adapters.db.decorators import transactional from ows_text_campaigns.campaigns.enums import CampaignStatus from ows_text_campaigns.campaigns.exceptions import ( CampaignAlreadyQueuedError, CampaignMustBeScheduledError, ) from ows_text_campaigns.campaigns.services import CampaignService @dataclass(kw_only=True) class UnscheduleCampaignRequest(AuthRequest): campaign_id: str @singleton class UnscheduleCampaignHandler: permission = Permission("text_campaign", "schedule") def __init__( self, auth_service: AuthService, campaign_service: CampaignService, ) -> None: self.auth_service = auth_service self.campaign_service = campaign_service @transactional def handle(self, request: UnscheduleCampaignRequest) -> None: campaign = self.campaign_service.get_campaign(request.campaign_id) self.auth_service.authorize_account( request.identity_id, permission=self.permission, account=campaign.account, ) # Do not allow un-scheduling of non-scheduled campaigns if not campaign.is_scheduled: raise CampaignMustBeScheduledError # Do not allow un-scheduling of campaigns # that are scheduled to be sent in the period if self.campaign_service.should_enqueue_campaign(campaign): raise CampaignAlreadyQueuedError campaign.status = CampaignStatus.DRAFT campaign.send_at = None campaign.send_at_timezone = None campaign.recipients_count = 0 campaign.updated_by = request.identity_id