reydb 1.1.47__py3-none-any.whl → 1.1.49__py3-none-any.whl

reydb/rbase.py

@@ -9,9 +9,10 @@
 """
 
 
-from typing import Any, TypedDict, Literal
+from typing import Any, TypedDict, Literal, overload
 from sqlalchemy.engine.base import Engine, Connection
 from sqlalchemy.engine.url import URL
+from sqlalchemy.sql.elements import TextClause
 from reykit.rbase import Base, throw
 from reykit.rre import search
 
@@ -19,7 +20,8 @@ from reykit.rre import search
 __all__ = (
     'DatabaseBase',
     'extract_url',
-    'extract_engine'
+    'extract_engine',
+    'extract_path'
 )
 
 
@@ -187,3 +189,100 @@ def extract_engine(engine: Engine | Connection) -> dict[
     }
 
     return params
+
+
+def extract_path(
+    path: str | tuple[str] | tuple[str, str] | tuple[str, str, str]
+) -> tuple[str, str | None, str | None]:
+    """
+    Extract table name and database name and column name from path.
+
+    Parameters
+    ----------
+    path : Path.
+        - `str` and not contain '.' or contain '`': Database name.
+        - `str` and contain '.': Database name and table name, column name is optional. Format is 'database.table[.column]'.
+        - `tuple`: Format is `(database[, table, column])`.
+
+    Returns
+    -------
+    Names.
+    """
+
+    # Type str.
+    if type(path) == str:
+
+        ## Single.
+        if (
+            '.' not in path
+            or '`' in path
+        ):
+            name = path.replace('`', '')
+            names = (name, None, None)
+
+        ## Multiple.
+        else:
+            names = path.split('.', 2)
+            if len(names) == 2:
+                names.append(None)
+            names = tuple(names)
+
+    # Type tuple.
+    else:
+        if len(path) == 1:
+            path += (None, None)
+        if len(path) == 2:
+            path += (None,)
+        names = path
+
+    return names
+
+
+def get_syntax(self, sql: str | TextClause) -> list[str]:
+    """
+    Extract SQL syntax type for each segment form SQL.
+
+    Parameters
+    ----------
+    sql : SQL text or TextClause object.
+
+    Returns
+    -------
+    SQL syntax type for each segment.
+    """
+
+    # Handle parameter.
+    if type(sql) == TextClause:
+        sql = sql.text
+
+    # Extract.
+    syntax = [
+        search('[a-zA-Z]+', sql_part).upper()
+        for sql_part in sql.split(';')
+        if sql_part != ''
+    ]
+
+    return syntax
+
+
+def is_multi_sql(self, sql: str | TextClause) -> bool:
+    """
+    Judge whether it is multi segment SQL.
+
+    Parameters
+    ----------
+    sql : SQL text or TextClause object.
+
+    Returns
+    -------
+    Judgment result.
+    """
+
+    # Handle parameter.
+    if type(sql) == TextClause:
+        sql = sql.text
+
+    # Judge.
+    if ';' in sql.rstrip()[:-1]:
+        return True
+    return False

reydb/rbuild.py

@@ -14,7 +14,7 @@ from copy import deepcopy
 from reykit.rbase import throw
 from reykit.rstdout import ask
 
-from .rbase import DatabaseBase
+from .rbase import DatabaseBase, extract_path
 from .rconn import DatabaseConnection
 from .rdb import Database
 
@@ -61,11 +61,6 @@ class DatabaseBuild(DatabaseBase):
         database : Database or DatabaseConnection instance.
         """
 
-        # SQLite.
-        if database.backend == 'sqlite':
-            text = 'not suitable for SQLite databases'
-            throw(AssertionError, text=text)
-
         # Set attribute.
         self.database = database
         self._schema: dict[str, dict[str, list[str]]] | None = None
@@ -285,7 +280,8 @@ class DatabaseBuild(DatabaseBase):
         """
 
         # Handle parameter.
-        database, table, _ = self.database.extract_path(path)
+        if type(path) == str:
+            database, table, _ = extract_path(path)
         if fields.__class__ == dict:
             fields = [fields]
         if primary.__class__ == str:
@@ -375,7 +371,7 @@ class DatabaseBuild(DatabaseBase):
         """
 
         # Handle parameter.
-        database, view, _ = self.database.extract_path(path)
+        database, view, _ = extract_path(path)
 
         # Generate SQL.
         select = select.replace('\n', '\n    ')
@@ -502,7 +498,7 @@ class DatabaseBuild(DatabaseBase):
         """
 
         # Handle parameter.
-        database, table, _ = self.database.extract_path(path)
+        database, table, _ = extract_path(path)
 
         # Generate.
         sql = f'DROP TABLE `{database}`.`{table}`'
@@ -535,7 +531,7 @@ class DatabaseBuild(DatabaseBase):
         """
 
         # Handle parameter.
-        database, view, _ = self.database.extract_path(path)
+        database, view, _ = extract_path(path)
 
         # Generate SQL.
         sql = 'DROP VIEW `%s`.`%s`' % (database, view)
@@ -649,7 +645,7 @@ class DatabaseBuild(DatabaseBase):
         """
 
         # Handle parameter.
-        database, table, _ = self.database.extract_path(path)
+        database, table, _ = extract_path(path)
         if fields.__class__ == dict:
             fields = [fields]
         if primary.__class__ == str:
@@ -739,7 +735,7 @@ class DatabaseBuild(DatabaseBase):
         """
 
         # Handle parameter.
-        database, table, _ = self.database.extract_path(path)
+        database, table, _ = extract_path(path)
         if fields.__class__ == str:
             fields = [fields]
         if indexes.__class__ == str:
@@ -829,7 +825,7 @@ class DatabaseBuild(DatabaseBase):
         """
 
         # Handle parameter.
-        database, table, _ = self.database.extract_path(path)
+        database, table, _ = extract_path(path)
         if fields.__class__ == dict:
             fields = [fields]
 
@@ -925,7 +921,7 @@ class DatabaseBuild(DatabaseBase):
         """
 
         # Handle parameter.
-        database, view, _ = self.database.extract_path(path)
+        database, view, _ = extract_path(path)
 
         # Generate SQL.
         sql = 'ALTER VIEW `%s`.`%s` AS\n%s' % (database, view, select)
@@ -958,7 +954,7 @@ class DatabaseBuild(DatabaseBase):
         """
 
         # Handle parameter.
-        database, table, _ = self.database.extract_path(path)
+        database, table, _ = extract_path(path)
 
         # Generate.
         sql = f'TRUNCATE TABLE `{database}`.`{table}`'
@@ -972,7 +968,7 @@ class DatabaseBuild(DatabaseBase):
 
     def exist(
         self,
-        path: str | tuple[str, str | None, str | None]
+        path: str | tuple[str] | tuple[str, str] | tuple[str, str, str]
     ) -> bool:
         """
         Judge database or table or column exists.
@@ -981,7 +977,7 @@ class DatabaseBuild(DatabaseBase):
         ----------
         path : Database name and table name and column name.
             - `str`: Automatic extract.
-            - `tuple[str, str | None, str | None]`: Database name, table name and column name is optional.
+            - `tuple`: Format is `(database[, table, column]).`
 
         Returns
         -------
@@ -989,7 +985,7 @@ class DatabaseBuild(DatabaseBase):
         """
 
         # Handle parameter.
-        database, table, column = self.database.extract_path(path, 'database')
+        database, table, column = extract_path(path)
         if self._schema is None:
             self._schema = self.database.schema(False)
 
@@ -1101,7 +1097,7 @@ class DatabaseBuild(DatabaseBase):
         # Table.
         for params in tables:
             path = params['path']
-            database, table, _ = self.database.extract_path(path)
+            database, table, _ = extract_path(path)
 
             ## Exist.
             exist = self.database.build.exist((database, table, None))
@@ -1124,7 +1120,7 @@ class DatabaseBuild(DatabaseBase):
         # View.
         for params in views:
             path = params['path']
-            database, view, _ = self.database.extract_path(path)
+            database, view, _ = extract_path(path)
 
             ## Exist.
             exist = self.database.build.exist((database, view, None))
@@ -1147,7 +1143,7 @@ class DatabaseBuild(DatabaseBase):
         # View stats.
         for params in views_stats:
             path = params['path']
-            database, view, _ = self.database.extract_path(path)
+            database, view, _ = extract_path(path)
 
             ## Exist.
             exist = self.database.build.exist((database, view, None))

reydb/rconfig.py

@@ -54,11 +54,6 @@ class DatabaseConfig(object):
         Build instance attributes.
         """
 
-        # SQLite.
-        if database.backend == 'sqlite':
-            text = 'not suitable for SQLite databases'
-            throw(AssertionError, text=text)
-
         # Build.
         self.database = database
 
@@ -199,7 +194,7 @@ class DatabaseConfig(object):
         """
 
         # Get.
-        result = self.database.execute_select(
+        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'
@@ -235,7 +230,7 @@ class DatabaseConfig(object):
 
         # Get.
         where = '`key` = :key'
-        result = self.database.execute_select(
+        result = self.database.execute.select(
             (self.db_names['base'], self.db_names['base.config']),
             '`value`',
             where,
@@ -281,7 +276,7 @@ class DatabaseConfig(object):
             'type': type(default).__name__,
             'note': default_note
         }
-        result = self.database.execute_insert(
+        result = self.database.execute.insert(
             (self.db_names['base'], self.db_names['base.config']),
             data,
             'ignore'
@@ -322,7 +317,7 @@ class DatabaseConfig(object):
                 row['type'] = type(row['value']).__name__
 
         # Update.
-        self.database.execute_insert(
+        self.database.execute.insert(
             (self.db_names['base'], self.db_names['base.config']),
             data,
             'update'
@@ -345,7 +340,7 @@ class DatabaseConfig(object):
         else:
             where = '`key` in :key'
             limit = None
-        result = self.database.execute_delete(
+        result = self.database.execute.delete(
             (self.db_names['base'], self.db_names['base.config']),
             where,
             limit=limit,
@@ -367,7 +362,7 @@ class DatabaseConfig(object):
         """
 
         # Get.
-        result = self.database.execute_select(
+        result = self.database.execute.select(
             (self.db_names['base'], self.db_names['base.config']),
             ['key', 'value']
         )
@@ -393,7 +388,7 @@ class DatabaseConfig(object):
         """
 
         # Get.
-        result = self.database.execute_select(
+        result = self.database.execute.select(
             (self.db_names['base'], self.db_names['base.config']),
             '`key`'
         )
@@ -418,7 +413,7 @@ class DatabaseConfig(object):
         """
 
         # Get.
-        result = self.database.execute_select(
+        result = self.database.execute.select(
             (self.db_names['base'], self.db_names['base.config']),
             '`value`'
         )
@@ -480,7 +475,7 @@ class DatabaseConfig(object):
             'type': type(value).__name__,
             'note': note
         }
-        self.database.execute_insert(
+        self.database.execute.insert(
             (self.db_names['base'], self.db_names['base.config']),
             data,
             'update'

reydb/rconn.py

@@ -10,10 +10,9 @@
 
 
 from typing import Self
-from sqlalchemy.engine.base import Connection
-from sqlalchemy.sql.elements import TextClause
 
-from .rdb import Result, Database
+from .rbase import DatabaseBase
+from .rdb import Database
 
 
 __all__ = (
@@ -21,7 +20,7 @@ __all__ = (
 )
 
 
-class DatabaseConnection(Database):
+class DatabaseConnection(DatabaseBase):
     """
     Database connection type.
     """
@@ -29,66 +28,46 @@ class DatabaseConnection(Database):
 
     def __init__(
         self,
-        connection: Connection,
-        rdatabase: Database
+        db: Database,
+        autocommit: bool
     ) -> None:
         """
         Build instance attributes.
 
         Parameters
         ----------
-        connection : Connection object.
-        rdatabase : Database object.
+        db : Database instance.
+        autocommit: Whether automatic commit connection.
         """
 
-        # Set parameter.
-        self.connection = connection
-        self.rdatabase = rdatabase
+        # Build.
+        self.db = db
+        self.autocommit = autocommit
+        self.conn = db.engine.connect()
         self.begin = None
-        self.username = rdatabase.username
-        self.password = rdatabase.password
-        self.host = rdatabase.host
-        self.port = rdatabase.port
-        self.database = rdatabase.database
-        self.query = rdatabase.query
-        self.pool_recycle = rdatabase.pool_recycle
 
 
-    def executor(
-        self,
-        sql: TextClause,
-        data: list[dict],
-        report: bool
-    ) -> Result:
+    @property
+    def execute(self):
         """
-        SQL executor.
-
-        Parameters
-        ----------
-        sql : TextClause object.
-        data : Data set for filling.
-        report : Whether report SQL execute information.
+        Build `database execute` instance.
 
         Returns
         -------
-        Result object.
+        Instance.
         """
 
+        # Import.
+        from .rexec import DatabaseExecute
+
         # Create transaction.
         if self.begin is None:
-            self.begin = self.connection.begin()
-
-        # Execute.
+            self.begin = self.conn.begin()
 
-        ## Report.
-        if report:
-            result = self.executor_report(self.connection, sql, data)
-
-        ## Not report.
-        else:
-            result = self.connection.execute(sql, data)
+        # Build.
+        exec = DatabaseExecute(self)
 
-        return result
+        return exec
 
 
     def commit(self) -> None:
@@ -119,7 +98,7 @@ class DatabaseConnection(Database):
         """
 
         # Close.
-        self.connection.close()
+        self.conn.close()
 
 
     def __enter__(self) -> Self: