feat: implement /deb and /city command handlers for the bot.
This commit is contained in:
@@ -55,16 +55,36 @@ class CityCommandHandler:
|
|||||||
self.guard = guard
|
self.guard = guard
|
||||||
self.city_service = city_service
|
self.city_service = city_service
|
||||||
self.io_layer = io_layer
|
self.io_layer = io_layer
|
||||||
|
self._handled_message_keys: set[tuple[Any, Any]] = set()
|
||||||
|
|
||||||
|
def _mark_handled_once(self, message: Any) -> bool:
|
||||||
|
chat_id = getattr(getattr(message, "chat", None), "id", None)
|
||||||
|
message_id = getattr(message, "message_id", None)
|
||||||
|
key = (chat_id, message_id)
|
||||||
|
if key in self._handled_message_keys:
|
||||||
|
return False
|
||||||
|
self._handled_message_keys.add(key)
|
||||||
|
if len(self._handled_message_keys) > 4096:
|
||||||
|
self._handled_message_keys.clear()
|
||||||
|
return True
|
||||||
|
|
||||||
def register(self) -> None:
|
def register(self) -> None:
|
||||||
|
@self.bot.message_handler(commands=["city"])
|
||||||
|
def _city_command(message):
|
||||||
|
self.handle(message)
|
||||||
|
|
||||||
@self.bot.message_handler(
|
@self.bot.message_handler(
|
||||||
func=lambda message: _is_city_command_text(getattr(message, "text", None)),
|
regexp=r"^[//⁄∕╱⧸]\s*city(@[A-Za-z0-9_]+)?(\s|$)",
|
||||||
content_types=["text"],
|
content_types=["text"],
|
||||||
)
|
)
|
||||||
def _city(message):
|
def _city_regex(message):
|
||||||
self.handle(message)
|
self.handle(message)
|
||||||
|
|
||||||
def handle(self, message: Any) -> None:
|
def handle(self, message: Any) -> None:
|
||||||
|
if not self._mark_handled_once(message):
|
||||||
|
return
|
||||||
|
if not _is_city_command_text(getattr(message, "text", None)):
|
||||||
|
return
|
||||||
trace = CommandTrace("/city", message)
|
trace = CommandTrace("/city", message)
|
||||||
try:
|
try:
|
||||||
parts = (message.text or "").split(maxsplit=1)
|
parts = (message.text or "").split(maxsplit=1)
|
||||||
|
|||||||
+22
-2
@@ -55,16 +55,36 @@ class DebCommandHandler:
|
|||||||
self.guard = guard
|
self.guard = guard
|
||||||
self.deb_service = deb_service
|
self.deb_service = deb_service
|
||||||
self.io_layer = io_layer
|
self.io_layer = io_layer
|
||||||
|
self._handled_message_keys: set[tuple[Any, Any]] = set()
|
||||||
|
|
||||||
|
def _mark_handled_once(self, message: Any) -> bool:
|
||||||
|
chat_id = getattr(getattr(message, "chat", None), "id", None)
|
||||||
|
message_id = getattr(message, "message_id", None)
|
||||||
|
key = (chat_id, message_id)
|
||||||
|
if key in self._handled_message_keys:
|
||||||
|
return False
|
||||||
|
self._handled_message_keys.add(key)
|
||||||
|
if len(self._handled_message_keys) > 4096:
|
||||||
|
self._handled_message_keys.clear()
|
||||||
|
return True
|
||||||
|
|
||||||
def register(self) -> None:
|
def register(self) -> None:
|
||||||
|
@self.bot.message_handler(commands=["deb"])
|
||||||
|
def _deb_command(message):
|
||||||
|
self.handle(message)
|
||||||
|
|
||||||
@self.bot.message_handler(
|
@self.bot.message_handler(
|
||||||
func=lambda message: _is_deb_command_text(getattr(message, "text", None)),
|
regexp=r"^[//⁄∕╱⧸]\s*deb(@[A-Za-z0-9_]+)?(\s|$)",
|
||||||
content_types=["text"],
|
content_types=["text"],
|
||||||
)
|
)
|
||||||
def _deb(message):
|
def _deb_regex(message):
|
||||||
self.handle(message)
|
self.handle(message)
|
||||||
|
|
||||||
def handle(self, message: Any) -> None:
|
def handle(self, message: Any) -> None:
|
||||||
|
if not self._mark_handled_once(message):
|
||||||
|
return
|
||||||
|
if not _is_deb_command_text(getattr(message, "text", None)):
|
||||||
|
return
|
||||||
trace = CommandTrace("/deb", message)
|
trace = CommandTrace("/deb", message)
|
||||||
try:
|
try:
|
||||||
parts = (message.text or "").split(maxsplit=1)
|
parts = (message.text or "").split(maxsplit=1)
|
||||||
|
|||||||
Reference in New Issue
Block a user