Merge branch 'release/3.4.0'

Author: wolph

.github/workflows/main.yml

@@ -14,9 +14,9 @@ jobs:
         python-version: ['3.7', '3.8', '3.9', '3.10', '3.11-dev']
 
     steps:
-      - uses: actions/checkout@v2
+      - uses: actions/checkout@v3
       - name: Set up Python ${{ matrix.python-version }}
-        uses: actions/setup-python@v2
+        uses: actions/setup-python@v3
         with:
           python-version: ${{ matrix.python-version }}
       - name: Install dependencies
@@ -32,19 +32,25 @@ jobs:
       - name: pytest
         run: py.test
 
-  docs:
+  docs_and_lint:
     runs-on: ubuntu-latest
     timeout-minutes: 2
     steps:
-    - uses: actions/checkout@v2
+    - uses: actions/checkout@v3
     - name: Set up Python
-      uses: actions/setup-python@v2
+      uses: actions/setup-python@v3
       with:
         python-version: '3.10'
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip setuptools
-        pip install -e '.[docs]'
+        pip install -e '.[docs,tests]' pyright flake8 mypy
     - name: build docs
       run: make html
       working-directory: docs/
+    - name: flake8
+      run: flake8 -v python_utils setup.py
+    - name: mypy
+      run: mypy python_utils setup.py
+    - name: pyright
+      run: pyright

_python_utils_tests/test_time.py

@@ -71,6 +71,9 @@ def test_timeout_generator(
 
 @pytest.mark.asyncio
 async def test_aio_generator_timeout_detector():
+    # Make pyright happy
+    i = None
+
     async def generator():
         for i in range(10):
             await asyncio.sleep(i / 100.0)
@@ -106,52 +109,55 @@ async def generator():
 
 @pytest.mark.asyncio
 async def test_aio_generator_timeout_detector_decorator():
+    # Make pyright happy
+    i = None
+
     # Test regular timeout with reraise
     @python_utils.aio_generator_timeout_detector_decorator(timeout=0.05)
-    async def generator():
+    async def generator_timeout():
         for i in range(10):
             await asyncio.sleep(i / 100.0)
             yield i
 
     with pytest.raises(asyncio.TimeoutError):
-        async for i in generator():
+        async for i in generator_timeout():
             pass
 
     # Test regular timeout with clean exit
     @python_utils.aio_generator_timeout_detector_decorator(
         timeout=0.05, on_timeout=None
     )
-    async def generator():
+    async def generator_clean():
         for i in range(10):
             await asyncio.sleep(i / 100.0)
             yield i
 
-    async for i in generator():
+    async for i in generator_clean():
         pass
 
     assert i == 4
 
     # Test total timeout with reraise
     @python_utils.aio_generator_timeout_detector_decorator(total_timeout=0.1)
-    async def generator():
+    async def generator_reraise():
         for i in range(10):
             await asyncio.sleep(i / 100.0)
             yield i
 
     with pytest.raises(asyncio.TimeoutError):
-        async for i in generator():
+        async for i in generator_reraise():
             pass
 
     # Test total timeout with clean exit
     @python_utils.aio_generator_timeout_detector_decorator(
         total_timeout=0.1, on_timeout=None
     )
-    async def generator():
+    async def generator_clean_total():
         for i in range(10):
             await asyncio.sleep(i / 100.0)
             yield i
 
-    async for i in generator():
+    async for i in generator_clean_total():
         pass
 
     assert i == 4

pyrightconfig.json

@@ -0,0 +1,8 @@
+{
+  "include": [
+    "python_utils"
+  ],
+  "exclude": [
+    "python_utils/types.py",
+  ],
+}

python_utils/__about__.py

@@ -7,4 +7,4 @@
 )
 __url__: str = 'https://github.com/WoLpH/python-utils'
 # Omit type info due to automatic versioning script
-__version__ = '3.3.3'
+__version__ = '3.4.0'

python_utils/aio.py

@@ -5,8 +5,15 @@
 import asyncio
 import itertools
 
+from . import types
 
-async def acount(start=0, step=1, delay=0, stop=None):
+
+async def acount(
+    start: float = 0,
+    step: float = 1,
+    delay: float = 0,
+    stop: types.Optional[float] = None,
+) -> types.AsyncIterator[float]:
     '''Asyncio version of itertools.count()'''
     for item in itertools.count(start, step):  # pragma: no branch
         if stop is not None and item >= stop:

python_utils/containers.py

@@ -31,8 +31,8 @@ def __init__(
         self,
         key_cast: KT_cast = None,
         value_cast: VT_cast = None,
-        *args,
-        **kwargs,
+        *args: DictUpdateArgs,
+        **kwargs: VT,
     ) -> None:
         self._value_cast = value_cast
         self._key_cast = key_cast
@@ -53,7 +53,7 @@ def __setitem__(self, key: Any, value: Any) -> None:
         return super().__setitem__(key, value)
 
 
-class CastedDict(CastedDictBase):
+class CastedDict(CastedDictBase[KT, VT]):
     '''
     Custom dictionary that casts keys and values to the specified typing.
 
@@ -88,14 +88,14 @@ class CastedDict(CastedDictBase):
     {1: 2, '3': '4', '5': '6', '7': '8'}
     '''
 
-    def __setitem__(self, key, value):
+    def __setitem__(self, key: Any, value: Any) -> None:
         if self._value_cast is not None:
             value = self._value_cast(value)
 
         super().__setitem__(key, value)
 
 
-class LazyCastedDict(CastedDictBase):
+class LazyCastedDict(CastedDictBase[KT, VT]):
     '''
     Custom dictionary that casts keys and lazily casts values to the specified
     typing. Note that the values are cast only when they are accessed and
@@ -141,13 +141,13 @@ class LazyCastedDict(CastedDictBase):
     '4'
     '''
 
-    def __setitem__(self, key, value):
+    def __setitem__(self, key: Any, value: Any) -> None:
         if self._key_cast is not None:
             key = self._key_cast(key)
 
         super().__setitem__(key, value)
 
-    def __getitem__(self, key) -> VT:
+    def __getitem__(self, key: Any) -> VT:
         if self._key_cast is not None:
             key = self._key_cast(key)
 

python_utils/converters.py

@@ -10,7 +10,7 @@ def to_int(
     input_: typing.Optional[str] = None,
     default: int = 0,
     exception: types.ExceptionsType = (ValueError, TypeError),
-    regexp: types.O[types.Pattern] = None,
+    regexp: types.Optional[types.Pattern] = None,
 ) -> int:
     r'''
     Convert the given input to an integer or return default
@@ -100,7 +100,7 @@ def to_float(
     input_: str,
     default: int = 0,
     exception: types.ExceptionsType = (ValueError, TypeError),
-    regexp: types.O[types.Pattern] = None,
+    regexp: types.Optional[types.Pattern] = None,
 ) -> types.Number:
     r'''
     Convert the given `input_` to an integer or return default

python_utils/generators.py

@@ -9,7 +9,7 @@ async def abatcher(
     generator: types.AsyncIterator,
     batch_size: types.Optional[int] = None,
     interval: types.Optional[types.delta_type] = None,
-):
+) -> types.AsyncIterator[list]:
     '''
     Asyncio generator wrapper that returns items with a given batch size or
     interval (whichever is reached first).
@@ -64,7 +64,9 @@ async def abatcher(
             next_yield = time.perf_counter() + interval_s
 
 
-def batcher(iterable, batch_size):
+def batcher(
+    iterable: types.Iterable, batch_size: int = 10
+) -> types.Iterator[list]:
     '''
     Generator wrapper that returns items with a given batch size
     '''

python_utils/loguru.py

@@ -1,15 +1,15 @@
 from __future__ import annotations
 
-from . import logger
-
 import loguru
 
+from . import logger as logger_module
+
 __all__ = ['Logurud']
 
 
-class Logurud(logger.LoggerBase):
+class Logurud(logger_module.LoggerBase):
     logger: loguru.Logger
 
     def __new__(cls, *args, **kwargs):
-        cls.logger: loguru.Loguru = loguru.logger.opt(depth=1)
+        cls.logger: loguru.Logger = loguru.logger.opt(depth=1)
         return super().__new__(cls)