Ejemplo n.º 1
0
def connect(s3_staging_dir=None,
            region_name=None,
            schema_name='default',
            poll_interval=1,
            encryption_option=None,
            kms_key=None,
            profile_name=None,
            converter=None,
            formatter=None,
            retry_exceptions=('ThrottlingException',
                              'TooManyRequestsException'),
            retry_attempt=5,
            retry_multiplier=1,
            retry_max_delay=1800,
            retry_exponential_base=2,
            **kwargs):
    from pyathena.connection import Connection
    return Connection(s3_staging_dir, region_name, schema_name, poll_interval,
                      encryption_option, kms_key, profile_name, converter,
                      formatter, retry_exceptions, retry_attempt,
                      retry_multiplier, retry_max_delay,
                      retry_exponential_base, **kwargs)
Ejemplo n.º 2
0
def connect(*args, **kwargs) -> "Connection":
    from pyathena.connection import Connection

    return Connection(*args, **kwargs)
Ejemplo n.º 3
0
 def connect(self):
     return Connection(region_name=self.region_name,
                       s3_staging_dir=self.staging_location)