reydb 1.1.59__py3-none-any.whl → 1.1.60__py3-none-any.whl

reydb/rconfig.py

@@ -49,109 +49,54 @@ class DatabaseConfig(DatabaseBase):
     """
 
 
-    def __init__(self, db: 'rdb.Database') -> None:
+    def __init__(
+        self,
+        db: 'rdb.Database',
+        db_names: dict[str, str] | None = None
+    ) -> None:
         """
         Build instance attributes.
 
         Parameters
         ----------
         db: Database instance.
+        db_names: Update build database table names.
         """
 
         # Build.
         self.db = db
-
-        ## Database path name.
         self.db_names = {
-            'base': 'base',
-            'base.config': 'config',
-            'base.stats_config': 'stats_config'
+            'config': 'config',
+            'stats_config': 'stats_config'
         }
+        if db_names is not None:
+            self.db_names.update(db_names)
 
 
     def build_db(self) -> None:
         """
-        Check and build all standard databases and tables, by `self.db_names`.
+        Check and build database tables, by `self.db_names`.
         """
 
-        # Set parameter.
-
-        ## Database.
-        databases = [
-            {
-                'name': self.db_names['base']
-            }
-        ]
+        # Handle parameter.
 
         ## Table.
-        tables = [
-
-            ### 'config'.
-            {
-                'path': (self.db_names['base'], self.db_names['base.config']),
-                'fields': [
-                    {
-                        'name': 'create_time',
-                        'type': 'datetime',
-                        'constraint': 'NOT NULL DEFAULT CURRENT_TIMESTAMP',
-                        'comment': 'Config create time.'
-                    },
-                    {
-                        'name': 'update_time',
-                        'type': 'datetime',
-                        'constraint': 'DEFAULT NULL ON UPDATE CURRENT_TIMESTAMP',
-                        'comment': 'Config update time.'
-                    },
-                    {
-                        'name': 'key',
-                        'type': 'varchar(50)',
-                        'constraint': 'NOT NULL',
-                        'comment': 'Config key.'
-                    },
-                    {
-                        'name': 'value',
-                        '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',
-                'indexes': [
-                    {
-                        'name': 'n_create_time',
-                        'fields': 'create_time',
-                        'type': 'noraml',
-                        'comment': 'Config create time normal index.'
-                    },
-                    {
-                        'name': 'n_update_time',
-                        'fields': 'update_time',
-                        'type': 'noraml',
-                        'comment': 'Config update time normal index.'
-                    }
-                ],
-                'comment': 'Config data table.'
-            }
-
-        ]
+        class Config(self.db.orm.Model, table=True):
+            __name__ = self.db_names['config']
+            __commment__ = 'Config data table.'
+            create_time: datetime = self.db.orm.Field(not_null=True, field_default='CURRENT_TIMESTAMP', index_n=True, comment='Config create time.')
+            update_time: datetime = self.db.orm.Field(field_default='CURRENT_TIMESTAMP', index_n=True, comment='Config update time.')
+            key = self.db.orm.Field(type=self.db.orm.types.VARCHAR(50), key=True, not_null=True, comment='Config key.')
+            value = self.db.orm.Field(type=self.db.orm.types.TEXT, not_null=True, comment='Config value.')
+            type = self.db.orm.Field(type=self.db.orm.types.VARCHAR(50), not_null=True, comment='Config value type.')
+            note = self.db.orm.Field(type=self.db.orm.types.VARCHAR(500), comment='Config note.')
+
+        tables = [Config]
 
         ## View stats.
         views_stats = [
-
-            ### 'stats_config'.
             {
-                'path': (self.db_names['base'], self.db_names['base.stats_config']),
+                'path': self.db_names['stats_config'],
                 'items': [
                     {
                         'name': 'count',
@@ -178,13 +123,12 @@ class DatabaseConfig(DatabaseBase):
                         'comment': 'Config last record update time.'
                     }
                 ]
-
             }
 
         ]
 
         # Build.
-        self.db.build.build(databases, tables, views_stats=views_stats)
+        self.db.build.build(tables=tables, views_stats=views_stats)
 
 
     @property
@@ -199,7 +143,7 @@ class DatabaseConfig(DatabaseBase):
 
         # Get.
         result = self.db.execute.select(
-            (self.db_names['base'], self.db_names['base.config']),
+            self.db_names['config'],
             ['key', 'value', 'type', 'note'],
             order='IFNULL(`update_time`, `create_time`) DESC'
         )
@@ -235,7 +179,7 @@ class DatabaseConfig(DatabaseBase):
         # Get.
         where = '`key` = :key'
         result = self.db.execute.select(
-            (self.db_names['base'], self.db_names['base.config']),
+            self.db_names['config'],
             '`value`',
             where,
             limit=1,
@@ -281,7 +225,7 @@ class DatabaseConfig(DatabaseBase):
             'note': default_note
         }
         result = self.db.execute.insert(
-            (self.db_names['base'], self.db_names['base.config']),
+            self.db_names['config'],
             data,
             'ignore'
         )
@@ -322,7 +266,7 @@ class DatabaseConfig(DatabaseBase):
 
         # Update.
         self.db.execute.insert(
-            (self.db_names['base'], self.db_names['base.config']),
+            self.db_names['config'],
             data,
             'update'
         )
@@ -345,7 +289,7 @@ class DatabaseConfig(DatabaseBase):
             where = '`key` in :key'
             limit = None
         result = self.db.execute.delete(
-            (self.db_names['base'], self.db_names['base.config']),
+            self.db_names['base.config'],
             where,
             limit=limit,
             key=key
@@ -367,7 +311,7 @@ class DatabaseConfig(DatabaseBase):
 
         # Get.
         result = self.db.execute.select(
-            (self.db_names['base'], self.db_names['base.config']),
+            self.db_names['config'],
             ['key', 'value']
         )
 
@@ -393,7 +337,7 @@ class DatabaseConfig(DatabaseBase):
 
         # Get.
         result = self.db.execute.select(
-            (self.db_names['base'], self.db_names['base.config']),
+            self.db_names['config'],
             '`key`'
         )
 
@@ -418,7 +362,7 @@ class DatabaseConfig(DatabaseBase):
 
         # Get.
         result = self.db.execute.select(
-            (self.db_names['base'], self.db_names['base.config']),
+            self.db_names['config'],
             '`value`'
         )
 
@@ -480,7 +424,7 @@ class DatabaseConfig(DatabaseBase):
             'note': note
         }
         self.db.execute.insert(
-            (self.db_names['base'], self.db_names['base.config']),
+            self.db_names['config'],
             data,
             'update'
         )

reydb/rconn.py

@@ -9,12 +9,12 @@
 """
 
 
-from typing import Self, Generic
+from typing import Self, TypeVar, Generic
 from sqlalchemy import Connection, Transaction
 from sqlalchemy.ext.asyncio import AsyncConnection, AsyncTransaction
 
 from . import rdb, rexec
-from .rbase import DatabaseT, DatabaseExecuteT, ConnectionT, TransactionT, DatabaseBase
+from .rbase import ConnectionT, TransactionT, DatabaseBase
 
 
 __all__ = (
@@ -24,6 +24,10 @@ __all__ = (
 )
 
 
+DatabaseT = TypeVar('DatabaseT', 'rdb.Database', 'rdb.DatabaseAsync')
+DatabaseExecuteT = TypeVar('DatabaseExecuteT', 'rexec.DatabaseExecute', 'rexec.DatabaseExecuteAsync')
+
+
 class DatabaseConnectionSuper(DatabaseBase, Generic[DatabaseT, DatabaseExecuteT, ConnectionT, TransactionT]):
     """
     Database connection super type.
@@ -53,8 +57,8 @@ class DatabaseConnectionSuper(DatabaseBase, Generic[DatabaseT, DatabaseExecuteT,
             case rdb.DatabaseAsync():
                 exec = rexec.DatabaseExecuteAsync(self)
         self.execute: DatabaseExecuteT = exec
-        self.conn: ConnectionT | None = None
-        self.begin: TransactionT | None = None
+        self.connection: ConnectionT | None = None
+        self.transaction: TransactionT | None = None
 
 
 class DatabaseConnection(DatabaseConnectionSuper['rdb.Database', 'rexec.DatabaseExecute', Connection, Transaction]):
@@ -106,10 +110,10 @@ class DatabaseConnection(DatabaseConnectionSuper['rdb.Database', 'rexec.Database
         """
 
         # Create.
-        if self.conn is None:
-            self.conn = self.db.engine.connect()
+        if self.connection is None:
+            self.connection = self.db.engine.connect()
 
-        return self.conn
+        return self.connection
 
 
     def get_begin(self) -> Transaction:
@@ -122,11 +126,11 @@ class DatabaseConnection(DatabaseConnectionSuper['rdb.Database', 'rexec.Database
         """
 
         # Create.
-        if self.begin is None:
+        if self.transaction is None:
             conn = self.get_conn()
-            self.begin = conn.begin()
+            self.transaction = conn.begin()
 
-        return self.begin
+        return self.transaction
 
 
     def commit(self) -> None:
@@ -135,9 +139,9 @@ class DatabaseConnection(DatabaseConnectionSuper['rdb.Database', 'rexec.Database
         """
 
         # Commit.
-        if self.begin is not None:
-            self.begin.commit()
-            self.begin = None
+        if self.transaction is not None:
+            self.transaction.commit()
+            self.transaction = None
 
 
     def rollback(self) -> None:
@@ -146,9 +150,9 @@ class DatabaseConnection(DatabaseConnectionSuper['rdb.Database', 'rexec.Database
         """
 
         # Rollback.
-        if self.begin is not None:
-            self.begin.rollback()
-            self.begin = None
+        if self.transaction is not None:
+            self.transaction.rollback()
+            self.transaction = None
 
 
     def close(self) -> None:
@@ -157,12 +161,12 @@ class DatabaseConnection(DatabaseConnectionSuper['rdb.Database', 'rexec.Database
         """
 
         # Close.
-        if self.begin is not None:
-            self.begin.close()
-            self.begin = None
-        if self.conn is not None:
-            self.conn.close()
-            self.conn = None
+        if self.transaction is not None:
+            self.transaction.close()
+            self.transaction = None
+        if self.connection is not None:
+            self.connection.close()
+            self.connection = None
 
 
     def insert_id(self) -> int:
@@ -232,10 +236,10 @@ class DatabaseConnectionAsync(DatabaseConnectionSuper['rdb.DatabaseAsync', 'rexe
         """
 
         # Create.
-        if self.conn is None:
-            self.conn = await self.db.engine.connect()
+        if self.connection is None:
+            self.connection = await self.db.engine.connect()
 
-        return self.conn
+        return self.connection
 
 
     async def get_begin(self) -> AsyncTransaction:
@@ -248,11 +252,11 @@ class DatabaseConnectionAsync(DatabaseConnectionSuper['rdb.DatabaseAsync', 'rexe
         """
 
         # Create.
-        if self.begin is None:
+        if self.transaction is None:
             conn = await self.get_conn()
-            self.begin = await conn.begin()
+            self.transaction = await conn.begin()
 
-        return self.begin
+        return self.transaction
 
 
     async def commit(self) -> None:
@@ -261,9 +265,9 @@ class DatabaseConnectionAsync(DatabaseConnectionSuper['rdb.DatabaseAsync', 'rexe
         """
 
         # Commit.
-        if self.begin is not None:
-            await self.begin.commit()
-            self.begin = None
+        if self.transaction is not None:
+            await self.transaction.commit()
+            self.transaction = None
 
 
     async def rollback(self) -> None:
@@ -272,9 +276,9 @@ class DatabaseConnectionAsync(DatabaseConnectionSuper['rdb.DatabaseAsync', 'rexe
         """
 
         # Rollback.
-        if self.begin is not None:
-            await self.begin.rollback()
-            self.begin = None
+        if self.transaction is not None:
+            await self.transaction.rollback()
+            self.transaction = None
 
 
     async def close(self) -> None:
@@ -283,12 +287,12 @@ class DatabaseConnectionAsync(DatabaseConnectionSuper['rdb.DatabaseAsync', 'rexe
         """
 
         # Close.
-        if self.begin is not None:
-            await self.begin.close()
-            self.begin = None
-        if self.conn is not None:
-            await self.conn.close()
-            self.conn = None
+        if self.transaction is not None:
+            await self.transaction.close()
+            self.transaction = None
+        if self.connection is not None:
+            await self.connection.close()
+            self.connection = None
 
 
     async def insert_id(self) -> int:

reydb/rdb.py

@@ -26,14 +26,22 @@ __all__ = (
 )
 
 
+DatabaseConnectionT = TypeVar('DatabaseConnectionT', 'rconn.DatabaseConnection', 'rconn.DatabaseConnectionAsync')
+DatabaseExecuteT = TypeVar('DatabaseExecuteT', 'rexec.DatabaseExecute', 'rexec.DatabaseExecuteAsync')
+DatabaseSchemaT = TypeVar('DatabaseSchemaT', 'rparam.DatabaseSchema', 'rparam.DatabaseSchemaAsync')
+DatabaseORMT = TypeVar('DatabaseORMT', 'rorm.DatabaseORM', 'rorm.DatabaseORMAsync')
+DatabaseBuildT = TypeVar('DatabaseBuildT')
+
+
 class DatabaseSuper(
     rbase.DatabaseBase,
     Generic[
         rbase.EngineT,
-        rbase.DatabaseConnectionT,
-        rbase.DatabaseExecuteT,
-        rbase.DatabaseSchemaT,
-        rbase.DatabaseORMT
+        DatabaseConnectionT,
+        DatabaseExecuteT,
+        DatabaseSchemaT,
+        DatabaseORMT,
+        DatabaseBuildT
     ]
 ):
     """
@@ -71,7 +79,7 @@ class DatabaseSuper(
         pool_recycle : Number of seconds `recycle` connection.
             - `None | Literal[-1]`: No recycle.
             - `int`: Use this value.
-        report : Whether report SQL execute information.
+        report : Whether report SQL execute information, not include ORM execute.
         query : Remote server database parameters.
         """
 
@@ -238,7 +246,7 @@ class DatabaseSuper(
         return keep_n, overflow_n
 
 
-    def connect(self, autocommit: bool = False) -> rbase.DatabaseConnectionT:
+    def connect(self, autocommit: bool = False) -> DatabaseConnectionT:
         """
         Build database connection instance.
 
@@ -262,7 +270,7 @@ class DatabaseSuper(
 
 
     @property
-    def execute(self) -> rbase.DatabaseExecuteT:
+    def execute(self) -> DatabaseExecuteT:
         """
         Build database execute instance.
 
@@ -279,7 +287,7 @@ class DatabaseSuper(
 
 
     @property
-    def orm(self) -> 'rbase.DatabaseORMT':
+    def orm(self) -> DatabaseORMT:
         """
         Build database ORM instance.
 
@@ -299,7 +307,7 @@ class DatabaseSuper(
 
 
     @property
-    def build(self):
+    def build(self) -> DatabaseBuildT:
         """
         Build database build instance.
 
@@ -309,9 +317,13 @@ class DatabaseSuper(
         """
 
         # Build.
-        dbbuild = rbuild.DatabaseBuild(self)
+        match self:
+            case Database():
+                build = rbuild.DatabaseBuild(self)
+            case DatabaseAsync():
+                build = rbuild.DatabaseBuildAsync(self)
 
-        return dbbuild
+        return build
 
 
     @property
@@ -399,7 +411,7 @@ class DatabaseSuper(
 
 
     @property
-    def schema(self) -> rbase.DatabaseSchemaT:
+    def schema(self) -> DatabaseSchemaT:
         """
         Build database schema instance.
 
@@ -490,7 +502,8 @@ class Database(
         'rconn.DatabaseConnection',
         'rexec.DatabaseExecute',
         'rparam.DatabaseSchema',
-        'rorm.DatabaseORM'
+        'rorm.DatabaseORM',
+        'rbuild.DatabaseBuild'
     ]
 ):
     """
@@ -504,7 +517,8 @@ class DatabaseAsync(
         'rconn.DatabaseConnectionAsync',
         'rexec.DatabaseExecuteAsync',
         'rparam.DatabaseSchemaAsync',
-        'rorm.DatabaseORMAsync'
+        'rorm.DatabaseORMAsync',
+        'rbuild.DatabaseBuildAsync'
     ]
 ):
     """

reydb/rerror.py

@@ -53,7 +53,7 @@ class DatabaseError(DatabaseBase):
 
     def build_db(self) -> None:
         """
-        Check and build all standard databases and tables, by `self.db_names`.
+        Check and build database tables, by `self.db_names`.
         """
 
         # Set parameter.
@@ -223,7 +223,7 @@ class DatabaseError(DatabaseBase):
 
         # Insert.
         self.db.execute.insert(
-            (self.db_names['base'], self.db_names['base.error']),
+            self.db_names['base.error'],
             data=data
         )