From 75592b0e20d3f63f1109d0f1526529b8de51834f Mon Sep 17 00:00:00 2001
From: Mmiirrkk <cot@wp.pl>
Date: Thu, 15 May 2025 08:59:10 +0200
Subject: [PATCH 1/2] add iterators and test_iterators 1

---
 serverish/base/iterators.py | 97 +++++++++++++++++++++++++++++++++++++
 tests/test_iterators.py     | 46 ++++++++++++++++++
 2 files changed, 143 insertions(+)
 create mode 100644 serverish/base/iterators.py
 create mode 100644 tests/test_iterators.py

diff --git a/serverish/base/iterators.py b/serverish/base/iterators.py
new file mode 100644
index 0000000..ccdd05c
--- /dev/null
+++ b/serverish/base/iterators.py
@@ -0,0 +1,97 @@
+import asyncio
+from typing import List, Dict, Any, Tuple, Set
+
+
+class AsyncRangeIter:
+    """This class represent async range iterator (instead of sync).
+
+    WARNING: Noticed that end is included (not like in standard iterator).
+    Example 1: start=1, end=4, will return [1, 2, 3, 4]
+    Example 2: async for n in AsyncRangeIter(start=1, end=3): ...
+    """
+
+    def __init__(self, start: int, end: int) -> None:
+        self.start = start
+        self.end = end
+
+    def __aiter__(self) -> Any:
+        self.current = self.start
+        return self
+
+    async def __anext__(self) -> int:
+        if self.current <= self.end:
+            value = self.current
+            self.current += 1
+            await asyncio.sleep(0)
+            return value
+        else:
+            raise StopAsyncIteration
+
+
+class AsyncListIter:
+    """This class represent async list iterator (instead of sync).
+
+    Example of use: async for n in AsyncListIter(some_list): ...
+    """
+    def __init__(self, iterable: List | Tuple | Set | Any):
+        self.iterable = iterable
+        self.index: int = 0
+
+    def __aiter__(self) -> Any:
+        self.index = 0
+        return self
+
+    async def __anext__(self):
+        if self.index < len(self.iterable):
+            value = self.iterable[self.index]
+            self.index += 1
+            await asyncio.sleep(0)
+            return value
+        else:
+            raise StopAsyncIteration
+
+
+class AsyncEnumerateIter:
+    """This class represent async enumerate iterator (instead of sync).
+
+    Example of use: async for current_index, value in AsyncEnumerateIter(some_iterable): ...
+    """
+
+    def __init__(self, iterable: List | Tuple | Set | Any) -> None:
+        self.iterable = iterable
+
+    def __aiter__(self):
+        self.index = 0
+        return self
+
+    async def __anext__(self):
+        if self.index < len(self.iterable):
+            value = self.iterable[self.index]
+            current_index = self.index
+            self.index += 1
+            await asyncio.sleep(0)
+            return current_index, value
+        else:
+            raise StopAsyncIteration
+
+
+class AsyncDictItemsIter:
+    """This class represent async dict items iterator (instead of sync).
+
+    Example of use: async for key, value in AsyncEnumerateIter(some_dict): ...
+    """
+
+    def __init__(self, data_dict: Dict) -> None:
+        self.data_dict = data_dict
+
+    def __aiter__(self) -> Any:
+        self.iterator = iter(self.data_dict.items())
+        return self
+
+    async def __anext__(self) -> Tuple:
+        try:
+            n, m = next(self.iterator)
+            await asyncio.sleep(0)
+            return n, m
+        except StopIteration:
+            raise StopAsyncIteration
diff --git a/tests/test_iterators.py b/tests/test_iterators.py
new file mode 100644
index 0000000..46ad5bd
--- /dev/null
+++ b/tests/test_iterators.py
@@ -0,0 +1,46 @@
+import unittest
+
+from serverish.base.iterators import AsyncRangeIter, AsyncListIter, AsyncDictItemsIter, AsyncEnumerateIter
+
+
+class TestIterAsync(unittest.IsolatedAsyncioTestCase):
+
+    async def test_async_range_iter(self):
+        target_list = [1, 2, 3, 4, 5]
+        new_list = []
+        async for n in AsyncRangeIter(1, 5):
+            new_list.append(n)
+        # print(target_list)
+        # print(new_list)
+        self.assertListEqual(target_list, new_list)
+
+    async def test_async_list_iter(self):
+        target_list = [1, 2, 3, 4, 5]
+        new_list = []
+        async for n in AsyncListIter(target_list):
+            new_list.append(n)
+        # print(target_list)
+        # print(new_list)
+        self.assertListEqual(target_list, new_list)
+
+    async def test_async_dict_items_iter(self):
+        target_dict = {'a': 2, 'b': 55}
+        new_dict = {}
+        async for n, m in AsyncDictItemsIter(target_dict):
+            new_dict[n] = m
+        # print(target_dict)
+        # print(new_dict)
+        self.assertDictEqual(target_dict, new_dict)
+
+    async def test_async_enumerate_items_iter(self):
+        target_dict = {0: 1, 1: 2, 2: 3}
+        new_dict = {}
+        async for n, m in AsyncEnumerateIter([m for n, m in target_dict.items()]):
+            new_dict[n] = m
+        # print(target_dict)
+        # print(new_dict)
+        self.assertDictEqual(target_dict, new_dict)
+
+
+if __name__ == '__main__':
+    unittest.main()

From 3e5900e06b5fbf0b1a1ce5c5061cd766c79a2f4a Mon Sep 17 00:00:00 2001
From: majkelx <mkalusz@camk.edu.pl>
Date: Thu, 15 May 2025 19:49:52 +0200
Subject: [PATCH 2/2] version bump on introducing iterators

---
 pyproject.toml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index 95ec996..d01a85c 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "serverish"
-version = "1.2.3"
+version = "1.3.0"
 description = "helpers for server alike projects"
 authors = ["Mikołaj Kałuszyński", "MMME team"]
 readme = "README.md"