mtsql 1.12.22__py3-none-any.whl → 1.12.25__py3-none-any.whl

mt/sql/psql.py

@@ -636,24 +636,27 @@ def rename_table(
     new_table_name,
     engine,
     schema: tp.Optional[str] = None,
+    foreign: bool = False,
     nb_trials: int = 3,
     logger: tp.Optional[logg.IndentedLoggerAdapter] = None,
 ):
-    """Renames a table of a schema.
+    """Renames a (foreign) table of a schema.
 
     Parameters
     ----------
-    old_table_name: str
+    old_table_name : str
         old table name
-    new_table_name: str
+    new_table_name : str
         new table name
-    engine: sqlalchemy.engine.Engine
+    engine : sqlalchemy.engine.Engine
         an sqlalchemy connection engine created by function `create_engine()`
-    schema: str or None
+    schema : str or None
         a valid schema name returned from `list_schemas()`
-    nb_trials: int
+    foreign : bool
+        whether the table to rename is a foreign table
+    nb_trials : int
         number of query trials
-    logger: mt.logg.IndentedLoggerAdapter, optional
+    logger : mt.logg.IndentedLoggerAdapter, optional
         logger for debugging
 
     Returns
@@ -661,8 +664,12 @@ def rename_table(
     whatever exec_sql() returns
     """
     frame_sql_str = frame_sql(old_table_name, schema=schema)
+    if foreign:
+        query_str = f'ALTER FOREIGN TABLE {frame_sql_str} RENAME TO "{new_table_name}";'
+    else:
+        query_str = f'ALTER TABLE {frame_sql_str} RENAME TO "{new_table_name}";'
     return exec_sql(
-        'ALTER TABLE {} RENAME TO "{}";'.format(frame_sql_str, new_table_name),
+        query_str,
         engine,
         nb_trials=nb_trials,
         logger=logger,
@@ -710,21 +717,24 @@ def drop_table(
     table_name,
     engine,
     schema: tp.Optional[str] = None,
-    restrict=True,
+    foreign: bool = False,
+    restrict: bool = True,
     nb_trials: int = 3,
     logger: tp.Optional[logg.IndentedLoggerAdapter] = None,
 ):
-    """Drops a table if it exists, with restrict or cascade options.
+    """Drops a (foreign) table if it exists, with restrict or cascade options.
 
     Parameters
     ----------
     table_name : str
         table name
-    engine: sqlalchemy.engine.Engine
+    engine : sqlalchemy.engine.Engine
         an sqlalchemy connection engine created by function `create_engine()`
-    schema: str or None
+    schema : str or None
         a valid schema name returned from `list_schemas()`
-    restrict: bool
+    foreign : bool
+        whether the table to drop is a foreign table
+    restrict : bool
         If True, refuses to drop table if there is any object depending on it. Otherwise it is the
         'cascade' option which allows you to remove those dependent objects together with the table
         automatically.
@@ -738,9 +748,11 @@ def drop_table(
     whatever exec_sql() returns
     """
     frame_sql_str = frame_sql(table_name, schema=schema)
-    query_str = "DROP TABLE IF EXISTS {} {};".format(
-        frame_sql_str, "RESTRICT" if restrict else "CASCADE"
-    )
+    if foreign:
+        query_str = "DROP FOREIGN TABLE "
+    else:
+        query_str = "DROP TABLE "
+    query_str += f'IF EXISTS {frame_sql_str} {"RESTRICT" if restrict else "CASCADE"};'
     return exec_sql(query_str, engine, nb_trials=nb_trials, logger=logger)
 
 

mt/sql/version.py

@@ -1,5 +1,5 @@
 MAJOR_VERSION = 1
 MINOR_VERSION = 12
-PATCH_VERSION = 22
+PATCH_VERSION = 25
 version = '{}.{}.{}'.format(MAJOR_VERSION, MINOR_VERSION, PATCH_VERSION)
 __all__  = ['MAJOR_VERSION', 'MINOR_VERSION', 'PATCH_VERSION', 'version']

{mtsql-1.12.22.dist-info → mtsql-1.12.25.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mtsql
-Version: 1.12.22
+Version: 1.12.25
 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']
@@ -12,8 +12,8 @@ Requires-Dist: tzlocal
 Requires-Dist: tqdm
 Requires-Dist: psycopg[binary]
 Requires-Dist: redshift_connector>=2.1.5
-Requires-Dist: mtbase>=4.32.27
-Requires-Dist: mtpandas>=1.17.13
+Requires-Dist: mtbase>=4.32.34
+Requires-Dist: mtpandas>=1.17.16
 Dynamic: author
 Dynamic: home-page
 Dynamic: license-file

{mtsql-1.12.22.dist-info → mtsql-1.12.25.dist-info}/RECORD

@@ -1,17 +1,17 @@
 mt/sql/__init__.py,sha256=b7zO50apZxt9Hg2eOkJhRLrXgACR8eS5b-Rphdn5qNQ,44
 mt/sql/base.py,sha256=j2fzeGUdh-TnvgotuWOmTM-A0pqBg31z2c19bH5E_xk,13504
 mt/sql/mysql.py,sha256=n2ENDctdUqZuSaDAcrqZYtPtawq3Wx4dOPCRsCB5Q4w,4894
-mt/sql/psql.py,sha256=w5klr_2lUZnAFee-Fzbn3G8jdYcf_iX4N_GfAGIdsMI,66670
+mt/sql/psql.py,sha256=8gWsF6ZobwkAxHsfecF95Em_cAlMitHHdQzinZLCi6s,67110
 mt/sql/sqlite.py,sha256=T2ak_hhNi_zRfpg_gp8JhNHn7D2kl4i-Ey6-9ANMtz0,8678
-mt/sql/version.py,sha256=TMep9gRjrVIi_raEj5ssyAJSqu8WjBqWN-k1B6KZSgg,208
+mt/sql/version.py,sha256=fESKN3a-51Gk518u42eRfS6jc2qynXacnonQdAruWec,208
 mt/sql/redshift/__init__.py,sha256=S-eRxJWcrvncF7LZXuulCdPV-UERu9eiw6uyDb5aGsM,443
 mt/sql/redshift/commands.py,sha256=xhGUBf3bL66EYdZI5HCUtOx-XqPCnXT_P-LnhPgtzrY,40193
 mt/sql/redshift/ddl.py,sha256=eUcZj9oIajiE1wKKBAP-V64gYJ7cVnSAt8dLFfluOJA,9777
 mt/sql/redshift/dialect.py,sha256=-0JjJubZZHRw0abhl6H6rKWaUE9pKtGwAuX-62T0e_c,53399
 mt/sql/redshift/main.py,sha256=H8_5sjtJ7dzWoCMXzPNjYhrMQ18eLUQ9xg-aYl5QeTc,17104
 mt/sql/redshift/redshift-ca-bundle.crt,sha256=532qYkOpQOstFE0mdXE1GVtL3v00XDKgZNTr6gK5-KE,8621
-mtsql-1.12.22.dist-info/licenses/LICENSE,sha256=PojkRlQzTT5Eg6Nj03XoIVEefN3u8iiIFf1p4rqe_t4,1070
-mtsql-1.12.22.dist-info/METADATA,sha256=3zityjnam0Lf_ihhR1UDvNhJJ6bxqOlM8QPA0flFU78,740
-mtsql-1.12.22.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-mtsql-1.12.22.dist-info/top_level.txt,sha256=WcqGFu9cV7iMZg09iam8eNxUvGpLSKKF2Iubf6SJVOo,3
-mtsql-1.12.22.dist-info/RECORD,,
+mtsql-1.12.25.dist-info/licenses/LICENSE,sha256=PojkRlQzTT5Eg6Nj03XoIVEefN3u8iiIFf1p4rqe_t4,1070
+mtsql-1.12.25.dist-info/METADATA,sha256=nMGzCZOmjrDiBh9wWHes41XqZ2Q9nvDcsdzNI1ln8tk,740
+mtsql-1.12.25.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+mtsql-1.12.25.dist-info/top_level.txt,sha256=WcqGFu9cV7iMZg09iam8eNxUvGpLSKKF2Iubf6SJVOo,3
+mtsql-1.12.25.dist-info/RECORD,,