reykit 1.1.0__py3-none-any.whl → 1.1.1__py3-none-any.whl

reydb/rparameter.py

@@ -0,0 +1,243 @@
+# !/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+"""
+@Time    : 2022-12-05 14:10:02
+@Author  : Rey
+@Contact : reyxbo@163.com
+@Explain : Database parameter methods.
+"""
+
+
+from typing import overload
+
+from .rconnection import RDatabase, RDBConnection
+
+
+__all__ = (
+    'RDBParameter',
+    'RDBPStatus',
+    'RDBPVariable'
+)
+
+
+class RDBParameter(object):
+    """
+    Rey's `database parameters` type.
+    """
+
+
+    def __init__(
+        self,
+        rdatabase: RDatabase | RDBConnection,
+        global_: bool
+    ) -> None:
+        """
+        Build `database parameters` attributes.
+
+        Parameters
+        ----------
+        rdatabase : RDatabase or RDBConnection instance.
+        global_ : Whether base global.
+        """
+
+        # Set parameter.
+        self.rdatabase = rdatabase
+        self.global_ = global_
+
+
+    def __getitem__(self, key: str) -> str | None:
+        """
+        Get item of parameter dictionary.
+
+        Parameters
+        ----------
+        key : Parameter key.
+
+        Returns
+        -------
+        Parameter value.
+        """
+
+        # Get.
+        value = self.get(key)
+
+        return value
+
+
+    def __setitem__(self, key: str, value: str | float) -> None:
+        """
+        Set item of parameter dictionary.
+
+        Parameters
+        ----------
+        key : Parameter key.
+        value : Parameter value.
+        """
+
+        # Set.
+        params = {key: value}
+
+        # Update.
+        self.update(params)
+
+
+class RDBPStatus(RDBParameter):
+    """
+    Rey's `database parameter status` type.
+    """
+
+
+    @overload
+    def get(self, key: None = None) -> dict[str, str]: ...
+
+    @overload
+    def get(self, key: str = None) -> str | None: ...
+
+    def get(self, key: str | None = None) -> dict[str, str] | str | None:
+        """
+        Get parameter.
+
+        Parameters
+        ----------
+        key : Parameter key.
+            - `None`: Return dictionary of all parameters.
+            - `str`: Return value of parameter.
+
+        Returns
+        -------
+        Status of database.
+        """
+
+        # Generate SQL.
+
+        ## Global.
+        if self.global_:
+            sql = 'SHOW GLOBAL STATUS'
+
+        ## Not global.
+        else:
+            sql = 'SHOW STATUS'
+
+        # Execute SQL.
+
+        ## Dictionary.
+        if key is None:
+            status = result.fetch_dict(val_field=1)
+
+        ## Value.
+        else:
+            sql += ' LIKE :key'
+            result = self.rdatabase(sql, key=key)
+            row = result.first()
+            if row is None:
+                status = None
+            else:
+                status = row['Value']
+
+        return status
+
+
+    def update(self, params: dict[str, str | float]) -> None:
+        """
+        Update parameter.
+
+        Parameters
+        ----------
+        params : Update parameter key value pairs.
+        """
+
+        # Throw exception.
+        raise AssertionError('database status not update')
+
+
+class RDBPVariable(RDBParameter):
+    """
+    Rey's `database parameter variable` type.
+    """
+
+
+    @overload
+    def get(self, key: None = None) -> dict[str, str]: ...
+
+    @overload
+    def get(self, key: str = None) -> str | None: ...
+
+    def get(self, key: str | None = None) -> dict[str, str] | str | None:
+        """
+        Get parameter.
+
+        Parameters
+        ----------
+        key : Parameter key.
+            - `None`: Return dictionary of all parameters.
+            - `str`: Return value of parameter.
+
+        Returns
+        -------
+        Variables of database.
+        """
+
+        # Generate SQL.
+
+        ## Global.
+        if self.global_:
+            sql = 'SHOW GLOBAL VARIABLES'
+
+        ## Not global.
+        else:
+            sql = 'SHOW VARIABLES'
+
+        # Execute SQL.
+
+        ## Dictionary.
+        if key is None:
+            variables = result.fetch_dict(val_field=1)
+
+        ## Value.
+        else:
+            sql += ' LIKE :key'
+            result = self.rdatabase(sql, key=key)
+            row = result.first()
+            if row is None:
+                variables = None
+            else:
+                variables = row['Value']
+
+        return variables
+
+
+
+    def update(self, params: dict[str, str | float]) -> None:
+        """
+        Update parameter.
+
+        Parameters
+        ----------
+        params : Update parameter key value pairs.
+        """
+
+        # Generate SQL.
+        sql_set_list = [
+            '%s = %s' % (
+                key,
+                (
+                    value
+                    if value.__class__ in (int, float)
+                    else "'%s'" % value
+                )
+            )
+            for key, value in params.items()
+        ]
+        sql_set = ',\n    '.join(sql_set_list)
+
+        # Global.
+        if self.global_:
+            sql = f'SET GLOBAL {sql_set}'
+
+        ## Not global.
+        else:
+            sql = f'SET {sql_set}'
+
+        # Execute SQL.
+        self.rdatabase(sql)

reykit/__init__.py

@@ -38,4 +38,4 @@ rzip : File compress methods.
 from typing import Final
 
 
-__version__: Final[str] = '1.1.0'
+__version__: Final[str] = '1.1.1'

reykit/rcomm.py

@@ -9,7 +9,7 @@
 """
 
 
-from typing import Any, Literal, Optional, Union
+from typing import Any, Literal
 from collections.abc import Callable, Iterable
 from warnings import filterwarnings
 from os.path import abspath as os_abspath, isfile as os_isfile
@@ -149,7 +149,7 @@ def split_cookie(cookie: str) -> dict[str, str]:
     return params
 
 
-def get_content_type(file: Union[str, bytes]) -> Optional[str]:
+def get_content_type(file: str | bytes) -> str | None:
     """
     Get HTTP content type of file.
 
@@ -184,17 +184,17 @@ def get_content_type(file: Union[str, bytes]) -> Optional[str]:
 
 def request(
     url: str,
-    params: Optional[dict] = None,
-    data: Optional[Union[dict, str, bytes]] = None,
-    json: Optional[dict] = None,
-    files: Optional[dict[str, Union[str, bytes, tuple[Union[str, bytes], dict]]]] = None,
+    params: dict | None = None,
+    data: dict | str | bytes | None = None,
+    json: dict | None = None,
+    files: dict[str, str | bytes | tuple[str | bytes, dict]] | None = None,
     headers: dict[str, str] = {},
-    timeout: Optional[float] = None,
+    timeout: float | None = None,
     proxies: dict[str, str] = {},
     stream: bool = False,
     verify: bool = False,
-    method: Optional[Literal['get', 'post', 'put', 'patch', 'delete', 'options', 'head']] = None,
-    check: Union[bool, int, Iterable[int]] = False
+    method: Literal['get', 'post', 'put', 'patch', 'delete', 'options', 'head'] | None = None,
+    check: bool | int | Iterable[int] = False
 ) -> Response:
     """
     Send request.
@@ -318,7 +318,7 @@ def request(
     return response
 
 
-def download(url: str, path: Optional[str] = None) -> str:
+def download(url: str, path: str | None = None) -> str:
     """
     Download file from URL.
 
@@ -363,7 +363,7 @@ def download(url: str, path: Optional[str] = None) -> str:
 
 
 def get_file_stream_time(
-    file: Union[str, bytes, int],
+    file: str | bytes | int,
     bandwidth: float
 ) -> float:
     """
@@ -402,7 +402,7 @@ def get_file_stream_time(
 
 def listen_socket(
     host: str,
-    port: Union[str, int],
+    port: str | int,
     handler: Callable[[bytes], Any]
 ) -> None:
     """

reykit/rdata.py

@@ -9,7 +9,7 @@
 """
 
 
-from typing import Any, TypedDict, Optional, Literal, NoReturn, TypeVar, overload
+from typing import Any, TypedDict, Literal, NoReturn, TypeVar, overload
 from collections.abc import Callable, Iterable, Generator
 
 from .rexception import check_least_one, check_most_one
@@ -87,7 +87,7 @@ def count(
     return result
 
 
-def flatten(data: Any, *, _flattern_data: Optional[list] = None) -> list:
+def flatten(data: Any, *, _flattern_data: list | None = None) -> list:
     """
     Flatten data.
 
@@ -136,9 +136,9 @@ def split(data: Iterable[Element], share: None = None, bin_size: None = None) ->
 def split(data: Iterable[Element], share: int = None, bin_size: int = None) -> NoReturn: ...
 
 @overload
-def split(data: Iterable[Element], share: Optional[int] = None, bin_size: Optional[int] = None) -> list[list[Element]]: ...
+def split(data: Iterable[Element], share: int | None = None, bin_size: int | None = None) -> list[list[Element]]: ...
 
-def split(data: Iterable[Element], share: Optional[int] = None, bin_size: Optional[int] = None) -> list[list[Element]]:
+def split(data: Iterable[Element], share: int | None = None, bin_size: int | None = None) -> list[list[Element]]:
     """
     Split data into multiple data.
 

reykit/remail.py

@@ -9,8 +9,6 @@
 """
 
 
-from typing import Optional, Union
-from io import BufferedIOBase
 from smtplib import SMTP
 from email.mime.multipart import MIMEMultipart
 from email.mime.text import MIMEText
@@ -101,12 +99,12 @@ class REmail(object):
 
     def create_email(
         self,
-        title: Optional[str],
-        text: Optional[str],
+        title: str | None,
+        text: str | None,
         attachment: dict[str, bytes],
-        show_from: Optional[str],
-        show_to: Optional[list[str]],
-        show_cc: Optional[list[str]]
+        show_from: str | None,
+        show_to: list[str] | None,
+        show_cc: list[str] | None
     ) -> str:
         """
         create email content.
@@ -169,14 +167,14 @@ class REmail(object):
 
     def send_email(
         self,
-        to: Union[str, list[str]],
-        title: Optional[str] = None,
-        text: Optional[str] = None,
+        to: str | list[str],
+        title: str | None = None,
+        text: str | None = None,
         attachment: dict[str, FileBytes] = {},
-        cc: Optional[Union[str, list[str]]] = None,
-        show_from: Optional[str] = None,
-        show_to: Optional[Union[str, list[str]]] = None,
-        show_cc: Optional[Union[str, list[str]]] = None
+        cc: str | list[str] | None = None,
+        show_from: str | None = None,
+        show_to: str | list[str] | None = None,
+        show_cc: str | list[str] | None = None
     ) -> None:
         """
         Send email.

reykit/rexception.py

@@ -20,7 +20,7 @@
 """
 
 
-from typing import Any, Optional, Union, NoReturn
+from typing import Any, NoReturn
 from types import TracebackType
 from collections.abc import Iterable
 from sys import exc_info as sys_exc_info
@@ -61,7 +61,7 @@ def throw(
     exception: type[BaseException] = AssertionError,
     value: Any = RNull,
     *values: Any,
-    text: Optional[str] = None,
+    text: str | None = None,
     frame: int = 2
 ) -> NoReturn:
     """
@@ -176,7 +176,7 @@ def warn(
 
 
 def catch_exc(
-    title: Optional[str] = None
+    title: str | None = None
 ) -> tuple[str, type[BaseException], BaseException, TracebackType]:
     """
     Catch exception information and print, must used in `except` syntax.
@@ -303,7 +303,7 @@ def check_file_exist(path: str) -> None:
 
 def check_response_code(
     code: int,
-    range_: Optional[Union[int, Iterable[int]]] = None
+    range_: int | Iterable[int] | None = None
 ) -> bool:
     """
     Check if the response code is in range.

reykit/rimage.py

@@ -9,7 +9,7 @@
 """
 
 
-from typing import Any, Union, Optional
+from typing import Any
 from io import BytesIO
 from qrcode import make as qrcode_make
 from qrcode.image.pil import PilImage
@@ -42,7 +42,7 @@ monkey_image_type = monkey_path_pil_image_get_bytes()
 RImage = monkey_image_type
 
 
-def encode_qrcode(text: str, path: Optional[str] = None) -> bytes:
+def encode_qrcode(text: str, path: str | None = None) -> bytes:
     """
     Encoding text to QR code image.
 
@@ -73,7 +73,7 @@ def encode_qrcode(text: str, path: Optional[str] = None) -> bytes:
     return file_bytes
 
 
-def decode_qrcode(image: Union[str, bytes]) -> list[str]:
+def decode_qrcode(image: str | bytes) -> list[str]:
     """
     Decoding QR code or bar code image.
 
@@ -108,14 +108,14 @@ def decode_qrcode(image: Union[str, bytes]) -> list[str]:
 
 
 def compress_image(
-    input_image: Union[str, bytes],
-    ouput_image: Optional[str] = None,
+    input_image: str | bytes,
+    ouput_image: str | None = None,
     target_size: float = 0.5,
     rate: int = 5,
     reduce: bool = False,
     max_quality: int = 75,
     min_quality: int = 0
-) -> Optional[bytes]:
+) -> bytes | None:
     """
     Compress image file.
 
@@ -187,7 +187,7 @@ def compress_image(
         rfile(content)
 
 
-def to_pimage(image: Union[str, bytes]) -> RImage:
+def to_pimage(image: str | bytes) -> RImage:
     """
     Get `Image` instance of `PIL` package.
 
@@ -215,8 +215,8 @@ def to_pimage(image: Union[str, bytes]) -> RImage:
 
 
 def generate_captcha_image(
-    text: Optional[Union[int, str]] = None,
-    path: Optional[str] = None,
+    text: int | str | None = None,
+    path: str | None = None,
     **kwargs: Any
 ) -> bytes:
     """

reykit/rlog.py

@@ -10,7 +10,7 @@
 
 
 from __future__ import annotations
-from typing import Any, Optional, Union, Literal, Final, NoReturn, overload, override
+from typing import Any, Literal, Final, NoReturn, overload, override
 from collections.abc import Callable
 from queue import Queue
 from os.path import abspath as os_abspath
@@ -291,7 +291,7 @@ class RLog(object):
     def get_default_filter_method(
         self,
         format_: str,
-        mode : Optional[Literal['print', 'file']] = None
+        mode : Literal['print', 'file'] | None = None
     ) -> Callable[[LogRecord], Literal[True]]:
         """
         Get default filter method of handler.
@@ -397,8 +397,8 @@ class RLog(object):
     def add_print(
         self,
         level: int = DEBUG,
-        format_: Optional[str] = None,
-        filter_: Optional[Callable[[LogRecord], bool]] = None
+        format_: str | None = None,
+        filter_: Callable[[LogRecord], bool] | None = None
     ) -> StreamHandler:
         """
         Add print output record handler.
@@ -443,56 +443,56 @@ class RLog(object):
     @overload
     def add_file(
         self,
-        path: Optional[str] = None,
-        mb: Optional[float] = None,
+        path: str | None = None,
+        mb: float | None = None,
         time: None = None,
         level: int = DEBUG,
-        format_: Optional[str] = None,
-        filter_: Optional[Callable[[LogRecord], bool]] = None
+        format_: str | None = None,
+        filter_: Callable[[LogRecord], bool] | None = None
     ) -> ConcurrentRotatingFileHandler: ...
 
     @overload
     def add_file(
         self,
-        path: Optional[str] = None,
+        path: str | None = None,
         mb: None = None,
-        time: Union[float, Literal['m', 'w0', 'w1', 'w2', 'w3', 'w4', 'w5', 'w6']] = None,
+        time: float | Literal['m', 'w0', 'w1', 'w2', 'w3', 'w4', 'w5', 'w6'] = None,
         level: int = DEBUG,
-        format_: Optional[str] = None,
-        filter_: Optional[Callable[[LogRecord], bool]] = None
+        format_: str | None = None,
+        filter_: Callable[[LogRecord], bool] | None = None
     ) -> ConcurrentTimedRotatingFileHandler: ...
 
     @overload
     def add_file(
         self,
-        path: Optional[str] = None,
+        path: str | None = None,
         mb: None = None,
         time: Any = None,
         level: int = DEBUG,
-        format_: Optional[str] = None,
-        filter_: Optional[Callable[[LogRecord], bool]] = None
+        format_: str | None = None,
+        filter_: Callable[[LogRecord], bool] | None = None
     ) -> NoReturn: ...
 
     @overload
     def add_file(
         self,
-        path: Optional[str] = None,
+        path: str | None = None,
         mb: float = None,
-        time: Union[float, Literal['m', 'w0', 'w1', 'w2', 'w3', 'w4', 'w5', 'w6']] = None,
+        time: float | Literal['m', 'w0', 'w1', 'w2', 'w3', 'w4', 'w5', 'w6'] = None,
         level: int = DEBUG,
-        format_: Optional[str] = None,
-        filter_: Optional[Callable[[LogRecord], bool]] = None
+        format_: str | None = None,
+        filter_: Callable[[LogRecord], bool] | None = None
     ) -> NoReturn: ...
 
     def add_file(
         self,
-        path: Optional[str] = None,
-        mb: Optional[float] = None,
-        time: Optional[Union[float, Literal['m', 'w0', 'w1', 'w2', 'w3', 'w4', 'w5', 'w6']]] = None,
+        path: str | None = None,
+        mb: float | None = None,
+        time: float | Literal['m', 'w0', 'w1', 'w2', 'w3', 'w4', 'w5', 'w6'] | None = None,
         level: int = DEBUG,
-        format_: Optional[str] = None,
-        filter_: Optional[Callable[[LogRecord], bool]] = None
-    ) -> Union[ConcurrentRotatingFileHandler, ConcurrentTimedRotatingFileHandler]:
+        format_: str | None = None,
+        filter_: Callable[[LogRecord], bool] | None = None
+    ) -> ConcurrentRotatingFileHandler | ConcurrentTimedRotatingFileHandler:
         """
         Add file output record handler, can split files based on size or time.
 
@@ -613,9 +613,9 @@ class RLog(object):
 
     def add_queue(
         self,
-        queue: Optional[Queue] = None,
+        queue: Queue | None = None,
         level: int = DEBUG,
-        filter_: Optional[Callable[[LogRecord], bool]] = None
+        filter_: Callable[[LogRecord], bool] | None = None
     ) -> tuple[QueueHandler, Queue[LogRecord]]:
         """
         Add queue output record handler.
@@ -655,7 +655,7 @@ class RLog(object):
         self,
         method: Callable[[LogRecord], Any],
         level: int = DEBUG,
-        filter_: Optional[Callable[[LogRecord], bool]] = None
+        filter_: Callable[[LogRecord], bool] | None = None
     ) -> None:
         """
         Add method record handler.
@@ -771,7 +771,7 @@ class RLog(object):
     def log(
         self,
         *messages: Any,
-        level: Optional[int] = None,
+        level: int | None = None,
         catch: bool = True,
         **params: Any
     ) -> None:
@@ -970,7 +970,7 @@ class RRecord(object):
 
     def __init__(
         self,
-        path: Optional[str] = '_rrecord'
+        path: str | None = '_rrecord'
     ) -> None:
         """
         Build `record` attributes.

reykit/rmonkey.py

@@ -29,7 +29,6 @@ def monkey_patch_sqlalchemy_result_more_fetch():
     """
 
 
-    from typing import Optional
     from sqlalchemy.engine.cursor import CursorResult
     from pandas import DataFrame, NA, concat
 
@@ -81,7 +80,7 @@ def monkey_patch_sqlalchemy_result_more_fetch():
 
 
     # Print result.
-    def method_show(self: RResult, limit: Optional[int] = None) -> None:
+    def method_show(self: RResult, limit: int | None = None) -> None:
         """
         Print result.
 
@@ -207,18 +206,18 @@ def monkey_patch_sqlalchemy_row_index_field():
     """
 
 
-    from typing import Any, Union, overload
+    from typing import Any, overload
     from sqlalchemy.engine.row import Row
 
 
     # Define.
     @overload
-    def __getitem__(self, index: Union[str, int]) -> Any: ...
+    def __getitem__(self, index: str | int) -> Any: ...
 
     @overload
     def __getitem__(self, index: slice) -> tuple: ...
 
-    def __getitem__(self, index: Union[str, int, slice]) -> Union[Any, tuple]:
+    def __getitem__(self, index: str | int | slice) -> Any | tuple:
         """
         Index row value.