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

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.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: reykit
-Version: 1.1.10
+Version: 1.1.11
 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.11.dist-info}/RECORD

@@ -9,7 +9,7 @@ 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/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.11.dist-info/METADATA,sha256=-bdW79_QpDoJ8moBzretuJd4QKiiQP9ML7Z5iXp1y2Q,1889
+reykit-1.1.11.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+reykit-1.1.11.dist-info/licenses/LICENSE,sha256=UYLPqp7BvPiH8yEZduJqmmyEl6hlM3lKrFIefiD4rvk,1059
+reykit-1.1.11.dist-info/RECORD,,