218 lines
9.3 KiB
Python
218 lines
9.3 KiB
Python
import datetime
|
|
import typing
|
|
from dataclasses import dataclass
|
|
from uuid import UUID
|
|
|
|
from src import domain
|
|
|
|
from .analytics.get_creatives_analytics import get_creatives_analytics
|
|
from .analytics.get_external_channels_analytics import get_external_channels_analytics
|
|
from .analytics.get_placements_analytics import get_placements_analytics
|
|
from .analytics.get_spending_analytics import get_spending_analytics
|
|
from .creative.create_creative import create_creative
|
|
from .creative.delete_creative import delete_creative
|
|
from .creative.get_creative import get_creative
|
|
from .creative.get_creatives import get_creatives
|
|
from .creative.update_creative import update_creative
|
|
from .external_channel.create_external_channel import create_external_channel
|
|
from .external_channel.delete_external_channel import delete_external_channel
|
|
from .external_channel.get_external_channels import get_external_channels
|
|
from .external_channel.update_external_channel import update_external_channel
|
|
from .external_channel.update_external_channel_links import update_external_channel_links
|
|
from .get_me import get_me
|
|
from .placement.create_placement import create_placement
|
|
from .placement.delete_placement import delete_placement
|
|
from .placement.get_placement import get_placement
|
|
from .placement.get_placements import get_placements
|
|
from .placement.update_placement import update_placement
|
|
from .subscription.handle_subscription import handle_subscription
|
|
from .subscription.handle_unsubscription import handle_unsubscription
|
|
from .target_channel.connect_target_chan import connect_target_chan
|
|
from .target_channel.disconnect_target_chan import disconnect_target_chan
|
|
from .target_channel.disconnect_target_chan_by_tg_id import disconnect_target_chan_by_tg_id
|
|
from .target_channel.get_user_target_chans import get_user_target_chans
|
|
from .target_channel.update_target_chan_permissions import update_target_chan_permissions
|
|
from .telegram_login import telegram_login
|
|
from .validate_login_token import validate_login_token
|
|
from .views.fetch_views_manually import fetch_views_manually
|
|
from .views.get_views_history import get_views_history
|
|
from .views.run_views_worker_cycle import run_views_worker_cycle
|
|
from .views.update_views_manually import update_views_manually
|
|
|
|
|
|
class Database(typing.Protocol):
|
|
def transaction(self) -> typing.AsyncContextManager[None]: ...
|
|
|
|
async def get_user(self, *, user_id: UUID | None = None, telegram_id: int | None = None) -> domain.User | None: ...
|
|
|
|
async def create_user(self, user: domain.User) -> domain.User: ...
|
|
|
|
async def create_login_token(
|
|
self, user_id: UUID, token: str, expires_at: datetime.datetime
|
|
) -> domain.LoginToken: ...
|
|
|
|
async def get_login_token(self, token: str) -> domain.LoginToken | None: ...
|
|
|
|
async def mark_token_as_used(self, token: str) -> None: ...
|
|
|
|
async def get_target_channel(
|
|
self, user_id: UUID, channel_id: UUID | None = None, telegram_id: int | None = None
|
|
) -> domain.TargetChannel | None: ...
|
|
|
|
async def create_target_channel(self, channel: domain.TargetChannel) -> domain.TargetChannel: ...
|
|
|
|
async def upsert_target_channel(self, channel: domain.TargetChannel) -> domain.TargetChannel: ...
|
|
|
|
async def get_user_target_channels(self, user_id: UUID) -> list[domain.TargetChannel]: ...
|
|
|
|
async def update_target_channel(self, channel: domain.TargetChannel) -> domain.TargetChannel: ...
|
|
|
|
async def get_external_channel(
|
|
self, user_id: UUID, channel_id: UUID | None = None, telegram_id: int | None = None
|
|
) -> domain.ExternalChannel | None: ...
|
|
|
|
async def create_external_channel(self, channel: domain.ExternalChannel) -> domain.ExternalChannel: ...
|
|
|
|
async def get_external_channels_for_target(self, target_channel_id: UUID) -> list[domain.ExternalChannel]: ...
|
|
|
|
async def get_user_external_channels(self, user_id: UUID) -> list[domain.ExternalChannel]: ...
|
|
|
|
async def add_external_channel_to_targets(
|
|
self, external_channel_id: UUID, target_channel_ids: list[UUID]
|
|
) -> None: ...
|
|
|
|
async def remove_external_channel_from_target(self, external_channel_id: UUID, target_channel_id: UUID) -> None: ...
|
|
|
|
async def delete_external_channel(self, channel_id: UUID) -> None: ...
|
|
|
|
async def update_external_channel(self, channel: domain.ExternalChannel) -> domain.ExternalChannel: ...
|
|
|
|
async def get_creative(self, user_id: UUID, creative_id: UUID) -> domain.Creative | None: ...
|
|
|
|
async def create_creative(self, creative: domain.Creative) -> domain.Creative: ...
|
|
|
|
async def update_creative(self, creative: domain.Creative) -> domain.Creative: ...
|
|
|
|
async def get_user_creatives(
|
|
self, user_id: UUID, target_channel_id: UUID | None = None, include_archived: bool = False
|
|
) -> list[domain.Creative]: ...
|
|
|
|
async def delete_creative(self, creative_id: UUID) -> None: ...
|
|
|
|
async def get_placement(self, user_id: UUID, placement_id: UUID) -> domain.Placement | None: ...
|
|
|
|
async def create_placement(self, placement: domain.Placement) -> domain.Placement: ...
|
|
|
|
async def update_placement(self, placement: domain.Placement) -> domain.Placement: ...
|
|
|
|
async def get_user_placements(
|
|
self,
|
|
user_id: UUID,
|
|
target_channel_id: UUID | None = None,
|
|
external_channel_id: UUID | None = None,
|
|
creative_id: UUID | None = None,
|
|
include_archived: bool = False,
|
|
) -> list[domain.Placement]: ...
|
|
|
|
async def delete_placement(self, placement_id: UUID) -> None: ...
|
|
|
|
async def check_creative_in_use(self, creative_id: UUID) -> bool: ...
|
|
|
|
async def get_placement_by_invite_link(self, invite_link: str) -> domain.Placement | None: ...
|
|
|
|
async def get_subscriber(self, telegram_id: int) -> domain.Subscriber | None: ...
|
|
|
|
async def create_subscriber(self, subscriber: domain.Subscriber) -> domain.Subscriber: ...
|
|
|
|
async def upsert_subscriber(self, subscriber: domain.Subscriber) -> domain.Subscriber: ...
|
|
|
|
async def create_subscription(self, subscription: domain.Subscription) -> domain.Subscription: ...
|
|
|
|
async def get_subscription_by_subscriber_and_placement(
|
|
self, subscriber_id: UUID, placement_id: UUID
|
|
) -> domain.Subscription | None: ...
|
|
|
|
async def update_subscription(self, subscription: domain.Subscription) -> domain.Subscription: ...
|
|
|
|
async def get_active_subscriptions_by_subscriber_and_channel(
|
|
self, subscriber_id: UUID, channel_telegram_id: int
|
|
) -> list[domain.Subscription]: ...
|
|
|
|
async def get_active_subscription_by_subscriber_and_channel(
|
|
self, subscriber_id: UUID, channel_telegram_id: int
|
|
) -> domain.Subscription | None: ...
|
|
|
|
async def create_placement_views_history(
|
|
self, history: domain.PlacementViewsHistory
|
|
) -> domain.PlacementViewsHistory: ...
|
|
|
|
async def get_views_history(
|
|
self,
|
|
placement_id: UUID,
|
|
*,
|
|
from_date: datetime.datetime | None = None,
|
|
to_date: datetime.datetime | None = None,
|
|
) -> list[domain.PlacementViewsHistory]: ...
|
|
|
|
async def get_active_placements_with_urls(self) -> list[domain.Placement]: ... # Для worker
|
|
|
|
|
|
class TelegramBotWriter(typing.Protocol):
|
|
async def send_message(self, text: str, chat_id: int) -> None: ...
|
|
|
|
async def send_message_with_button(self, text: str, chat_id: int, button_text: str, button_url: str) -> None: ...
|
|
|
|
async def get_chat_member(self, chat_id: int, user_id: int) -> dict[str, typing.Any]: ...
|
|
|
|
async def create_chat_invite_link(self, chat_id: int, requires_approval: bool = False) -> str: ...
|
|
|
|
|
|
class TelegramParser(typing.Protocol):
|
|
async def get_post_views(self, post_url: str) -> int | None: ...
|
|
|
|
|
|
class JWTEncoder(typing.Protocol):
|
|
def encode_access_token(self, user_id: UUID, telegram_id: int, username: str | None = None) -> str: ...
|
|
|
|
|
|
@dataclass
|
|
class Usecase:
|
|
database: Database
|
|
telegram_bot: TelegramBotWriter
|
|
telegram_parser: TelegramParser
|
|
jwt_encoder: JWTEncoder
|
|
|
|
validate_login_token = validate_login_token
|
|
telegram_login = telegram_login
|
|
get_me = get_me
|
|
connect_target_chan = connect_target_chan
|
|
get_user_target_chans = get_user_target_chans
|
|
disconnect_target_chan = disconnect_target_chan
|
|
disconnect_target_chan_by_tg_id = disconnect_target_chan_by_tg_id
|
|
update_target_chan_permissions = update_target_chan_permissions
|
|
get_external_channels = get_external_channels
|
|
create_external_channel = create_external_channel
|
|
delete_external_channel = delete_external_channel
|
|
update_external_channel = update_external_channel
|
|
update_external_channel_links = update_external_channel_links
|
|
get_creatives = get_creatives
|
|
get_creative = get_creative
|
|
create_creative = create_creative
|
|
update_creative = update_creative
|
|
delete_creative = delete_creative
|
|
get_placements = get_placements
|
|
get_placement = get_placement
|
|
create_placement = create_placement
|
|
update_placement = update_placement
|
|
delete_placement = delete_placement
|
|
handle_subscription = handle_subscription
|
|
handle_unsubscription = handle_unsubscription
|
|
fetch_views = fetch_views_manually
|
|
run_views_worker_cycle = run_views_worker_cycle
|
|
get_views_history = get_views_history
|
|
update_views_manually = update_views_manually
|
|
get_placements_analytics = get_placements_analytics
|
|
get_creatives_analytics = get_creatives_analytics
|
|
get_external_channels_analytics = get_external_channels_analytics
|
|
get_spending_analytics = get_spending_analytics
|