mtsql 1.12.7__py3-none-any.whl → 1.12.8__py3-none-any.whl

mt/sql/base.py

@@ -6,8 +6,7 @@ import sqlalchemy as sa
 import sqlalchemy.exc as se
 import psycopg as ps
 
-from mt import tp, logg, pd, ctx
-from mt.halo import Halo
+from mt import tp, logg, pd, ctx, halo
 
 
 __all__ = [
@@ -27,6 +26,7 @@ __all__ = [
     "temp_table_name",
     "temp_table_find_new_id",
     "temp_table_drop",
+    "to_temp_table",
 ]
 
 
@@ -166,7 +166,7 @@ def read_sql(
 
     if chunksize is not None:
         s = "read_sql: '{}'".format(text_sql)
-        spinner = Halo(s, spinner="dots", enabled=bool(logger))
+        spinner = halo.Halo(s, spinner="dots", enabled=bool(logger))
         spinner.start()
         ts = pd.Timestamp.now()
         cnt = 0
@@ -453,3 +453,28 @@ def temp_table_drop(
     name = id if isinstance(id, str) else temp_table_name(id)
     sql = f"DROP TABLE IF EXISTS {name}"
     return engine_execute(engine, sql)
+
+
+@ctx.contextmanager
+def to_temp_table(df: pd.DataFrame, engine: sa.engine.Engine):
+    """
+    A context manager that uploads a dataframe to a temp table and cleans up the table when done.
+
+    You can use the class in a with statement to work with the temp table, whose name is returned
+    as the context manager object.
+
+    Parameters
+    ----------
+    df : pandas.DataFrame
+        dataframe to be uploaded to the database as a temporary table
+    engine : sqlalchemy.engine.Engine
+        engine connrecting to the database
+    """
+
+    temp_table_id = temp_table_find_new_id(engine)
+    temp_table_name = temp_table_name(temp_table_id)
+    try:
+        df.to_sql(temp_table_name, engine)
+        yield temp_table_name
+    finally:
+        temp_table_drop(engine, temp_table_id)

mt/sql/version.py

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

{mtsql-1.12.7.dist-info → mtsql-1.12.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mtsql
-Version: 1.12.7
+Version: 1.12.8
 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.31.5
-Requires-Dist: mtpandas>=1.16.8
+Requires-Dist: mtbase>=4.32.20
+Requires-Dist: mtpandas>=1.17.10
 Dynamic: author
 Dynamic: home-page
 Dynamic: license-file

{mtsql-1.12.7.dist-info → mtsql-1.12.8.dist-info}/RECORD

@@ -1,17 +1,17 @@
 mt/sql/__init__.py,sha256=b7zO50apZxt9Hg2eOkJhRLrXgACR8eS5b-Rphdn5qNQ,44
-mt/sql/base.py,sha256=CPrI28CZcEEp3cBQBg5_9APotlCiF3U6sg3moeUKZVM,12242
+mt/sql/base.py,sha256=fqNA0gDeUKYdMGPOIo33y_E2s-t0sBNtoR2hYxTdJGM,13037
 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=TZkj7gDWCVgkQFfdQQxVsySbeb72IGyS62UsT1inxS4,207
+mt/sql/version.py,sha256=9cPEw4vxFA-JdEQQR7uevWv6fRYCk_Vbs4X79gmGUzg,207
 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.7.dist-info/licenses/LICENSE,sha256=PojkRlQzTT5Eg6Nj03XoIVEefN3u8iiIFf1p4rqe_t4,1070
-mtsql-1.12.7.dist-info/METADATA,sha256=27DDVa8knElWPBe05BX1GJR2dPhLZ0MvYVULygmMNyg,737
-mtsql-1.12.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-mtsql-1.12.7.dist-info/top_level.txt,sha256=WcqGFu9cV7iMZg09iam8eNxUvGpLSKKF2Iubf6SJVOo,3
-mtsql-1.12.7.dist-info/RECORD,,
+mtsql-1.12.8.dist-info/licenses/LICENSE,sha256=PojkRlQzTT5Eg6Nj03XoIVEefN3u8iiIFf1p4rqe_t4,1070
+mtsql-1.12.8.dist-info/METADATA,sha256=3psQ6918Qf0xhB5-133fvGLzVBZc9eo6Xoj3kjzm4IA,739
+mtsql-1.12.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+mtsql-1.12.8.dist-info/top_level.txt,sha256=WcqGFu9cV7iMZg09iam8eNxUvGpLSKKF2Iubf6SJVOo,3
+mtsql-1.12.8.dist-info/RECORD,,