mtsql 1.11.3__py3-none-any.whl → 1.11.4__py3-none-any.whl

mt/sql/base.py

@@ -23,6 +23,9 @@ __all__ = [
     "list_views",
     "table_exists",
     "create_temp_id_table",
+    "temp_table_name",
+    "temp_table_find_new_id",
+    "temp_table_drop",
 ]
 
 
@@ -387,3 +390,50 @@ def create_temp_id_table(l_ids: list, conn: sa.engine.Connection) -> str:
     conn.execute(sa.text(query_str))
 
     return table_name
+
+
+def temp_table_name(id: int) -> str:
+    """Converts a temp table id into a temp table name."""
+    return f"mttmp_{id}"
+
+
+def temp_table_find_new_id(engine: sa.engine.Engine) -> int:
+    """Finds a new temp table id that does not exist in the public schema.
+
+    Parameters
+    ----------
+    engine : sqlalchemy.engine.Engine
+        connection engine to the server
+
+    Returns
+    -------
+    id : int
+        table id that has not been existent in the public schema.
+    """
+    l_tableNames = list_tables(engine)
+    max_id = -1
+    for table_name in l_tableNames:
+        s = re.match(r"mttmp_(\d+)", table_name)
+        if s:
+            id = int(s[1])
+            if max_id < id:
+                max_id = id
+    return max_id + 1
+
+
+def temp_table_drop(
+    id: int,
+    engine: sa.engine.Engine,
+):
+    """Drops a temp table.
+
+    Parameters
+    ----------
+    id : int
+        table id. Can be generated by invoking :func:`temp_table_find_new_id`
+    engine : sqlalchemy.engine.Engine
+        connection engine to the server
+    """
+
+    sql = f"DROP TABLE IF EXISTS {temp_table_name(id)}"
+    return engine_execute(engine, sql)

mt/sql/version.py

@@ -1,5 +1,5 @@
 MAJOR_VERSION = 1
 MINOR_VERSION = 11
-PATCH_VERSION = 3
+PATCH_VERSION = 4
 version = '{}.{}.{}'.format(MAJOR_VERSION, MINOR_VERSION, PATCH_VERSION)
 __all__  = ['MAJOR_VERSION', 'MINOR_VERSION', 'PATCH_VERSION', 'version']

{mtsql-1.11.3.dist-info → mtsql-1.11.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mtsql
-Version: 1.11.3
+Version: 1.11.4
 Summary: Extra Python modules to deal with the interaction between pandas dataframes and remote SQL servers, for Minh-Tri Pham
 Home-page: https://github.com/inteplus/mtsql
 Author: ['Minh-Tri Pham']

mtsql-1.11.4.dist-info/RECORD

@@ -0,0 +1,12 @@
+mt/sql/__init__.py,sha256=b7zO50apZxt9Hg2eOkJhRLrXgACR8eS5b-Rphdn5qNQ,44
+mt/sql/base.py,sha256=H_bDZWgtaleNcb45atrgrk3fYwVw9OskrkZsfFWP5VA,11786
+mt/sql/mysql.py,sha256=n2ENDctdUqZuSaDAcrqZYtPtawq3Wx4dOPCRsCB5Q4w,4894
+mt/sql/psql.py,sha256=tzsIN2XyBJX8t7p0D2yXgEQwUTkTcDzllI4HunCfq3s,66345
+mt/sql/redshift.py,sha256=Nve12tQ6_vO3M6yOpQQn3TkI1r4eDRBLsU0agOLRDvs,17037
+mt/sql/sqlite.py,sha256=T2ak_hhNi_zRfpg_gp8JhNHn7D2kl4i-Ey6-9ANMtz0,8678
+mt/sql/version.py,sha256=EJI7ks7VazIXVEXRmxPmjvpx4qjsXGHfiJ5XSOskhlI,207
+mtsql-1.11.4.dist-info/LICENSE,sha256=PojkRlQzTT5Eg6Nj03XoIVEefN3u8iiIFf1p4rqe_t4,1070
+mtsql-1.11.4.dist-info/METADATA,sha256=rxQ6i-8lbpMiFqPw7ZTASYmEhRDXj5ZQTQWoYvRyPo4,579
+mtsql-1.11.4.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
+mtsql-1.11.4.dist-info/top_level.txt,sha256=WcqGFu9cV7iMZg09iam8eNxUvGpLSKKF2Iubf6SJVOo,3
+mtsql-1.11.4.dist-info/RECORD,,

{mtsql-1.11.3.dist-info → mtsql-1.11.4.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.43.0)
+Generator: setuptools (72.1.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

mtsql-1.11.3.dist-info/RECORD

@@ -1,12 +0,0 @@
-mt/sql/__init__.py,sha256=b7zO50apZxt9Hg2eOkJhRLrXgACR8eS5b-Rphdn5qNQ,44
-mt/sql/base.py,sha256=iz60AAer-kSFNKotRP6D2JtxIwPyIyL8BJYVVLtmZco,10569
-mt/sql/mysql.py,sha256=n2ENDctdUqZuSaDAcrqZYtPtawq3Wx4dOPCRsCB5Q4w,4894
-mt/sql/psql.py,sha256=tzsIN2XyBJX8t7p0D2yXgEQwUTkTcDzllI4HunCfq3s,66345
-mt/sql/redshift.py,sha256=Nve12tQ6_vO3M6yOpQQn3TkI1r4eDRBLsU0agOLRDvs,17037
-mt/sql/sqlite.py,sha256=T2ak_hhNi_zRfpg_gp8JhNHn7D2kl4i-Ey6-9ANMtz0,8678
-mt/sql/version.py,sha256=Qi5LGoO7TZ4qAVw4wY48Y93y0s5fIxvoXnq2OAVlJtk,207
-mtsql-1.11.3.dist-info/LICENSE,sha256=PojkRlQzTT5Eg6Nj03XoIVEefN3u8iiIFf1p4rqe_t4,1070
-mtsql-1.11.3.dist-info/METADATA,sha256=rc9T8lsrBi6DENup8tRb9l8b9C8AIKzbcHMR0te8zGw,579
-mtsql-1.11.3.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-mtsql-1.11.3.dist-info/top_level.txt,sha256=WcqGFu9cV7iMZg09iam8eNxUvGpLSKKF2Iubf6SJVOo,3
-mtsql-1.11.3.dist-info/RECORD,,