reydb 1.1.35__py3-none-any.whl → 1.1.36__py3-none-any.whl

reydb/rconfig.py

@@ -9,8 +9,15 @@
 """
 
 
-from typing import Any
-from reykit.rbase import T, throw
+from typing import Any, TypedDict, TypeVar
+import datetime
+from datetime import (
+    datetime as Datetime,
+    date as Date,
+    time as Time,
+    timedelta as Timedelta
+)
+from reykit.rbase import throw
 
 from .rconn import DatabaseConnection
 from .rdb import Database
@@ -21,10 +28,24 @@ __all__ = (
 )
 
 
+type ConfigValue = bool | str | int | float | list | tuple | dict | set | Datetime | Date | Time | Timedelta | None
+ConfigRow = TypedDict('ConfigRow', {'key': str, 'value': ConfigValue, 'type': str, 'note': str | None})
+type ConfigTable = list[ConfigRow]
+ConfigValueT = TypeVar('T', bound=ConfigValue) # Any.
+
+
 class DatabaseConfig(object):
     """
     Database config type.
     Can create database used `self.build` method.
+
+    Examples
+    --------
+    >>> config = DatabaseConfig()
+    >>> config['key1'] = 1
+    >>> config['key2', 'note'] = 2
+    >>> config['key1'], config['key2']
+    (1, 2)
     """
 
 
@@ -90,16 +111,24 @@ class DatabaseConfig(object):
                     },
                     {
                         'name': 'value',
-                        'type': 'json',
+                        'type': 'text',
                         'constraint': 'NOT NULL',
                         'comment': 'Config value.'
                     },
+                    {
+                        'name': 'type',
+                        'type': 'varchar(50)',
+                        'constraint': 'NOT NULL',
+                        'comment': 'Config value type.'
+                    },
                     {
                         'name': 'note',
                         'type': 'varchar(500)',
                         'comment': 'Config note.'
                     }
-                ]
+                ],
+                'primary': 'key',
+                'comment': 'Config data table.'
             }
 
         ]
@@ -145,7 +174,38 @@ class DatabaseConfig(object):
         self.database.build.build(databases, tables, views_stats=views_stats)
 
 
-    def get(self, key: str, default: T | None = None) -> Any | T:
+    @property
+    def data(self) -> ConfigTable:
+        """
+        Get config data table.
+
+        Returns
+        -------
+        Config data table.
+        """
+
+        # Get.
+        result = self.database.execute_select(
+            (self.db_names['base'], self.db_names['base.config']),
+            ['key', 'value', 'type', 'note'],
+            order='IFNULL(`update_time`, `create_time`) DESC'
+        )
+
+        # Convert.
+        global_dict = {'datetime': datetime}
+        result = [
+            {
+                'key': row['key'],
+                'value': eval(row['value'], global_dict),
+                'note': row['note']
+            }
+            for row in result
+        ]
+
+        return result
+
+
+    def get(self, key: str, default: ConfigValueT | None = None) -> ConfigValue | ConfigValueT:
         """
         Get config value, when not exist, then return default value.
 
@@ -163,7 +223,7 @@ class DatabaseConfig(object):
         where = '`key` = :key'
         result = self.database.execute_select(
             (self.db_names['base'], self.db_names['base.config']),
-            'value',
+            '`value`',
             where,
             limit=1,
             key=key
@@ -173,11 +233,19 @@ class DatabaseConfig(object):
         # Default.
         if value is None:
             value = default
+        else:
+            global_dict = {'datetime': datetime}
+            value = eval(value, global_dict)
 
         return value
 
 
-    def setdefault(self, key: str, default: T | None = None) -> Any | T:
+    def setdefault(
+        self,
+        key: str,
+        default: ConfigValueT | None = None,
+        default_note: str | None = None
+    ) -> ConfigValue | ConfigValueT:
         """
         Set config value.
 
@@ -185,6 +253,7 @@ class DatabaseConfig(object):
         ----------
         key : Config key.
         default : Config default value.
+        default_note : Config default note.
 
         Returns
         -------
@@ -192,7 +261,12 @@ class DatabaseConfig(object):
         """
 
         # Set.
-        data = {'key': key, 'value': default}
+        data = {
+            'key': key,
+            'value': repr(default),
+            'type': type(default).__name__,
+            'note': default_note
+        }
         result = self.database.execute_insert(
             (self.db_names['base'], self.db_names['base.config']),
             data,
@@ -201,30 +275,39 @@ class DatabaseConfig(object):
 
         # Get.
         if result.rowcount == 0:
-            result = self[key]
-        else:
-            result = default
+            default = self.get(key)
 
-        return result
+        return default
 
 
-    def update(self, data: dict[str, Any]) -> None:
+    def update(self, data: dict[str, ConfigValue] | ConfigTable) -> None:
         """
         Update config values.
 
         Parameters
         ----------
         data : Config update data.
+            - `dict[str, Any]`: Config key and value.
+            - `ConfigTable`: Config key and value and note.
         """
 
+        # Handle parameter.
+        if type(data) == dict:
+            data = [
+                {
+                    'key': key,
+                    'value': repr(value),
+                    'type': type(value).__name__
+                }
+                for key, value in data.items()
+            ]
+        else:
+            data = data.copy()
+            for row in data:
+                row['value'] = repr(row['value'])
+                row['type'] = type(row['value']).__name__
+
         # Update.
-        data = [
-            {
-                'key': key,
-                'value': value
-            }
-            for key, value in data.items
-        ]
         self.database.execute_insert(
             (self.db_names['base'], self.db_names['base.config']),
             data,
@@ -232,21 +315,26 @@ class DatabaseConfig(object):
         )
 
 
-    def remove(self, key: str) -> None:
+    def remove(self, key: str | list[str]) -> None:
         """
         Remove config.
 
         Parameters
         ----------
-        key : Config key.
+        key : Config key or key list.
         """
 
         # Remove.
-        where = '`key` = :key'
+        if type(key) == str:
+            where = '`key` = :key'
+            limit = 1
+        else:
+            where = '`key` in :key'
+            limit = None
         result = self.database.execute_delete(
             (self.db_names['base'], self.db_names['base.config']),
             where,
-            limit=1,
+            limit=limit,
             key=key
         )
 
@@ -255,7 +343,7 @@ class DatabaseConfig(object):
             throw(KeyError, key)
 
 
-    def items(self) -> dict:
+    def items(self) -> dict[str, ConfigValue]:
         """
         Get all config keys and values.
 
@@ -271,7 +359,12 @@ class DatabaseConfig(object):
         )
 
         # Convert.
+        global_dict = {'datetime': datetime}
         result = result.to_dict('key', 'value')
+        result = {
+            key: eval(value, global_dict)
+            for key, value in result.items()
+        }
 
         return result
 
@@ -288,16 +381,20 @@ class DatabaseConfig(object):
         # Get.
         result = self.database.execute_select(
             (self.db_names['base'], self.db_names['base.config']),
-            'key'
+            '`key`'
         )
 
         # Convert.
-        result = result.to_list()
+        global_dict = {'datetime': datetime}
+        result = [
+            eval(value, global_dict)
+            for value in result
+        ]
 
         return result
 
 
-    def values(self) -> list[Any]:
+    def values(self) -> list[ConfigValue]:
         """
         Get all config value.
 
@@ -309,16 +406,20 @@ class DatabaseConfig(object):
         # Get.
         result = self.database.execute_select(
             (self.db_names['base'], self.db_names['base.config']),
-            'value'
+            '`value`'
         )
 
         # Convert.
-        result = result.to_list()
+        global_dict = {'datetime': datetime}
+        result = [
+            eval(value, global_dict)
+            for value in result
+        ]
 
         return result
 
 
-    def __getitem__(self, key: str) -> Any:
+    def __getitem__(self, key: str) -> ConfigValue:
         """
         Get config value.
 
@@ -341,18 +442,30 @@ class DatabaseConfig(object):
         return value
 
 
-    def __setitem__(self, key: str, value: Any) -> None:
+    def __setitem__(self, key_note: str | tuple[str, str], value: ConfigValue) -> None:
         """
         Set config value.
 
         Parameters
         ----------
-        key : Config key.
+        key_note : Config key and note.
         value : Config value.
         """
 
+        # Handle parameter.
+        if type(key_note) != str:
+            key, note = key_note
+        else:
+            key = key_note
+            note = None
+
         # Set.
-        data = {'key': key, 'value': value}
+        data = {
+            'key': key,
+            'value': repr(value),
+            'type': type(value).__name__,
+            'note': note
+        }
         self.database.execute_insert(
             (self.db_names['base'], self.db_names['base.config']),
             data,

{reydb-1.1.35.dist-info → reydb-1.1.36.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: reydb
-Version: 1.1.35
+Version: 1.1.36
 Summary: Database method set.
 Project-URL: homepage, https://github.com/reyxbo/reydb/
 Author-email: Rey <reyxbo@163.com>

{reydb-1.1.35.dist-info → reydb-1.1.36.dist-info}/RECORD

@@ -2,7 +2,7 @@ reydb/__init__.py,sha256=SqjJEBMiUMnKkNfmOvw_jprZcj9Edi0jyBKt67xeYUE,544
 reydb/rall.py,sha256=UWnbtl4oG4YqXyqTMN_5uqE-QqD5nb_-dvarotlTUeU,388
 reydb/rbase.py,sha256=X8bDBmKyIiLXGvOVnTh5nD0ltkhOPWU6g4BebifOZYY,312
 reydb/rbuild.py,sha256=s4lLVQONfL5a9B6mw5D7bfbPyJoYNIqBxlas9rCM1Y8,32458
-reydb/rconfig.py,sha256=d8xvpkoLbyVDVUNGPb0_TlrqxMdJsbjEXluXj7xWbN8,8801
+reydb/rconfig.py,sha256=x9NXZ_Z_LjEZH7xHm7mUFv_1Y93Ky50Vx72FRAXu08w,12303
 reydb/rconn.py,sha256=kdw2xQb0JqSUXM-BD7XfJ-ZqaWHxYkeT8psRhOK43mA,6211
 reydb/rdb.py,sha256=OQ3ztC4G53usAbocqGwTqo1ry1AIGjoWTN7pn8Pn3Uk,62184
 reydb/rerror.py,sha256=xHLvRxu8KBvUdfVS7mD906jx9XTQPGwEl7psBLGEZ5w,9966
@@ -10,7 +10,7 @@ reydb/rexec.py,sha256=dGdRkG1XR0Z66T0r4nPCSdQzSRWc_Q3t6TPSSrDTIxY,9042
 reydb/rfile.py,sha256=7g6hPBz33p-mkGFc6LEmL2hpFes-LM-AWQ0SxgJe2BI,15254
 reydb/rinfo.py,sha256=KXTkcpTGAD3p9RVKKcnmc_FjJtiKRPk-K5ZepPOnphQ,15253
 reydb/rparam.py,sha256=3BGDBD8QshOf2J70ZJ6LJ9PiH-1ZU3ruZwoE0bN6OOw,7017
-reydb-1.1.35.dist-info/METADATA,sha256=gXTzXVZRTCIRkoKawXAeaEpsF9f9QUaUgQ6Q3mUpQUs,1550
-reydb-1.1.35.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-reydb-1.1.35.dist-info/licenses/LICENSE,sha256=UYLPqp7BvPiH8yEZduJqmmyEl6hlM3lKrFIefiD4rvk,1059
-reydb-1.1.35.dist-info/RECORD,,
+reydb-1.1.36.dist-info/METADATA,sha256=At9eApGRgD86nYSVG-ZWBL8ndc55uPRwc4K0Ex5c7gs,1550
+reydb-1.1.36.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+reydb-1.1.36.dist-info/licenses/LICENSE,sha256=UYLPqp7BvPiH8yEZduJqmmyEl6hlM3lKrFIefiD4rvk,1059
+reydb-1.1.36.dist-info/RECORD,,