mtsql 1.12.9__py3-none-any.whl → 1.12.11__py3-none-any.whl

mt/sql/base.py

@@ -31,7 +31,7 @@ __all__ = [
 
 
 def frame_sql(frame_name, schema: tp.Optional[str] = None):
-    return frame_name if schema is None else "{}.{}".format(schema, frame_name)
+    return frame_name if schema is None else f"{schema}.{frame_name}"
 
 
 def indices(df):
@@ -48,26 +48,26 @@ def run_func(
     *args,
     nb_trials: int = 3,
     logger: tp.Optional[logg.IndentedLoggerAdapter] = None,
-    **kwargs,
+    **kwds,
 ):
     """Attempt to run a function a number of times to overcome OperationalError exceptions.
 
     Parameters
     ----------
-    func: function
+    func : function
         function to be invoked
-    args: sequence
-        arguments to be passed to the function
-    nb_trials: int
+    nb_trials : int
         number of query trials
-    logger: mt.logg.IndentedLoggerAdapter, optional
+    logger : mt.logg.IndentedLoggerAdapter, optional
         logger for debugging
-    kwargs: dict
+    *args : iterable
+        arguments to be passed to the function
+    **kwds : dict
         keyword arguments to be passed to the function
     """
     for x in range(nb_trials):
         try:
-            return func(*args, **kwargs)
+            return func(*args, **kwds)
         except (se.ProgrammingError, se.IntegrityError):
             raise
         except (
@@ -92,10 +92,10 @@ def conn_ctx(engine):
     return ctx.nullcontext(engine)
 
 
-def engine_execute(engine, sql, *args, **kwargs):
+def engine_execute(engine, sql, *args, **kwds):
     text_sql = sa.text(sql) if isinstance(sql, str) else sql
     with conn_ctx(engine) as conn:
-        return conn.execute(text_sql, *args, **kwargs)
+        return conn.execute(text_sql, *args, **kwds)
 
 
 def trim_sql_query(sql_query: str) -> str:
@@ -112,7 +112,7 @@ def read_sql(
     nb_trials: int = 3,
     exception_handling: str = "raise",
     logger: tp.Optional[logg.IndentedLoggerAdapter] = None,
-    **kwargs,
+    **kwds,
 ) -> pd.DataFrame:
     """Read an SQL query with a number of trials to overcome OperationalError.
 
@@ -165,7 +165,7 @@ def read_sql(
     text_sql = trim_sql_query(text_sql)
 
     if chunksize is not None:
-        s = "read_sql: '{}'".format(text_sql)
+        s = f"read_sql: '{text_sql}'"
         spinner = halo.Halo(s, spinner="dots", enabled=bool(logger))
         spinner.start()
         ts = pd.Timestamp.now()
@@ -180,7 +180,7 @@ def read_sql(
             chunksize=chunksize,
             nb_trials=nb_trials,
             logger=logger,
-            **kwargs,
+            **kwds,
         )
 
     if chunksize is None:
@@ -192,24 +192,21 @@ def read_sql(
             dfs.append(df)
             cnt += len(df)
             td = (pd.Timestamp.now() - ts).total_seconds() + 0.001
-            s = "{} rows ({} rows/sec)".format(cnt, cnt / td)
+            s = f"{cnt} rows ({cnt / td} rows/sec)"
             spinner.text = s
         df = pd.concat(dfs)
-        s = "{} rows".format(cnt)
+        s = f"{cnt} rows"
         spinner.succeed(s)
     except:
-        s = "{} rows".format(cnt)
+        s = f"{cnt} rows"
         spinner.fail(s)
         if logger:
             logger.warn_last_exception()
         if exception_handling == "raise":
             raise
         if exception_handling != "warn":
-            raise ValueError(
-                "Unknown value for argument 'exception_handling': '{}'.".format(
-                    exception_handling
-                )
-            )
+            msg = f"Unknown value for argument 'exception_handling': '{exception_handling}'."
+            raise ValueError(msg)
         df = pd.concat(dfs)
 
     return df
@@ -220,7 +217,7 @@ def read_sql_table(
     engine,
     nb_trials: int = 3,
     logger: tp.Optional[logg.IndentedLoggerAdapter] = None,
-    **kwargs,
+    **kwds,
 ):
     """Read an SQL table with a number of trials to overcome OperationalError.
 
@@ -246,7 +243,7 @@ def read_sql_table(
         engine,
         nb_trials=nb_trials,
         logger=logger,
-        **kwargs,
+        **kwds,
     )
 
 
@@ -256,7 +253,7 @@ def exec_sql(
     *args,
     nb_trials: int = 3,
     logger: tp.Optional[logg.IndentedLoggerAdapter] = None,
-    **kwargs,
+    **kwds,
 ):
     """Execute an SQL query with a number of trials to overcome OperationalError.
 
@@ -280,7 +277,7 @@ def exec_sql(
     """
 
     return run_func(
-        engine_execute, engine, sql, *args, nb_trials=nb_trials, logger=logger, **kwargs
+        engine_execute, engine, sql, *args, nb_trials=nb_trials, logger=logger, **kwds
     )
 
 
@@ -456,7 +453,7 @@ def temp_table_drop(
 
 
 @ctx.contextmanager
-def to_temp_table(df: pd.DataFrame, engine: sa.engine.Engine):
+def to_temp_table(df: pd.DataFrame, engine: sa.engine.Engine, **kwds):
     """
     A context manager that uploads a dataframe to a temp table and cleans up the table when done.
 
@@ -469,12 +466,14 @@ def to_temp_table(df: pd.DataFrame, engine: sa.engine.Engine):
         dataframe to be uploaded to the database as a temporary table
     engine : sqlalchemy.engine.Engine
         engine connrecting to the database
+    **kwds : dict
+        other keyword arguments passed as-is to :func:`pandas.DataFrame.to_sql`
     """
 
     tid = temp_table_find_new_id(engine)
     name = temp_table_name(tid)
     try:
-        df.to_sql(name, engine)
+        df.to_sql(name, engine, **kwds)
         yield name
     finally:
         temp_table_drop(engine, tid)

mt/sql/version.py

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

{mtsql-1.12.9.dist-info → mtsql-1.12.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mtsql
-Version: 1.12.9
+Version: 1.12.11
 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,7 +12,7 @@ Requires-Dist: tzlocal
 Requires-Dist: tqdm
 Requires-Dist: psycopg[binary]
 Requires-Dist: redshift_connector>=2.1.5
-Requires-Dist: mtbase>=4.32.20
+Requires-Dist: mtbase>=4.32.21
 Requires-Dist: mtpandas>=1.17.10
 Dynamic: author
 Dynamic: home-page

{mtsql-1.12.9.dist-info → mtsql-1.12.11.dist-info}/RECORD

@@ -1,17 +1,17 @@
 mt/sql/__init__.py,sha256=b7zO50apZxt9Hg2eOkJhRLrXgACR8eS5b-Rphdn5qNQ,44
-mt/sql/base.py,sha256=XxVmhunppBy6v_ungSP4rU6ko5LXrd15wA7JQfk1hP8,12974
+mt/sql/base.py,sha256=rEDlXivxH06ZsdpSXkraiqBCDDYh3FLVZfuH-spVw8w,12976
 mt/sql/mysql.py,sha256=n2ENDctdUqZuSaDAcrqZYtPtawq3Wx4dOPCRsCB5Q4w,4894
 mt/sql/psql.py,sha256=dNPOvJBxZ8S88uOmIKnfjo1ssRiMLDLwVeJcVcdcNco,66580
 mt/sql/sqlite.py,sha256=T2ak_hhNi_zRfpg_gp8JhNHn7D2kl4i-Ey6-9ANMtz0,8678
-mt/sql/version.py,sha256=se_nwA6cbwTpCLNvbZ6BMeTu4QBxOLCvjJ6FtbKwEsg,207
+mt/sql/version.py,sha256=w6SVgrZRGJU7ZkLGiZn_JhRKVxIF0tW5Z2yvN47_WVg,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=oomLiQib0iMhotRY3GjDrQI3iWGXgcIQyK5WTbmj8kc,53431
 mt/sql/redshift/main.py,sha256=6dwnwNJ1F0_V9o2oqrSOkyN_pAMrgE01CCoqAjoyOME,17116
 mt/sql/redshift/redshift-ca-bundle.crt,sha256=532qYkOpQOstFE0mdXE1GVtL3v00XDKgZNTr6gK5-KE,8621
-mtsql-1.12.9.dist-info/licenses/LICENSE,sha256=PojkRlQzTT5Eg6Nj03XoIVEefN3u8iiIFf1p4rqe_t4,1070
-mtsql-1.12.9.dist-info/METADATA,sha256=2QHOan3WIwjtg21WSO9uBZ1HFvYAtjK6SQ59G82qhfo,739
-mtsql-1.12.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-mtsql-1.12.9.dist-info/top_level.txt,sha256=WcqGFu9cV7iMZg09iam8eNxUvGpLSKKF2Iubf6SJVOo,3
-mtsql-1.12.9.dist-info/RECORD,,
+mtsql-1.12.11.dist-info/licenses/LICENSE,sha256=PojkRlQzTT5Eg6Nj03XoIVEefN3u8iiIFf1p4rqe_t4,1070
+mtsql-1.12.11.dist-info/METADATA,sha256=ieYqOgiWjVGvSbp1kjcql2In-R36c6bB3UXr9GzcS6k,740
+mtsql-1.12.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+mtsql-1.12.11.dist-info/top_level.txt,sha256=WcqGFu9cV7iMZg09iam8eNxUvGpLSKKF2Iubf6SJVOo,3
+mtsql-1.12.11.dist-info/RECORD,,