reydb 1.2.6__py3-none-any.whl → 1.2.8__py3-none-any.whl

reydb/rbuild.py

@@ -11,6 +11,7 @@
 
 from typing import TypedDict, NotRequired, Literal, Type, TypeVar, Generic
 from copy import deepcopy
+from sqlalchemy import UniqueConstraint
 from reykit.rbase import throw, is_instance
 from reykit.rstdout import ask
 
@@ -2487,7 +2488,7 @@ class DatabaseBuildSuper(DatabaseBase, Generic[DatabaseT]):
                     if column.autoincrement
                     else ' | KEY'
                 ) + (
-                    f" | DEFAULT '{column.server_default.arg}'"
+                    f" | DEFAULT {column.server_default.arg}"
                     if column.server_default
                     else ''
                 ) + (
@@ -2500,14 +2501,10 @@ class DatabaseBuildSuper(DatabaseBase, Generic[DatabaseT]):
         )
 
         ## Index.
-        if (table.indexes):
+        if table.indexes:
             text += '\n' + '\n'.join(
                 [
-                    (
-                        '    UNIQUE'
-                        if index.unique
-                        else '    NORMAL'
-                    ) + f' INDEX `{index.name}` : ' + ', '.join(
+                    '    NORMAL INDEX: ' + ', '.join(
                         [
                             f'`{column.name}`'
                             for column in index.expressions
@@ -2517,6 +2514,24 @@ class DatabaseBuildSuper(DatabaseBase, Generic[DatabaseT]):
                 ]
             )
 
+        ## Constraint.
+        if table.constraints:
+            text += '\n' + '\n'.join(
+                [
+                    (
+                        '    UNIQUE CONSTRAIN: '
+                        if type(constraint) == UniqueConstraint
+                        else '    PRIMARY KEY CONSTRAIN: '
+                    ) + ', '.join(
+                        [
+                            f'`{column.name}`'
+                            for column in constraint.columns
+                        ]
+                    )
+                    for constraint in table.constraints
+                ]
+            )
+
         return text
 
 

reydb/rconfig.py

@@ -44,12 +44,12 @@ class DatabaseTableConfig(rorm.Model, table=True):
     """
 
     __comment__ = 'Config data table.'
-    create_time: rorm.Datetime = rorm.Field(field_default='CURRENT_TIMESTAMP', not_null=True, index_n=True, comment='Config create time.')
-    update_time: rorm.Datetime = rorm.Field(field_default='ON UPDATE CURRENT_TIMESTAMP', index_n=True, comment='Config update time.')
-    key: str = rorm.Field(field_type=rorm.types.VARCHAR(50), key=True, comment='Config key.')
-    value: str = rorm.Field(field_type=rorm.types.TEXT, not_null=True, comment='Config value.')
-    type: str = rorm.Field(field_type=rorm.types.VARCHAR(50), not_null=True, comment='Config value type.')
-    note: str = rorm.Field(field_type=rorm.types.VARCHAR(500), comment='Config note.')
+    create_time: rorm.Datetime = rorm.Field(field_default=':create_time', not_null=True, index_n=True, comment='Config create time.')
+    update_time: rorm.Datetime = rorm.Field(field_default=':update_time', index_n=True, comment='Config update time.')
+    key: str = rorm.Field(rorm.types.VARCHAR(50), key=True, comment='Config key.')
+    value: str = rorm.Field(rorm.types.TEXT, not_null=True, comment='Config value.')
+    type: str = rorm.Field(rorm.types.VARCHAR(50), not_null=True, comment='Config value type.')
+    note: str = rorm.Field(rorm.types.VARCHAR(500), comment='Config note.')
 
 
 class DatabaseConfigSuper(DatabaseBase, Generic[DatabaseT]):

reydb/rerror.py

@@ -38,12 +38,12 @@ class DatabaseTableError(rorm.Model, table=True):
     """
 
     __comment__ = 'Error log table.'
-    create_time: rorm.Datetime = rorm.Field(field_default='CURRENT_TIMESTAMP', not_null=True, index_n=True, comment='Record create time.')
-    id: int = rorm.Field(field_type=rorm.types_mysql.INTEGER(unsigned=True), key_auto=True, comment='ID.')
-    type: str = rorm.Field(field_type=rorm.types.VARCHAR(50), not_null=True, index_n=True, comment='Error type.')
-    data: str = rorm.Field(field_type=rorm.types.JSON, comment='Error data.')
-    stack: str = rorm.Field(field_type=rorm.types.JSON, comment='Error code traceback stack.')
-    note: str = rorm.Field(field_type=rorm.types.VARCHAR(500), comment='Error note.')
+    create_time: rorm.Datetime = rorm.Field(field_default=':create_time', not_null=True, index_n=True, comment='Record create time.')
+    id: int = rorm.Field(rorm.types_mysql.INTEGER(unsigned=True), key_auto=True, comment='ID.')
+    type: str = rorm.Field(rorm.types.VARCHAR(50), not_null=True, index_n=True, comment='Error type.')
+    data: str = rorm.Field(rorm.types.JSON, comment='Error data.')
+    stack: str = rorm.Field(rorm.types.JSON, comment='Error code traceback stack.')
+    note: str = rorm.Field(rorm.types.VARCHAR(500), comment='Error note.')
 
 
 class DatabaseErrorSuper(DatabaseBase, Generic[DatabaseT]):

reydb/rorm.py

@@ -21,12 +21,13 @@ from sqlalchemy import types
 from sqlalchemy.dialects.mysql import types as types_mysql
 from sqlalchemy.sql.sqltypes import TypeEngine
 from sqlalchemy.sql.dml import Insert, Update, Delete
+from sqlalchemy.sql import func as sqlalchemy_func
 from sqlmodel import SQLModel, Session, Table
 from sqlmodel.ext.asyncio.session import AsyncSession
 from sqlmodel.main import SQLModelMetaclass, FieldInfo, default_registry
 from sqlmodel.sql._expression_select_cls import SelectOfScalar as Select
 from datetime import datetime, date, time, timedelta
-from reykit.rbase import CallableT, Null, throw, is_instance
+from reykit.rbase import CallableT, throw, is_instance
 
 from . import rdb
 from .rbase import (
@@ -122,14 +123,18 @@ class DatabaseORMModelMeta(DatabaseORMBase, SQLModelMetaclass):
                 table_args['comment'] = attrs.pop('__comment__')
 
             ## Field.
-            for __name__ in attrs['__annotations__']:
-                field = attrs.get(__name__)
-                if field is None:
-                    field = attrs[__name__] = DatabaseORMModelField()
+            for attr_name in attrs['__annotations__']:
+                attr_name: str
+                if attr_name in attrs:
+                    field = attrs[attr_name]
+                    if type(field) != DatabaseORMModelField:
+                        field = attrs[attr_name] = DatabaseORMModelField(field)
+                else:
+                    field = attrs[attr_name] = DatabaseORMModelField()
                 sa_column_kwargs: dict = field.sa_column_kwargs
-                sa_column_kwargs.setdefault('name', __name__)
+                sa_column_kwargs.setdefault('name', attr_name)
 
-            ## Unique.
+            ## Replace.
             table = default_registry.metadata.tables.get(table_name)
             if table is not None:
                 default_registry.metadata.remove(table)
@@ -155,11 +160,13 @@ class DatabaseORMModelMeta(DatabaseORMBase, SQLModelMetaclass):
         super().__init__(name, bases, attrs, **kwargs)
 
         # Set parameter.
-        if '__annotations__' in attrs:
+        if (
+            '__annotations__' in attrs
+            and hasattr(cls, '__table__')
+        ):
             table: Table = cls.__table__
             for index in table.indexes:
-                index_name_prefix = ['u_', 'n_'][index.unique]
-                index_name = index_name_prefix + '_'.join(
+                index_name = '_'.join(
                     column.key
                     for column in index.expressions
                 )
@@ -180,12 +187,12 @@ class DatabaseORMModelField(DatabaseORMBase, FieldInfo):
     @overload
     def __init__(
         self,
-        arg_default: Any | Callable[[], Any] | Null = Null,
-        *,
-        arg_name: str | None = None,
-        field_default: str | Literal['CURRENT_TIMESTAMP'] | Literal['ON UPDATE CURRENT_TIMESTAMP'] | None = None,
-        filed_name: str | None = None,
         field_type: TypeEngine | None = None,
+        *,
+        field_default: str | Literal[':time'] | Literal[':create_time'] | Literal[':update_time'] = None,
+        arg_default: Any | Callable[[], Any] | None = None,
+        arg_update: Any | Callable[[], Any] = None,
+        name: str | None = None,
         key: bool = False,
         key_auto: bool = False,
         not_null: bool = False,
@@ -208,7 +215,7 @@ class DatabaseORMModelField(DatabaseORMBase, FieldInfo):
 
     def __init__(
         self,
-        arg_default: Any | Callable[[], Any] | Null = Null,
+        field_type: TypeEngine | None = None,
         **kwargs: Any
     ) -> None:
         """
@@ -216,19 +223,22 @@ class DatabaseORMModelField(DatabaseORMBase, FieldInfo):
 
         Parameters
         ----------
-        arg_default : Call argument default value.
-        arg_name : Call argument name.
-            - `None`: Same as attribute name.
-        field_default : Database field defualt value.
-            - `Literal['current_timestamp']`: Set SQL syntax 'current_timestamp', case insensitive.
-            - `Literal['on update current_timestamp']`: Set SQL syntax 'on update current_timestamp', case insensitive.
-        field_name : Database field name.
-            - `None`: Same as attribute name.
         field_type : Database field type.
             - `None`: Based type annotation automatic judgment.
+        field_default : Database field defualt value.
+            - `Literal[':time']`: Set SQL syntax 'DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP'.
+            - `Literal[':create_time']`: Set SQL syntax 'DEFAULT CURRENT_TIMESTAMP'.
+            - `Literal[':update_time']`: Set SQL syntax 'ON UPDATE CURRENT_TIMESTAMP'.
+        arg_default : Call argument default value.
+            - `Callable[[], Any]`: Call function and use return value.
+        arg_update : In `Session` management, When commit update record, then default value is this value.
+            - `Callable[[], Any]`: Call function and use return value.
+        name : Call argument name and database field name.
+            - `None`: Same as attribute name.
         key : Whether the field is primary key.
-        key_auto : Whether the field is automatic increment primary key.
+        key_auto : Whether the field is primary key and automatic increment.
         not_null : Whether the field is not null constraint.
+            - `Litreal[False]`: When argument `arg_default` is `Null`, then set argument `arg_default` is `None`.
         index_n : Whether the field add normal index.
         index_u : Whether the field add unique index.
         comment : Field commment.
@@ -257,7 +267,6 @@ class DatabaseORMModelField(DatabaseORMBase, FieldInfo):
 
         ## Convert argument name.
         mapping_keys = {
-            'arg_name': 'alias',
             'key': 'primary_key',
             'index_n': 'index',
             'index_u': 'unique',
@@ -272,7 +281,6 @@ class DatabaseORMModelField(DatabaseORMBase, FieldInfo):
             'num_places': 'max_digits',
             'num_places_dec': 'decimal_places'
         }
-
         for key_old, key_new in mapping_keys.items():
             if type(key_new) != tuple:
                 key_new = (key_new,)
@@ -281,30 +289,13 @@ class DatabaseORMModelField(DatabaseORMBase, FieldInfo):
                 for key in key_new:
                     kwargs[key] = value
 
-        ## Argument default.
-        if (
-            arg_default != Null
-            and callable(arg_default)
-        ):
-            kwargs['default_factory'] = arg_default
-        else:
-            kwargs['default'] = arg_default
-
-        ## Field default.
-        if 'field_default' in kwargs:
-            field_default: str = kwargs.pop('field_default')
-            field_default_upper = field_default.upper()
-            if field_default_upper in ('CURRENT_TIMESTAMP', 'ON UPDATE CURRENT_TIMESTAMP'):
-                field_default = sqlalchemy_text(field_default_upper)
-            kwargs['sa_column_kwargs']['server_default'] = field_default
-
-        ## Field name.
-        if 'field_name' in kwargs:
-            kwargs['sa_column_kwargs']['name'] = kwargs.pop('field_name')
-
         ## Field type.
-        if 'field_type' in kwargs:
-            kwargs['sa_type'] = kwargs.pop('field_type')
+        if field_type is not None:
+            kwargs['sa_type'] = field_type
+
+        ## Name.
+        if 'name' in kwargs:
+            kwargs['alias'] = kwargs['sa_column_kwargs']['name'] = kwargs.pop('field_name')
 
         ## Key auto.
         if kwargs.get('key_auto'):
@@ -313,12 +304,45 @@ class DatabaseORMModelField(DatabaseORMBase, FieldInfo):
         else:
             kwargs['sa_column_kwargs']['autoincrement'] = False
 
+        ## Key.
+        if kwargs.get('primary_key'):
+            kwargs['nullable'] = False
+
         ## Non null.
         if 'not_null' in kwargs:
             kwargs['nullable'] = not kwargs.pop('not_null')
         else:
             kwargs['nullable'] = True
 
+        ## Field default.
+        if 'field_default' in kwargs:
+            field_default: str = kwargs.pop('field_default')
+            if field_default == ':time':
+                field_default = sqlalchemy_text('CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP')
+            if field_default == ':create_time':
+                field_default = sqlalchemy_text('CURRENT_TIMESTAMP')
+            elif field_default == ':update_time':
+                if kwargs['nullable']:
+                    field_default = sqlalchemy_text('NULL ON UPDATE CURRENT_TIMESTAMP')
+                else:
+                    field_default = sqlalchemy_text('NOT NULL ON UPDATE CURRENT_TIMESTAMP')
+            kwargs['sa_column_kwargs']['server_default'] = field_default
+
+        ## Argument default.
+        arg_default = kwargs.pop('arg_default', None)
+        if arg_default is not None:
+            if callable(arg_default):
+                kwargs['default_factory'] = arg_default
+            else:
+                kwargs['default'] = arg_default
+        elif kwargs['nullable']:
+            kwargs['default'] = None
+
+        ## Argument update.
+        if 'arg_update' in kwargs:
+            arg_update = kwargs.pop('arg_update')
+            kwargs['sa_column_kwargs']['onupdate'] = arg_update
+
         ## Comment.
         if 'comment' in kwargs:
             kwargs['sa_column_kwargs']['comment'] = kwargs.pop('comment')
@@ -1646,6 +1670,9 @@ types = types
 ## Database ORM model MySQL filed types.
 types_mysql = types_mysql
 
+## Database ORM model functions.
+funcs = sqlalchemy_func
+
 ## Create decorator of validate database ORM model.
 wrap_validate_model = pydantic_model_validator
 

{reydb-1.2.6.dist-info → reydb-1.2.8.dist-info}/METADATA

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

{reydb-1.2.6.dist-info → reydb-1.2.8.dist-info}/RECORD

@@ -1,15 +1,15 @@
 reydb/__init__.py,sha256=4mnlkfJfkBfxBpCotVUJ86f4AnT8plqlFbGiH3vZ4PM,550
 reydb/rall.py,sha256=IxSPGh77xz7ndDC7J8kZ_66Gq_xTAztGtnELUku1Ouw,364
 reydb/rbase.py,sha256=vx37yV6LlWP89nWAfYyOf0Xm3N_e9eB8z5Mxe-aTEo4,8248
-reydb/rbuild.py,sha256=imakdlqQXzQlbAukti3_REVU_h7IRRE3fwUk91LaIIM,80430
-reydb/rconfig.py,sha256=JvNf9LU6CzD9eThM8BZz0EVJeDgALmX8JHpY_mXR5mg,19207
+reydb/rbuild.py,sha256=0jI0MNl-BIhQtuT_-GeXBPTjG76k9EE3wwEQXGhTJyc,80938
+reydb/rconfig.py,sha256=B_lwfmMg4V4C_SmuLZi8dqGhb3El3ahxouj1snhS_Ek,19143
 reydb/rconn.py,sha256=guRaR8N6RuzZzujwaeq7HhKWTizF9SrUBqEAFjfjpoo,6909
 reydb/rdb.py,sha256=syyqZbEu92NbCj9O6_T6iAv7E46CyfQOC4T8qtPfHNs,14364
-reydb/rerror.py,sha256=Cmbcz5CgpI2ro7CbFHUYWP6zuiFJoxy_ENzSUNuuwe8,14924
+reydb/rerror.py,sha256=Hqb2CA6sG7yEPDt3UHzqrgdt63BqoWbCzji3mKBIlp0,14864
 reydb/rexec.py,sha256=djHx311c6mr1IjzNLqnGe-4yr3qNmYGUy4pHQA3WElQ,53042
 reydb/rinfo.py,sha256=LjrqTA7JJbWJsjXwV-zKpbE1htv-whg6239hoQj4yIU,18151
-reydb/rorm.py,sha256=baHqpNqnBaE1qQpsc5TaegJBbOhWnl748rPPzO7Ab6Q,42314
-reydb-1.2.6.dist-info/METADATA,sha256=O94lfy2GvYZTAHXYwRlts15238S9IqTmoTIyDiwmJyo,1621
-reydb-1.2.6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-reydb-1.2.6.dist-info/licenses/LICENSE,sha256=UYLPqp7BvPiH8yEZduJqmmyEl6hlM3lKrFIefiD4rvk,1059
-reydb-1.2.6.dist-info/RECORD,,
+reydb/rorm.py,sha256=BXRmAaZRrIS6ZprtKRDMUHXtxuaMa6RNC7-_uqARSpo,43650
+reydb-1.2.8.dist-info/METADATA,sha256=LW49lxiggNPo5NgnLVeKo1Euvi11f3QlrD40falMIOE,1621
+reydb-1.2.8.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+reydb-1.2.8.dist-info/licenses/LICENSE,sha256=UYLPqp7BvPiH8yEZduJqmmyEl6hlM3lKrFIefiD4rvk,1059
+reydb-1.2.8.dist-info/RECORD,,