reykit 1.1.10__py3-none-any.whl → 1.1.12__py3-none-any.whl

reykit/rmultitask.py

@@ -10,8 +10,8 @@
 
 
 from __future__ import annotations
-from typing import Any, Literal
-from collections.abc import Callable, Iterable, Generator, Coroutine
+from typing import Any, Literal, Self
+from collections.abc import Callable, Iterable, Iterator, Generator, Coroutine, AsyncIterator
 from threading import RLock as TRLock, get_ident as threading_get_ident
 from concurrent.futures import ThreadPoolExecutor, Future as CFuture, as_completed as concurrent_as_completed
 from asyncio import (
@@ -36,7 +36,8 @@ __all__ = (
     'RThreadLock',
     'RAsyncLock',
     'RThreadPool',
-    'RAsyncPool'
+    'RAsyncPool',
+    'RAsyncIterator'
 )
 
 
@@ -868,3 +869,67 @@ class RAsyncPool(object):
 
 
     __mul__ = repeat
+
+
+class RAsyncIterator(AsyncIterator):
+    """
+    Rey's `asynchronous iterator` type.
+    """
+
+
+    def __init__(
+        self,
+        task: Callable,
+        args_iter: Iterable[Iterable] | None = None,
+        kwargs_iter: Iterable[dict] | None = None
+    ) -> None:
+        """
+        Build `asynchronous iterator` attributes.
+
+        Parameters
+        ----------
+        task : Asynchronous task.
+        args_iter : Iterable of task position arguments.
+        kwargs_iter : Iterable of task keyword arguments.
+        """
+
+        # Set attribute.
+        self.task = task
+        if args_iter is not None:
+            args_iter: Iterator[Iterable] = iter(args_iter)
+        self.args_iter = args_iter
+        if kwargs_iter is not None:
+            kwargs_iter: Iterator[dict] = iter(kwargs_iter)
+        self.kwargs_iter = kwargs_iter
+
+
+    def __aiter__(self) -> Self:
+        """
+        Get asynchronous iterator.
+        """
+
+        return self
+
+
+    async def __anext__(self):
+        """
+        Get next value from asynchronous iterator.
+        """
+
+        # Get parameter.
+        args = ()
+        kwargs = {}
+
+        ## Next.
+        try:
+            if self.args_iter is not None:
+                args = next(self.args_iter)
+            if self.kwargs_iter is not None:
+                kwargs = next(self.kwargs_iter)
+        except StopIteration:
+            raise StopAsyncIteration
+
+        # Execute.
+        result = self.task(*args, **kwargs)
+
+        return result

reykit/ros.py

@@ -10,7 +10,9 @@
 
 
 from __future__ import annotations
-from typing import Any, Literal, overload
+from typing import Any, Literal, TextIO, BinaryIO, overload, TYPE_CHECKING
+if TYPE_CHECKING:
+    from _typeshed import OpenTextMode, OpenBinaryMode
 from io import TextIOBase, BufferedIOBase
 from os import (
     walk as os_walk,
@@ -317,6 +319,84 @@ class RFile(object):
         self.path = os_abspath(path)
 
 
+    @overload
+    def open(
+        self,
+        mode: OpenTextMode = 'wb+'
+    ) -> TextIO: ...
+
+    @overload
+    def open(
+        self,
+        mode: OpenBinaryMode = 'wb+'
+    ) -> BinaryIO: ...
+
+    def open(
+        self,
+        mode: OpenTextMode | OpenBinaryMode = 'wb+'
+    ) -> TextIO | BinaryIO:
+        """
+        Open file.
+
+        Parameters
+        ----------
+        mode : Open mode.
+
+        Returns
+        -------
+        IO object.
+        """
+
+        # Get parameter.
+        if (
+            (
+                'r' in mode
+                or '+' in mode
+            )
+            and 'b' not in mode
+        ):
+            encoding = 'utf-8'
+        else:
+            encoding = None
+
+        # Open.
+        io = open(self.path, mode, encoding=encoding)
+
+        return io
+
+
+    @overload
+    def __getattr__(self, name: Literal['open_r', 'open_w', 'open_a']) -> TextIO: ...
+
+    @overload
+    def __getattr__(self, name: Literal['open_rb', 'open_wb', 'open_ab']) -> BinaryIO: ...
+
+    def __getattr__(self, name: Literal['open_r', 'open_w', 'open_a', 'open_rb', 'open_wb', 'open_ab']) -> TextIO | BinaryIO:
+        """
+        Get attribute.
+
+        Parameters
+        ----------
+        name : Open mode.
+
+        Returns
+        -------
+        IO object.
+        """
+
+        if name in (
+            'open_r',
+            'open_w',
+            'open_a',
+            'open_rb',
+            'open_wb',
+            'open_ab'
+        ):
+            mode: Literal['r', 'w', 'a', 'rb', 'wb', 'ab']  = name[5:]
+            io = self.open(mode)
+            return io
+
+
     @overload
     def read(
         self,
@@ -351,13 +431,11 @@ class RFile(object):
         match type_:
             case 'bytes':
                 mode = 'rb'
-                encoding = None
             case 'str':
                 mode = 'r'
-                encoding='utf-8'
 
         # Read.
-        with open(self.path, mode, encoding=encoding) as file:
+        with self.open(mode) as file:
             content = file.read()
 
         return content
@@ -398,7 +476,7 @@ class RFile(object):
                 data = str(data)
 
         # Write.
-        with open(self.path, mode) as file:
+        with self.open(mode) as file:
             file.write(data)
 
 

{reykit-1.1.10.dist-info → reykit-1.1.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: reykit
-Version: 1.1.10
+Version: 1.1.12
 Summary: Rey's kit method set.
 Project-URL: homepage, https://github.com/reyxbo/reykit/
 Author-email: Rey <reyxbo@163.com>

{reykit-1.1.10.dist-info → reykit-1.1.12.dist-info}/RECORD

@@ -7,9 +7,9 @@ reykit/rexception.py,sha256=ftXVdEVkU0B0HXbKjQTiJRJVVGbvhorwWtMiplpN9Zk,8118
 reykit/rimage.py,sha256=fQpIHX6Go3Jk_MDgsSDnZx27EZHumyGdgI9xyjP5lYQ,6275
 reykit/rlog.py,sha256=qhmATMv3_bJRaiuN9mo8hSf_95HH5Lw838aJFio98sg,25760
 reykit/rmonkey.py,sha256=RqhmKXabl11s2RJaGizpm00Q1yEkul1Je5uxw8_thUk,7584
-reykit/rmultitask.py,sha256=9UVAg8P85UsLS_exW-e6tZTa_b5r3ceL7LQRDrP8eAw,21981
+reykit/rmultitask.py,sha256=5oWApPnOrYrDDDZunTasAf8nmvrwBlKhS63cvSsi1iE,23599
 reykit/rnumber.py,sha256=XseLDNLDOt-XYP2P9oDbkKYuHudeSoWzIHsOENO8vvE,3196
-reykit/ros.py,sha256=UXX7FENgVCTR8m6-53CxzJvmPzHIzC7vLK2vyrkySiw,39096
+reykit/ros.py,sha256=6fHKCpxoWmCpxlRpGag5E-GzGsh0xt8YEmx9iqDvN_o,40780
 reykit/rrandom.py,sha256=TfQKYXRw379MMRIZYfjBtCkmsartA9thhDIG1fc3tM4,9233
 reykit/rregex.py,sha256=XTlnDLior8yyncFdrTr9FsVlBcqMXvsWRfpmvQS-BR8,6089
 reykit/rschedule.py,sha256=7EH_6TdEhwV-T6YyBEGYEcy85I1vTSNutDci-e_veTY,5796
@@ -24,7 +24,7 @@ reykit/rzip.py,sha256=i6KkmeSWCnq025d-O1mbuCYezNRUhyY9OGVK0CRlNAM,3522
 reykit/rdll/__init__.py,sha256=vM9V7wSNno-WH9RrxgHTIgCkQm8LmBFoLFO8z7qovNo,306
 reykit/rdll/rdll_inject.py,sha256=bETl8tywtN1OiQudbA21u6GwBM_bqVX7jbiisNj_JBg,645
 reykit/rdll/rdll_inject_core.py,sha256=Trgh_pdJs_Lw-Y-0Kkn8kHr4BnilM9dBKnHnX25T_pM,5092
-reykit-1.1.10.dist-info/METADATA,sha256=91hZfrvkD0Kf6t5cvqk4XNoGrk-ln1UpFHmacwcDMEk,1889
-reykit-1.1.10.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-reykit-1.1.10.dist-info/licenses/LICENSE,sha256=UYLPqp7BvPiH8yEZduJqmmyEl6hlM3lKrFIefiD4rvk,1059
-reykit-1.1.10.dist-info/RECORD,,
+reykit-1.1.12.dist-info/METADATA,sha256=5mIrvjJ0mmU7omM998GB5RHL566x2lfhXgIN3ZKuHSk,1889
+reykit-1.1.12.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+reykit-1.1.12.dist-info/licenses/LICENSE,sha256=UYLPqp7BvPiH8yEZduJqmmyEl6hlM3lKrFIefiD4rvk,1059
+reykit-1.1.12.dist-info/RECORD,,