81 lines
2.5 KiB
Python
81 lines
2.5 KiB
Python
from __future__ import annotations
|
|
|
|
from dataclasses import dataclass
|
|
from typing import TYPE_CHECKING, Any, Optional
|
|
|
|
from aiogram.utils.dataclass import dataclass_kwargs
|
|
|
|
if TYPE_CHECKING:
|
|
from aiogram.types import LinkPreviewOptions
|
|
|
|
|
|
# @dataclass ??
|
|
class Default:
|
|
# Is not a dataclass because of JSON serialization.
|
|
|
|
__slots__ = ("_name",)
|
|
|
|
def __init__(self, name: str) -> None:
|
|
self._name = name
|
|
|
|
@property
|
|
def name(self) -> str:
|
|
return self._name
|
|
|
|
def __str__(self) -> str:
|
|
return f"Default({self._name!r})"
|
|
|
|
def __repr__(self) -> str:
|
|
return f"<{self}>"
|
|
|
|
|
|
@dataclass(**dataclass_kwargs(slots=True, kw_only=True))
|
|
class DefaultBotProperties:
|
|
"""
|
|
Default bot properties.
|
|
"""
|
|
|
|
parse_mode: Optional[str] = None
|
|
"""Default parse mode for messages."""
|
|
disable_notification: Optional[bool] = None
|
|
"""Sends the message silently. Users will receive a notification with no sound."""
|
|
protect_content: Optional[bool] = None
|
|
"""Protects content from copying."""
|
|
allow_sending_without_reply: Optional[bool] = None
|
|
"""Allows to send messages without reply."""
|
|
link_preview: Optional[LinkPreviewOptions] = None
|
|
"""Link preview settings."""
|
|
link_preview_is_disabled: Optional[bool] = None
|
|
"""Disables link preview."""
|
|
link_preview_prefer_small_media: Optional[bool] = None
|
|
"""Prefer small media in link preview."""
|
|
link_preview_prefer_large_media: Optional[bool] = None
|
|
"""Prefer large media in link preview."""
|
|
link_preview_show_above_text: Optional[bool] = None
|
|
"""Show link preview above text."""
|
|
show_caption_above_media: Optional[bool] = None
|
|
"""Show caption above media."""
|
|
|
|
def __post_init__(self) -> None:
|
|
has_any_link_preview_option = any(
|
|
(
|
|
self.link_preview_is_disabled,
|
|
self.link_preview_prefer_small_media,
|
|
self.link_preview_prefer_large_media,
|
|
self.link_preview_show_above_text,
|
|
)
|
|
)
|
|
|
|
if has_any_link_preview_option and self.link_preview is None:
|
|
from ..types import LinkPreviewOptions
|
|
|
|
self.link_preview = LinkPreviewOptions(
|
|
is_disabled=self.link_preview_is_disabled,
|
|
prefer_small_media=self.link_preview_prefer_small_media,
|
|
prefer_large_media=self.link_preview_prefer_large_media,
|
|
show_above_text=self.link_preview_show_above_text,
|
|
)
|
|
|
|
def __getitem__(self, item: str) -> Any:
|
|
return getattr(self, item, None)
|