queutils 0.8.5__py3-none-any.whl → 0.9.1__py3-none-any.whl

queutils/__init__.py

@@ -2,10 +2,15 @@ from .countable import Countable as Countable
 from .asyncqueue import AsyncQueue as AsyncQueue
 from .iterablequeue import IterableQueue as IterableQueue, QueueDone as QueueDone
 from .filequeue import FileQueue as FileQueue
+from .categorycounterqueue import (
+    QCounter as QCounter,
+    CategoryCounterQueue as CategoryCounterQueue,
+)
 
 __all__ = [
     "asyncqueue",
     "countable",
+    "categorycounterqueue",
     "filequeue",
     "iterablequeue",
 ]

queutils/categorycounterqueue.py

@@ -0,0 +1,116 @@
+from asyncio import Queue
+from typing import TypeVar
+from deprecated import deprecated
+from .countable import Countable
+from .iterablequeue import IterableQueue, QueueDone
+from collections import defaultdict
+import logging
+
+logger = logging.getLogger()
+error = logger.error
+message = logger.warning
+verbose = logger.info
+debug = logger.debug
+
+###########################################
+#
+# class CounterQueue
+#
+###########################################
+T = TypeVar("T")
+
+
+@deprecated(version="0.9.1", reason="Use CategoryCounterQueue instead")
+class CounterQueue(Queue[T], Countable):
+    """
+    CounterQueue is a asyncio.Queue for counting items
+    """
+
+    _counter: int
+    _count_items: bool
+    _batch: int
+
+    def __init__(
+        self, *args, count_items: bool = True, batch: int = 1, **kwargs
+    ) -> None:
+        super().__init__(*args, **kwargs)
+        self._counter = 0
+        self._count_items = count_items
+        self._batch = batch
+
+    def task_done(self) -> None:
+        super().task_done()
+        if self._count_items:
+            self._counter += 1
+        return None
+
+    @property
+    def count(self) -> int:
+        """Return number of completed tasks"""
+        return self._counter * self._batch
+
+    @property
+    def count_items(self) -> bool:
+        """Whether or not count items"""
+        return self._count_items
+
+
+class CategoryCounterQueue(IterableQueue[tuple[str, int]]):
+    """
+    CategorySummerQueue is a asyncio.Queue for summing up values by category
+    """
+
+    _counter: defaultdict[str, int]
+    _batch: int
+
+    def __init__(self, *args, batch: int = 1, **kwargs) -> None:
+        super().__init__(*args, **kwargs)
+        self._batch = batch
+        self._counter = defaultdict(int)
+
+    async def receive(self) -> tuple[str, int]:
+        """Receive a category value from the queue and sum it"""
+        category: str
+        value: int
+        category, value = await super().get()
+        self._counter[category] += value
+        super().task_done()
+        return (category, value)
+
+    async def send(self, category: str = "count", value: int = 1) -> None:
+        """Send count of a category"""
+        await super().put((category, value))
+        return None
+
+    def get_count(self, category: str = "count") -> int:
+        """Return count of a category"""
+        return self._counter[category]
+
+    def get_counts(self) -> defaultdict[str, int]:
+        """Return counts of all categories"""
+        return self._counter
+
+    async def listen(self) -> defaultdict[str, int]:
+        """Listen for category values"""
+        try:
+            while True:
+                await self.receive()
+        except QueueDone:
+            pass
+        return self.get_counts()
+
+
+class QCounter:
+    def __init__(self, Q: Queue[int]):
+        self._count = 0
+        self._Q: Queue[int] = Q
+
+    @property
+    def count(self) -> int:
+        return self._count
+
+    async def start(self) -> None:
+        """Read and count items from Q"""
+        while True:
+            self._count += await self._Q.get()
+            self._Q.task_done()

{queutils-0.8.5.dist-info → queutils-0.9.1.dist-info}/METADATA

@@ -1,10 +1,11 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: queutils
-Version: 0.8.5
+Version: 0.9.1
 Summary: Handy Python Queue utilies
 Project-URL: Homepage, https://github.com/Jylpah/queutils
 Project-URL: Bug Tracker, https://github.com/Jylpah/queutils/issues
 Author-email: Jylpah <jylpah@gmail.com>
+License-File: LICENSE
 Classifier: Development Status :: 4 - Beta
 Classifier: Framework :: AsyncIO
 Classifier: License :: OSI Approved :: MIT License
@@ -13,6 +14,7 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Topic :: Software Development :: Libraries
 Requires-Python: >=3.11
 Requires-Dist: aioconsole>=0.6
+Requires-Dist: deprecated>=1.2.18
 Provides-Extra: dev
 Requires-Dist: build>=0.10; extra == 'dev'
 Requires-Dist: hatchling>=1.22.4; extra == 'dev'
@@ -24,6 +26,7 @@ Requires-Dist: pytest-datafiles>=3.0; extra == 'dev'
 Requires-Dist: pytest-timeout>=2.2; extra == 'dev'
 Requires-Dist: pytest>=8.0; extra == 'dev'
 Requires-Dist: ruff>=0.1.9; extra == 'dev'
+Requires-Dist: types-deprecated>=1.2.15; extra == 'dev'
 Description-Content-Type: text/markdown
 
 [![Python package](https://github.com/Jylpah/queutils/actions/workflows/python-package.yml/badge.svg)](https://github.com/Jylpah/queutils/actions/workflows/python-package.yml)  [![codecov](https://codecov.io/gh/Jylpah/queutils/graph/badge.svg?token=rMKdbfHOFs)](https://codecov.io/gh/Jylpah/queutils)

queutils-0.9.1.dist-info/RECORD

@@ -0,0 +1,11 @@
+queutils/__init__.py,sha256=buANyRlxO1x4G_EBE3tYFpPtZGSepJDKRkWl-zKct1M,453
+queutils/asyncqueue.py,sha256=GZRmlWTBQoKzTf7xr4MI-qhNqvIiaNWswAxFokP91Lg,2789
+queutils/categorycounterqueue.py,sha256=Ku5pAJXSsoNyUsuWkQe9w7SaPeiPwPHFMdZFqA0LtBw,3097
+queutils/countable.py,sha256=YSi7ILf9CuB5Tm3T4UUMEFlveqzqcmomfqJAlLGHEz8,172
+queutils/filequeue.py,sha256=q2ly9H-lSCq6xuOqT1IlWgyCVyLoZiKbc0NuzmkF4aw,5360
+queutils/iterablequeue.py,sha256=ZXwa9040yTawL1DMMNBXgQKiYr32nmzuSbgi-raAtZc,8664
+queutils/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+queutils-0.9.1.dist-info/METADATA,sha256=zT3br-kHUXjDGp2QhPxW_K-EvP212VUYWZNbjTb-MUA,4281
+queutils-0.9.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+queutils-0.9.1.dist-info/licenses/LICENSE,sha256=J1zeIKU2JVQmhwO2hHQDK8WR6zjVZ-wX8r7ZlL45AbI,1063
+queutils-0.9.1.dist-info/RECORD,,

{queutils-0.8.5.dist-info → queutils-0.9.1.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: hatchling 1.26.3
+Generator: hatchling 1.27.0
 Root-Is-Purelib: true
 Tag: py3-none-any

queutils-0.8.5.dist-info/RECORD

@@ -1,10 +0,0 @@
-queutils/__init__.py,sha256=MboAom80iQTQURz0liGv_f2ae2Np8YA5ORX1LwsNa50,311
-queutils/asyncqueue.py,sha256=GZRmlWTBQoKzTf7xr4MI-qhNqvIiaNWswAxFokP91Lg,2789
-queutils/countable.py,sha256=YSi7ILf9CuB5Tm3T4UUMEFlveqzqcmomfqJAlLGHEz8,172
-queutils/filequeue.py,sha256=q2ly9H-lSCq6xuOqT1IlWgyCVyLoZiKbc0NuzmkF4aw,5360
-queutils/iterablequeue.py,sha256=ZXwa9040yTawL1DMMNBXgQKiYr32nmzuSbgi-raAtZc,8664
-queutils/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-queutils-0.8.5.dist-info/METADATA,sha256=6sJ7O4naraaat7-8ZFeTD9c_r1RFM9uDtkotLuX6aq0,4169
-queutils-0.8.5.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-queutils-0.8.5.dist-info/licenses/LICENSE,sha256=J1zeIKU2JVQmhwO2hHQDK8WR6zjVZ-wX8r7ZlL45AbI,1063
-queutils-0.8.5.dist-info/RECORD,,