mtsql 1.12.30__py3-none-any.whl → 1.12.31__py3-none-any.whl

mt/sql/base.py

@@ -377,7 +377,7 @@ def table_exists(
 def create_temp_id_table(
     l_ids: list,
     conn: sa.engine.Connection,
-    int_type="int",
+    int_type: str = "int",
     chunksize: int = 1000000,
     logger: tp.Optional[logg.IndentedLoggerAdapter] = None,
 ) -> str:
@@ -504,6 +504,7 @@ def find_common_ids(
     engine: sa.engine.Engine,
     schema: tp.Optional[str] = None,
     id_col: str = "id",
+    int_type: str = "int",
     logger: tp.Optional[logg.IndentedLoggerAdapter] = None,
 ) -> tp.List[int]:
     """Finds common ids between a list of ids and the ids in a given frame.
@@ -520,6 +521,8 @@ def find_common_ids(
         schema of the frame. If None, the default schema is used.
     id_col : str
         name of the id column in the frame
+    int_type : str
+        an SQL string representing the int type of the id column
     logger : mt.logg.IndentedLoggerAdapter, optional
         logger for debugging
 
@@ -530,7 +533,7 @@ def find_common_ids(
     """
 
     with conn_ctx(engine) as conn:
-        temp_table = create_temp_id_table(l_ids, conn, logger=logger)
+        temp_table = create_temp_id_table(l_ids, conn, int_type=int_type, logger=logger)
 
         full_frame_name = frame_sql(frame_name, schema=schema)
 
@@ -553,6 +556,7 @@ def remove_records_by_id(
     engine: sa.engine.Engine,
     schema: tp.Optional[str] = None,
     id_col: str = "id",
+    int_type: str = "int",
     logger: tp.Optional[logg.IndentedLoggerAdapter] = None,
 ):
     """Removes records from a frame by a list of ids.
@@ -569,12 +573,14 @@ def remove_records_by_id(
         schema of the frame. If None, the default schema is used.
     id_col : str
         name of the id column in the frame
+    int_type : str
+        an SQL string representing the int type of the id column
     logger : mt.logg.IndentedLoggerAdapter, optional
         logger for debugging
     """
 
     with conn_ctx(engine) as conn:
-        temp_table = create_temp_id_table(l_ids, conn, logger=logger)
+        temp_table = create_temp_id_table(l_ids, conn, int_type=int_type, logger=logger)
 
         full_frame_name = frame_sql(frame_name, schema=schema)
 

mt/sql/version.py

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

{mtsql-1.12.30.dist-info → mtsql-1.12.31.dist-info}/METADATA

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

@@ -1,17 +1,17 @@
 mt/sql/__init__.py,sha256=b7zO50apZxt9Hg2eOkJhRLrXgACR8eS5b-Rphdn5qNQ,44
-mt/sql/base.py,sha256=AAoiWIpN8jyHdnfbmUvee6iU2pxo7pDcgO5dLrsybrY,16243
+mt/sql/base.py,sha256=CLFKd-rJtIUGLm37WtxeFLzhX48LgGpEoz6noR4qfXo,16510
 mt/sql/mysql.py,sha256=n2ENDctdUqZuSaDAcrqZYtPtawq3Wx4dOPCRsCB5Q4w,4894
 mt/sql/psql.py,sha256=U8XEyg4rQYr5gm8KohRWrpCNJKl5WC1yxJMBkkm1k_A,68125
 mt/sql/sqlite.py,sha256=T2ak_hhNi_zRfpg_gp8JhNHn7D2kl4i-Ey6-9ANMtz0,8678
-mt/sql/version.py,sha256=ATznhpd6AvkQUj3yf0sDQ4RYHO5ZWLiQdaKgfHdbxlE,208
+mt/sql/version.py,sha256=e7OodvUPDd0qLszM8zWrGxkmyhHFVodAR8B3CnSzs7g,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.30.dist-info/licenses/LICENSE,sha256=PojkRlQzTT5Eg6Nj03XoIVEefN3u8iiIFf1p4rqe_t4,1070
-mtsql-1.12.30.dist-info/METADATA,sha256=mxs0fj1WO1myL6Zlf3iTyPSDDe6Do4TnZtif48dPbms,734
-mtsql-1.12.30.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-mtsql-1.12.30.dist-info/top_level.txt,sha256=WcqGFu9cV7iMZg09iam8eNxUvGpLSKKF2Iubf6SJVOo,3
-mtsql-1.12.30.dist-info/RECORD,,
+mtsql-1.12.31.dist-info/licenses/LICENSE,sha256=PojkRlQzTT5Eg6Nj03XoIVEefN3u8iiIFf1p4rqe_t4,1070
+mtsql-1.12.31.dist-info/METADATA,sha256=bJET1fnqtrkAZ1Qz8UzM7c2E0PL1X65I5AN1h6wlY1Q,734
+mtsql-1.12.31.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+mtsql-1.12.31.dist-info/top_level.txt,sha256=WcqGFu9cV7iMZg09iam8eNxUvGpLSKKF2Iubf6SJVOo,3
+mtsql-1.12.31.dist-info/RECORD,,