Support pandas.to_sql

Author: laughingman7743

Pipfile.lock

@@ -95,7 +95,8 @@ def __init__(
 
         self._start_jvm(jvm_path, jvm_options, driver_path, log4j_conf)
 
-        props = self._build_driver_args(**driver_kwargs)
+        self._driver_kwargs = driver_kwargs
+        props = self._build_driver_args()
         jpype.JClass(ATHENA_DRIVER_CLASS_NAME)
         self._jdbc_conn = jpype.java.sql.DriverManager.getConnection(
             ATHENA_CONNECTION_STRING.format(
@@ -147,7 +148,7 @@ def _start_jvm(cls, jvm_path, jvm_options, driver_path, log4j_conf):
             )
             jpype.java.lang.Thread.currentThread().setContextClassLoader(class_loader)
 
-    def _build_driver_args(self, **kwargs):
+    def _build_driver_args(self):
         props = jpype.java.util.Properties()
         if self.credential_file:
             props.setProperty(
@@ -176,7 +177,7 @@ def _build_driver_args(self, **kwargs):
             props.setProperty("S3OutputLocation", self.s3_staging_dir)
         if self.work_group:
             props.setProperty("Workgroup", self.work_group)
-        for k, v in iteritems(kwargs):
+        for k, v in iteritems(self._driver_kwargs):
             if k and v:
                 props.setProperty(k, v)
         return props

pyathenajdbc/connection.py

@@ -129,7 +129,10 @@ def execute(self, operation, parameters=None):
             raise_from(DatabaseError(e), e)
 
     def executemany(self, operation, seq_of_parameters):
-        raise NotSupportedError
+        for parameters in seq_of_parameters:
+            self.execute(operation, parameters)
+        # Operations that have result sets are not allowed with executemany.
+        self._reset_state()
 
     @attach_thread_to_jvm
     @synchronized