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

mt/sql/base.py

@@ -76,6 +76,7 @@ def run_func(
             se.InterfaceError,
             se.PendingRollbackError,
             ssl.SSLEOFError,
+            ssl.SSLZeroReturnError,
         ):
             if x < nb_trials - 1:
                 if logger:
@@ -370,6 +371,7 @@ def create_temp_id_table(
     conn: sa.engine.Connection,
     int_type="int",
     chunksize: int = 1000000,
+    logger: tp.Optional[logg.IndentedLoggerAdapter] = None,
 ) -> str:
     """Creates a temporary table to containing a list of ids.
 
@@ -383,6 +385,8 @@ def create_temp_id_table(
         an SQL string representing the int type
     chunksize : int
         maximum number of ids to be inserted in each INSERT statement
+    logger : mt.logg.IndentedLoggerAdapter, optional
+        logger for debugging
 
     Returns
     -------
@@ -393,7 +397,7 @@ def create_temp_id_table(
     table_name = f"tab_{uuid.uuid4().hex}"
 
     query_str = f"CREATE TEMP TABLE {table_name}(id {int_type});"
-    exec_sql(sa.text(query_str), conn)
+    exec_sql(sa.text(query_str), conn, logger=logger)
 
     while True:
         l_ids2 = l_ids[:chunksize]
@@ -402,7 +406,7 @@ def create_temp_id_table(
 
         values = ",".join((f"({id})" for id in l_ids2))
         query_str = f"INSERT INTO {table_name}(id) VALUES {values};"
-        exec_sql(sa.text(query_str), conn)
+        exec_sql(sa.text(query_str), conn, logger=logger)
         l_ids = l_ids[chunksize:]
 
     return table_name
@@ -440,6 +444,7 @@ def temp_table_find_new_id(engine: sa.engine.Engine) -> int:
 def temp_table_drop(
     engine: sa.engine.Engine,
     id: int,
+    logger: tp.Optional[logg.IndentedLoggerAdapter] = None,
 ):
     """Drops a temp table.
 
@@ -449,11 +454,13 @@ def temp_table_drop(
         table id or table name. An id can be generated by invoking :func:`temp_table_find_new_id`.
     engine : sqlalchemy.engine.Engine
         connection engine to the server
+    logger : mt.logg.IndentedLoggerAdapter, optional
+        logger for debugging
     """
 
     name = id if isinstance(id, str) else temp_table_name(id)
     sql = f"DROP TABLE IF EXISTS {name}"
-    return exec_sql(sql, engine)
+    return exec_sql(sql, engine, logger=logger)
 
 
 @ctx.contextmanager

mt/sql/version.py

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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mtsql
-Version: 1.12.20
+Version: 1.12.22
 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.12.20.dist-info → mtsql-1.12.22.dist-info}/RECORD

@@ -1,17 +1,17 @@
 mt/sql/__init__.py,sha256=b7zO50apZxt9Hg2eOkJhRLrXgACR8eS5b-Rphdn5qNQ,44
-mt/sql/base.py,sha256=DDoE3bfAJOjDA7YqDctQqA1L9mkasubDCf6SwCNPe9c,13139
+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/sqlite.py,sha256=T2ak_hhNi_zRfpg_gp8JhNHn7D2kl4i-Ey6-9ANMtz0,8678
-mt/sql/version.py,sha256=cIHvxCMlXlz8gD9kQVucrmouSEzUOeBeLpuyYMhmx4c,208
+mt/sql/version.py,sha256=TMep9gRjrVIi_raEj5ssyAJSqu8WjBqWN-k1B6KZSgg,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.20.dist-info/licenses/LICENSE,sha256=PojkRlQzTT5Eg6Nj03XoIVEefN3u8iiIFf1p4rqe_t4,1070
-mtsql-1.12.20.dist-info/METADATA,sha256=nZHmvYsUawi9JTKBXHOBiJxLarfw9ukvF_u6GCio6nw,740
-mtsql-1.12.20.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-mtsql-1.12.20.dist-info/top_level.txt,sha256=WcqGFu9cV7iMZg09iam8eNxUvGpLSKKF2Iubf6SJVOo,3
-mtsql-1.12.20.dist-info/RECORD,,
+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,,