from dataclasses import dataclass 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 ows_text_campaigns.campaigns.dtos import ( CampaignDetailedAnalytics, ) from ows_text_campaigns.campaigns.services import CampaignService from ows_text_campaigns.campaigns.utils import ( extract_shortened_urls_from_markup, ) @dataclass(kw_only=True) class GetCampaignAnalyticsRequest(AuthAccountFilterRequest): campaign_id: str @singleton class GetCampaignAnalyticsHandler: permission = Permission("text_campaign", "view") def __init__( self, auth_service: AuthService, campaign_service: CampaignService, ) -> None: self.auth_service = auth_service self.campaign_service = campaign_service def handle(self, request: GetCampaignAnalyticsRequest) -> CampaignDetailedAnalytics: campaign = self.campaign_service.get_campaign(request.campaign_id) self.auth_service.authorize_account( request.identity_id, account=campaign.account, permission=self.permission, ) short_urls = extract_shortened_urls_from_markup(campaign.get_content()) has_shortened_urls = len(short_urls) > 0 return self.campaign_service.get_campaign_analytics( campaign.id, has_shortened_urls )