Tiny improvements (#187)

prostomarkeloff · Jun 11, 2023 · 862655a · 862655a
1 parent d88311a
commit 862655a
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/vkwave/bots/addons/easy/clones_bot.py b/vkwave/bots/addons/easy/clones_bot.py
@@ -1,5 +1,6 @@
 import asyncio
 import typing
+from collections.abc import Iterator
 
 from .easy_userbot import SimpleLongPollUserBot
 from .easy_bot import SimpleLongPollBot
@@ -20,6 +21,12 @@ def __init__(
         self.router = self.base_bot.router
         self.clones: typing.List[typing.Union[SimpleLongPollUserBot, SimpleLongPollBot]] = list(clones)
 
+    def __iter__(self) -> Iterator[typing.Union[SimpleLongPollUserBot, SimpleLongPollBot]]:
+        return iter(self.clones)
+
+    def __len__(self) -> int:
+        return len(self.clones)
+
     def add_clone(self, clone: typing.Union[SimpleLongPollUserBot, SimpleLongPollBot]) -> None:
         self.clones.append(clone)