telegrinder 0.2.0.post2__py3-none-any.whl → 0.2.1__py3-none-any.whl

telegrinder/__init__.py

@@ -79,6 +79,7 @@ from .bot import (
     WaiterMachine,
     register_manager,
 )
+from .bot.rules import StateMeta
 from .client import ABCClient, AiohttpClient
 from .model import Model
 from .modules import logger
@@ -86,6 +87,7 @@ from .tools import (
     ABCErrorHandler,
     ABCGlobalContext,
     ABCLoopWrapper,
+    ABCStateStorage,
     ABCTranslator,
     ABCTranslatorMiddleware,
     AnyMarkup,
@@ -104,10 +106,12 @@ from .tools import (
     KeyboardSetYAML,
     Lifespan,
     LoopWrapper,
+    MemoryStateStorage,
     ParseMode,
     RowButtons,
     SimpleI18n,
     SimpleTranslator,
+    StateData,
     ctx_var,
     magic_bundle,
 )
@@ -207,4 +211,8 @@ __all__ = (
     "logger",
     "magic_bundle",
     "register_manager",
+    "ABCStateStorage",
+    "MemoryStateStorage",
+    "StateData",
+    "StateMeta",
 )

telegrinder/bot/dispatch/waiter_machine/machine.py

@@ -45,13 +45,13 @@ class WaiterMachine:
 
     async def drop(
         self,
-        state_view: "ABCStateView[EventModel]",
+        state_view: "ABCStateView[EventModel] | str",
         id: Identificator,
         event: EventModel,
         update: Update,
         **context: typing.Any,
     ) -> None:
-        view_name = state_view.__class__.__name__
+        view_name = state_view if isinstance(state_view, str) else state_view.__class__.__name__
         if view_name not in self.storage:
             raise LookupError("No record of view {!r} found.".format(view_name))
 
@@ -61,13 +61,26 @@ class WaiterMachine:
 
         short_state.cancel()
         await self.call_behaviour(
-            state_view,
             event,
             update,
             behaviour=short_state.on_drop_behaviour,
             **context,
         )
 
+    async def drop_all(self) -> None:
+        """Drops all waiters in storage"""
+        for view_name in self.storage:
+            for ident, short_state in self.storage[view_name].items():
+                if short_state.context:
+                    await self.drop(
+                        view_name,
+                        ident,
+                        short_state.context.event,
+                        short_state.context.context.raw_update
+                    )
+                else:
+                    short_state.cancel()
+
     async def wait(
         self,
         state_view: "BaseStateView[EventModel]",
@@ -114,7 +127,6 @@ class WaiterMachine:
 
     async def call_behaviour(
         self,
-        view: "ABCStateView[EventModel]",
         event: EventModel,
         update: Update,
         behaviour: Behaviour[EventModel] | None = None,
@@ -147,7 +159,7 @@ class WaiterMachine:
             now = datetime.datetime.now()
             for ident, short_state in self.storage.get(view_name, {}).copy().items():
                 if short_state.expiration_date is not None and now > short_state.expiration_date:
-                    assert short_state.context
+                    assert short_state.context  # FIXME: why???
                     await self.drop(
                         view,
                         ident,

telegrinder/bot/dispatch/waiter_machine/middleware.py

@@ -59,7 +59,6 @@ class WaiterMiddleware(ABCMiddleware[EventType]):
 
         # before running the handler we check if the user wants to exit waiting
         if short_state.exit_behaviour is not None and await self.machine.call_behaviour(
-            self.view,
             event,
             ctx.raw_update,
             behaviour=short_state.exit_behaviour,
@@ -87,7 +86,6 @@ class WaiterMiddleware(ABCMiddleware[EventType]):
 
         elif short_state.default_behaviour is not None:
             await self.machine.call_behaviour(
-                self.view,
                 event,
                 ctx.raw_update,
                 behaviour=short_state.default_behaviour,

telegrinder/bot/rules/__init__.py

@@ -53,6 +53,7 @@ from telegrinder.bot.rules.node import NodeRule
 from telegrinder.bot.rules.regex import Regex
 from telegrinder.bot.rules.rule_enum import RuleEnum
 from telegrinder.bot.rules.start import StartCommand
+from telegrinder.bot.rules.state import State, StateMeta
 from telegrinder.bot.rules.text import HasText, Text
 from telegrinder.bot.rules.update import IsUpdateType
 
@@ -113,4 +114,6 @@ __all__ = (
     "StartCommand",
     "Text",
     "NodeRule",
+    "State",
+    "StateMeta",
 )

telegrinder/bot/rules/state.py

@@ -0,0 +1,34 @@
+import enum
+import typing
+
+from telegrinder.bot.dispatch.context import Context
+from telegrinder.bot.rules.abc import ABCRule
+from telegrinder.node import Source
+
+if typing.TYPE_CHECKING:
+    from telegrinder.tools.state_storage import ABCStateStorage
+
+
+class StateMeta(enum.Enum):
+    NO_STATE = enum.auto()
+    ANY = enum.auto()
+
+
+class State(ABCRule):
+    def __init__(self, storage: "ABCStateStorage", key: str | StateMeta = StateMeta.ANY):
+        self.storage = storage
+        self.key = key
+
+    async def check(self, source: Source, ctx: Context) -> bool:
+        state = await self.storage.get(source.from_user.id)
+        if not state:
+            return self.key == StateMeta.NO_STATE
+
+        if self.key != StateMeta.ANY and self.key != state.unwrap().key:
+            return False
+
+        ctx["state"] = state.unwrap()
+        return True
+
+
+__all__ = ("State", "StateMeta")

telegrinder/tools/__init__.py

@@ -67,6 +67,7 @@ from .limited_dict import LimitedDict
 from .loop_wrapper import ABCLoopWrapper, DelayedTask, Lifespan, LoopWrapper
 from .magic import impl, magic_bundle, resolve_arg_names
 from .parse_mode import ParseMode
+from .state_storage import ABCStateStorage, MemoryStateStorage, StateData
 
 __all__ = (
     "ABCErrorHandler",
@@ -139,4 +140,7 @@ __all__ = (
     "block_quote",
     "impl",
     "resolve_arg_names",
+    "ABCStateStorage",
+    "MemoryStateStorage",
+    "StateData",
 )

telegrinder/tools/keyboard.py

@@ -26,11 +26,6 @@ class KeyboardModel:
     keyboard: list[list[DictStrAny]]
 
 
-
-def copy_keyboard(keyboard: list[list[DictStrAny]]) -> list[list[DictStrAny]]:
-    return [row.copy() for row in keyboard]
-
-
 class ABCMarkup(ABC, typing.Generic[ButtonT]):
     BUTTON: type[ButtonT]
     keyboard: list[list[DictStrAny]]
@@ -78,7 +73,7 @@ class ABCMarkup(ABC, typing.Generic[ButtonT]):
         return copy_keyboard
 
     def merge(self, other: typing.Self) -> typing.Self:
-        self.keyboard.extend(copy_keyboard(other.keyboard))
+        self.keyboard.extend(other.keyboard)
         return self
 
 

telegrinder/tools/state_storage/__init__.py

@@ -0,0 +1,4 @@
+from .abc import ABCStateStorage, StateData
+from .memory import MemoryStateStorage
+
+__all__ = ("ABCStateStorage", "StateData", "MemoryStateStorage")

telegrinder/tools/state_storage/abc.py

@@ -0,0 +1,33 @@
+import abc
+import typing
+from dataclasses import dataclass
+
+from fntypes import Option
+
+from telegrinder.bot.rules.state import State, StateMeta
+
+Payload = typing.TypeVar("Payload")
+
+
+@dataclass
+class StateData(typing.Generic[Payload]):
+    key: str
+    payload: Payload
+
+
+class ABCStateStorage(abc.ABC, typing.Generic[Payload]):
+    @abc.abstractmethod
+    async def get(self, user_id: int) -> Option[StateData[Payload]]:
+        ...
+
+    @abc.abstractmethod
+    async def delete(self, user_id: int) -> None:
+        ...
+
+    @abc.abstractmethod
+    async def set(self, user_id: int, key: str, payload: Payload) -> None:
+        ...
+
+    def State(self, key: str | StateMeta = StateMeta.ANY) -> State:  # noqa: N802
+        """Can be used as a shortcut to get a state rule dependant on current storage"""
+        return State(storage=self, key=key)

telegrinder/tools/state_storage/memory.py

@@ -0,0 +1,22 @@
+import typing
+
+from fntypes import Nothing, Option, Some
+
+from .abc import ABCStateStorage, StateData
+
+Payload = dict[str, typing.Any]
+
+
+class MemoryStateStorage(ABCStateStorage[Payload]):
+    def __init__(self):
+        self.storage: dict[int, StateData[Payload]] = {}
+
+    async def get(self, user_id: int) -> Option[StateData[Payload]]:
+        state = self.storage.get(user_id)
+        return Some(state) if state is not None else Nothing()
+
+    async def set(self, user_id: int, key: str, payload: Payload) -> None:
+        self.storage[user_id] = StateData(key, payload)
+
+    async def delete(self, user_id: int) -> None:
+        self.storage.pop(user_id)

telegrinder/types/methods.py

@@ -4178,7 +4178,7 @@ class APIMethods:
         :param description: Product description, 1-255 characters.
 
         :param payload: Bot-defined invoice payload, 1-128 bytes. This will not be displayed to \
-        the user, use for your internal processes.
+        the user, use it for your internal processes.
 
         :param provider_token: Payment provider token, obtained via @BotFather. Pass an empty string \
         for payments in Telegram Stars.
@@ -4296,7 +4296,7 @@ class APIMethods:
         :param description: Product description, 1-255 characters.
 
         :param payload: Bot-defined invoice payload, 1-128 bytes. This will not be displayed to \
-        the user, use for your internal processes.
+        the user, use it for your internal processes.
 
         :param provider_token: Payment provider token, obtained via @BotFather. Pass an empty string \
         for payments in Telegram Stars.

telegrinder/types/objects.py

@@ -2795,6 +2795,15 @@ class ChatInviteLink(Model):
     pending_join_request_count: Option[int] = Nothing
     """Optional. Number of pending join requests created using this link."""
 
+    subscription_period: Option[int] = Nothing
+    """Optional. The number of seconds the subscription will be active for before
+    the next payment."""
+
+    subscription_price: Option[int] = Nothing
+    """Optional. The amount of Telegram Stars a user must pay initially and after
+    each subsequent subscription period to be a member of the chat using the
+    link."""
+
 
 class ChatAdministratorRights(Model):
     """Object `ChatAdministratorRights`, see the [documentation](https://core.telegram.org/bots/api#chatadministratorrights).
@@ -5393,7 +5402,7 @@ class InputInvoiceMessageContent(InputMessageContent):
 
     payload: str
     """Bot-defined invoice payload, 1-128 bytes. This will not be displayed to
-    the user, use for your internal processes."""
+    the user, use it for your internal processes."""
 
     currency: Currency
     """Three-letter ISO 4217 currency code, see more on currencies. Pass `XTR`

{telegrinder-0.2.0.post2.dist-info → telegrinder-0.2.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: telegrinder
-Version: 0.2.0.post2
+Version: 0.2.1
 Summary: Modern visionary telegram bot framework.
 Home-page: https://github.com/timoniq/telegrinder
 License: MIT

{telegrinder-0.2.0.post2.dist-info → telegrinder-0.2.1.dist-info}/RECORD

@@ -1,4 +1,4 @@
-telegrinder/__init__.py,sha256=6w6pzPVw-MOX4zBcNLxEeWfX5yCAbrltcjILhs0yA1I,4219
+telegrinder/__init__.py,sha256=vngxup29t8Hv4mnFFI9MZHKO33i0i3ny8gDa8KzgqQ8,4395
 telegrinder/api/__init__.py,sha256=AFZ07UvdL4rhq9lZpB-j9JuGOONmfkvt8RfdG71ND38,226
 telegrinder/api/api.py,sha256=OSwm2_mgHS9jCsKy_vdOJzVtH-j6v68ypkdtvKFlQ3w,2877
 telegrinder/api/error.py,sha256=6_KBR819Tg9mLI7w5aHHYnrS8VSDYNkWIrHCnfgCFKk,422
@@ -41,13 +41,13 @@ telegrinder/bot/dispatch/view/inline_query.py,sha256=wgel-euw601uk5qHIUZF7vyuhP5
 telegrinder/bot/dispatch/view/message.py,sha256=FzfL8Yn42exnvMj8VuKwNlKaHPhij5eBzLCrju7FEc0,1356
 telegrinder/bot/dispatch/view/raw.py,sha256=XMlJ0-XeuUfFUjSaI7ApR-gcGmSZ1eB_USHsCKdeMyY,3536
 telegrinder/bot/dispatch/waiter_machine/__init__.py,sha256=IeoNZIxAut4QZCqiQrA6g_BcSFyUdA-oFgfBhsG6ipA,363
-telegrinder/bot/dispatch/waiter_machine/machine.py,sha256=bMMYSwQPZbu58GYk364xu10_G8pLv0OAbvQRz69RhOs,5954
-telegrinder/bot/dispatch/waiter_machine/middleware.py,sha256=f4IJgjMwACOetmAIo6mey5WYOmO3Sw3OG3pga5ptsCc,3528
+telegrinder/bot/dispatch/waiter_machine/machine.py,sha256=DhRtWrVasYsQtB-VkcxVGNbU46HftDLxMZxh1cBH9qU,6492
+telegrinder/bot/dispatch/waiter_machine/middleware.py,sha256=1qQYT77foCSjtcxbMkiay4rPCGNPJJWaQOPEF353830,3478
 telegrinder/bot/dispatch/waiter_machine/short_state.py,sha256=V1NZUwM_pjO-NVvN1kOnx4yo6Mdt5SXMmQnSY5xp21w,2116
 telegrinder/bot/polling/__init__.py,sha256=OqfIFPS_V6UrCg-vCv9pkMFzTKdNbDP2faBfATs_TGg,94
 telegrinder/bot/polling/abc.py,sha256=qFiKzWTWENK-sSuShC5cPlM-JS4In2c8-1_ARdwdTms,442
 telegrinder/bot/polling/polling.py,sha256=w6_d6hOWY2ER5UG6EgGl2ceLvgJNDjeK13sUUmxWcfc,4714
-telegrinder/bot/rules/__init__.py,sha256=0uKfqa0myqjIBcwzfq5YxoRn0aZFR4IlQ3joFROIFEA,2772
+telegrinder/bot/rules/__init__.py,sha256=ue5Lt9BoaUcwIJcwBItQR80_DdNZ2MznNmki2MIJyNs,2859
 telegrinder/bot/rules/abc.py,sha256=-y5dlftZZaYVPuvJsEmgYcM9nd8MDj3mdlN4Et1Vje8,6122
 telegrinder/bot/rules/adapter/__init__.py,sha256=kWcGnK0N26WsOwtF2n_yy1M2YMOtOp75931-ssVtOps,445
 telegrinder/bot/rules/adapter/abc.py,sha256=lPyieSDDr1z98rZ4W6Ic8QzHDzCEU9xIYrPKU9qDCas,682
@@ -72,6 +72,7 @@ telegrinder/bot/rules/node.py,sha256=IP7Ke-4hILb12XHNIresMf_87kXhlLfKOloShoQvWN0
 telegrinder/bot/rules/regex.py,sha256=mjXXPX-NB2e3gFU6LFihLnKANcgUoAL5lOnfmhC3Qnc,1153
 telegrinder/bot/rules/rule_enum.py,sha256=35GwPKLBTG_ESn4TZLcFJTLNjYXAi_d9wrfIDexoEH8,2113
 telegrinder/bot/rules/start.py,sha256=3ciA28m9JNcM-1H2YnLrUNyS1Y4UsyYPR5MClrpdAdA,1154
+telegrinder/bot/rules/state.py,sha256=WxVsNRGVCJkzyLt48wtYXFkbXfFB2a7WpToGK3Pc0Uk,890
 telegrinder/bot/rules/text.py,sha256=h8SBiqF3_kkITDRCeDDanmNG80NRoSgj_dutjoV-26Y,964
 telegrinder/bot/rules/update.py,sha256=k73pZzlzhw77gHAmeIsqehy0-EI2ec8wVGxGRIF_8ao,398
 telegrinder/bot/scenario/__init__.py,sha256=nnPjdxdvjoEYYMRUEfWvIhZStiY1C984x1azdRRP9II,136
@@ -105,7 +106,7 @@ telegrinder/node/tools/generator.py,sha256=xuhZ5-TwoctFLlDBdYvjOQxHTpqC4IAdnkzMV
 telegrinder/node/update.py,sha256=2baHe10IJ0f6oOeQIqBXduRio9PGVd5Q2lbGyfU9yU8,441
 telegrinder/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 telegrinder/rules.py,sha256=qQHT63Yek8I2zZFbfH2SzvQLUdZCcWBz-DFlz05i39I,1115
-telegrinder/tools/__init__.py,sha256=N_EGjedn096mPw1Tozc4uMCSXH2h9NCFnKuubFZu5ec,2837
+telegrinder/tools/__init__.py,sha256=xFx3DfmBoHPp2xmjaxpYmSDjR1CFlmWIHLxd9-9hpCY,2977
 telegrinder/tools/buttons.py,sha256=4Mc8KZ8i8l2XSiJaybbO23ywIx8T3bQiMH0UkyXuj4I,2905
 telegrinder/tools/error_handler/__init__.py,sha256=WmYWZCNhhSk32j4lIOltEwzoYUx086TGTbOF5h3Ps7s,207
 telegrinder/tools/error_handler/abc.py,sha256=BpBtysvMZKxAYAIrQHYXNxP9mCZRzjtshFFwzIXvVi8,882
@@ -127,19 +128,22 @@ telegrinder/tools/i18n/simple.py,sha256=yiaGtEAaMkrzYV3O7esA2wi3A2n9YyndxKPKp8Fs
 telegrinder/tools/kb_set/__init__.py,sha256=k1KCQTnvEgJ2y4KlghhJWOh5rccwg_27cs8264NtMmk,156
 telegrinder/tools/kb_set/base.py,sha256=mbZs-ViUErfSibzyN064IqZp76LBJPg3NB4D9v4VFtg,243
 telegrinder/tools/kb_set/yaml.py,sha256=glk27r0L9Y8ibaPmTHMAEJZw-ED-ehmSo_NdJNKkrNs,2007
-telegrinder/tools/keyboard.py,sha256=bEC7NqLcuYIxH6CPBH8oykXD5Ud-CuTpl035FlK5SEc,3825
+telegrinder/tools/keyboard.py,sha256=3zaBOMWMV5rPQg6BsLI7S_6_Nhyl2yKosFCbQAk9GzM,3684
 telegrinder/tools/limited_dict.py,sha256=tb1WT4P3Oia5CsCBXTHzlFjrPnIgf9eHCkQ0zhAbEUg,1078
 telegrinder/tools/loop_wrapper/__init__.py,sha256=ZQ5jmE1lOKnqJlMZ9k2OYmjvOEhOlHPijUWqZ4nHIgk,165
 telegrinder/tools/loop_wrapper/abc.py,sha256=ET_Dp-kRz75Jo1fZB3qofUgEXN4FqlU0xH2diESKGCM,266
 telegrinder/tools/loop_wrapper/loop_wrapper.py,sha256=k5NaZ-18c8zuzBHpHNk03_33dHFNww8HiCJEGz2g5W0,6815
 telegrinder/tools/magic.py,sha256=oPaEqqtbFEVXHVCjdbmUvbjtWiQCw2yuMOfoQLLyFjE,4332
 telegrinder/tools/parse_mode.py,sha256=JyQ-x9YAMPLhIIiUX01acyKkpWgs5TBA07W-iUyPHpE,92
+telegrinder/tools/state_storage/__init__.py,sha256=wQHybNOqF1O7ri_UUQSvtKgLuesQqa2wxETaEgMyMSU,149
+telegrinder/tools/state_storage/abc.py,sha256=NAoEepnNGVv-SMoRrV-upKLQEIpJWtHIxZFbKWjnDAs,850
+telegrinder/tools/state_storage/memory.py,sha256=KNEyAqjcCeZhk2YbLnoQbio9LU9s9VZzRnT7A5eEMMU,662
 telegrinder/types/__init__.py,sha256=_7ANa63K4rMRWHgSBSQJXi2mNMtOJgp_E4yGattsgJE,6612
 telegrinder/types/enums.py,sha256=q9URlXZvrjOUQKpLfV6v9uspBcLrdW0gtU-m8YDnj7w,19017
-telegrinder/types/methods.py,sha256=Ah7DVfyb99EJzYHCX9feW9hIGBYcnHLKEzu5MUp4rvg,201058
-telegrinder/types/objects.py,sha256=6bN1BsYZ-Kmh1GcXgtumfZvvLJqxjjjJ9OzyfqBtGZM,244504
+telegrinder/types/methods.py,sha256=cZQzITIQbEVLLZCkscl8jMqDoS5tIqXKkIFyWJoi2-Q,201064
+telegrinder/types/objects.py,sha256=0UQ60e8CoqqpRjLSG-9RqMaSro3qUF_LxMJXXhpW8Ng,244881
 telegrinder/verification_utils.py,sha256=X7N0mHoOzbcYeKa5XxI_EFhmEGX5XNU3qqgbV8YRRa4,987
-telegrinder-0.2.0.post2.dist-info/LICENSE,sha256=Q0tKgU8mPOCQAkc6m__BrNIpRge8mPBQJDd59s21NZo,1095
-telegrinder-0.2.0.post2.dist-info/METADATA,sha256=bKp_kZ28oIvDegWbLKZQw1o8imRU0LyO85H19q1XAOg,3008
-telegrinder-0.2.0.post2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-telegrinder-0.2.0.post2.dist-info/RECORD,,
+telegrinder-0.2.1.dist-info/LICENSE,sha256=Q0tKgU8mPOCQAkc6m__BrNIpRge8mPBQJDd59s21NZo,1095
+telegrinder-0.2.1.dist-info/METADATA,sha256=o0lMt6wBzckvY4GaO7azBXsIjTXEuSwoDP1eaT2LRXQ,3002
+telegrinder-0.2.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+telegrinder-0.2.1.dist-info/RECORD,,