chore: init repository

Author: Brooke-white

.pre-commit-config.yaml

@@ -0,0 +1,16 @@
+repos:
+  - repo: https://github.com/pre-commit/pre-commit-hooks
+    rev: v2.3.0
+    hooks:
+      - id: check-yaml
+      - id: end-of-file-fixer
+      - id: trailing-whitespace
+  - repo: https://github.com/pre-commit/mirrors-isort
+    rev: v5.1.0 # must be >5.0.0 for black compatibility
+    hooks:
+      - id: isort
+        args: ["--profile", "black", "."]
+  - repo: https://github.com/ambv/black
+    rev: stable
+    hooks:
+      - id: black

build.sh

@@ -1,2 +1 @@
-pip3 install setuptools --user
-python3 setup.py install --user
+python3 setup.py sdist bdist_wheel

redshift_connector/CredentialsHolder.py

@@ -1,8 +1,8 @@
 import logging
 import os
-from typing import Any, Dict, List, Optional
+import typing
 
-from redshift_connector.core import BINARY, Connection, Cursor, Interval
+from redshift_connector.core import BINARY, Connection, Cursor
 from redshift_connector.error import (
     ArrayContentNotHomogenousError,
     ArrayContentNotSupportedError,
@@ -18,7 +18,7 @@
     ProgrammingError,
     Warning,
 )
-from redshift_connector.iamHelper import set_iam_properties
+from redshift_connector.iam_helper import set_iam_properties
 from redshift_connector.objects import (
     Binary,
     Date,
@@ -36,14 +36,12 @@
     PGTsvector,
     PGVarchar,
 )
-from redshift_connector.RedshiftProperty import RedshiftProperty
+from redshift_connector.redshift_property import RedshiftProperty
 
-from ._version import get_versions
+from .version import __version__
 
-__version__ = get_versions()['version']
-del get_versions
-path = os.path.abspath(__file__)
-log_path = '/'.join(path.split('/')[:-1]) + '/driver.log'
+path: str = os.path.abspath(__file__)
+log_path: str = "/".join(path.split("/")[:-1]) + "/driver.log"
 
 # Copyright (c) 2007-2009, Mathieu Fenniak
 # Copyright (c) The Contributors
@@ -77,62 +75,112 @@
 
 
 def connect(
-        user: Optional[str], host: str = 'localhost', database: Optional[str] = None,
-        port: int = 5432, password: Optional[str] = None,
-        source_address: Optional[str] = None, unix_sock: Optional[str] = None,
-        ssl: bool = True, sslmode: str = "verify-ca", timeout: Optional[int] = None,
-        max_prepared_statements: int = 1000, tcp_keepalive: bool = True,
-        application_name: Optional[str] = None, replication: Optional[str] = None,
-        idp_host: Optional[str] = None, db_user: Optional[str] = None,
-        app_id: Optional[str] = None, app_name: str = 'amazon_aws_redshift',
-        preferred_role: Optional[str] = None, principal_arn: Optional[str] = None,
-        credentials_provider: Optional[str] = None, region: Optional[str] = None,
-        cluster_identifier: Optional[str] = None, iam: bool = False,
-        client_id: Optional[str] = None, idp_tenant: Optional[str] = None,
-        client_secret: Optional[str] = None, partner_sp_id: Optional[str] = None,
-        idp_response_timeout: int = 120, listen_port: int = 7890,
-        login_url: Optional[str] = None, auto_create: bool = False,
-        db_groups: Optional[List[str]] = None, force_lowercase: bool = False,
-        allow_db_user_override: bool = False, dsilog_level: int = 0,
-        log_path: str = log_path
+    user: str,
+    database: str,
+    password: str,
+    port: int = 5439,
+    host: str = "localhost",
+    source_address: typing.Optional[str] = None,
+    unix_sock: typing.Optional[str] = None,
+    ssl: bool = True,
+    sslmode: str = "verify-ca",
+    timeout: typing.Optional[int] = None,
+    max_prepared_statements: int = 1000,
+    tcp_keepalive: bool = True,
+    application_name: typing.Optional[str] = None,
+    replication: typing.Optional[str] = None,
+    idp_host: typing.Optional[str] = None,
+    db_user: typing.Optional[str] = None,
+    app_id: typing.Optional[str] = None,
+    app_name: str = "amazon_aws_redshift",
+    preferred_role: typing.Optional[str] = None,
+    principal_arn: typing.Optional[str] = None,
+    credentials_provider: typing.Optional[str] = None,
+    region: typing.Optional[str] = None,
+    cluster_identifier: typing.Optional[str] = None,
+    iam: bool = False,
+    client_id: typing.Optional[str] = None,
+    idp_tenant: typing.Optional[str] = None,
+    client_secret: typing.Optional[str] = None,
+    partner_sp_id: typing.Optional[str] = None,
+    idp_response_timeout: int = 120,
+    listen_port: int = 7890,
+    login_url: typing.Optional[str] = None,
+    auto_create: bool = False,
+    db_groups: typing.Optional[typing.List[str]] = None,
+    force_lowercase: bool = False,
+    allow_db_user_override: bool = False,
+    log_level: int = 0,
+    log_path: str = log_path,
 ) -> Connection:
 
-    FORMAT_TO_USE: str = (
-        "%(levelname)s|%(asctime)s|%(name)s|%(filename)s|" "%(funcName)s|%(lineno)d: %(message)s"
-    )
-    log_level_dic: Dict[int, int] = {
+    FORMAT_TO_USE: str = "%(levelname)s|%(asctime)s|%(name)s|%(filename)s|" "%(funcName)s|%(lineno)d: %(message)s"
+    log_level_dic: typing.Dict[int, int] = {
         0: logging.CRITICAL,
         1: logging.ERROR,
         2: logging.WARN,
         3: logging.INFO,
-        4: logging.DEBUG
+        4: logging.DEBUG,
     }
-    logging.basicConfig(filename=log_path, filemode='w', format=FORMAT_TO_USE, level=logging.INFO)
-    logging.disable(log_level_dic[dsilog_level])
+    logging.basicConfig(filename=log_path, filemode="w", format=FORMAT_TO_USE, level=logging.INFO)
+    logging.disable(log_level_dic[log_level])
 
     info: RedshiftProperty = RedshiftProperty()
-    set_iam_properties(info, user, host=host, database=database, port=port, password=password,
-                       source_address=source_address, unix_sock=unix_sock,
-                       ssl=ssl, sslmode=sslmode, timeout=timeout,
-                       max_prepared_statements=max_prepared_statements,
-                       tcp_keepalive=tcp_keepalive, application_name=application_name,
-                       replication=replication, idp_host=idp_host, db_user=db_user,
-                       app_id=app_id, app_name=app_name, preferred_role=preferred_role, principal_arn=principal_arn,
-                       credentials_provider=credentials_provider,
-                       region=region, cluster_identifier=cluster_identifier, iam=iam,
-                       client_id=client_id, idp_tenant=idp_tenant, client_secret=client_secret,
-                       partner_sp_id=partner_sp_id, idp_response_timeout=idp_response_timeout, listen_port=listen_port,
-                       login_url=login_url, auto_create=auto_create, db_groups=db_groups,
-                       force_lowercase=force_lowercase, allow_db_user_override=allow_db_user_override)
+    set_iam_properties(
+        info,
+        user=user,
+        host=host,
+        database=database,
+        port=port,
+        password=password,
+        source_address=source_address,
+        unix_sock=unix_sock,
+        ssl=ssl,
+        sslmode=sslmode,
+        timeout=timeout,
+        max_prepared_statements=max_prepared_statements,
+        tcp_keepalive=tcp_keepalive,
+        application_name=application_name,
+        replication=replication,
+        idp_host=idp_host,
+        db_user=db_user,
+        app_id=app_id,
+        app_name=app_name,
+        preferred_role=preferred_role,
+        principal_arn=principal_arn,
+        credentials_provider=credentials_provider,
+        region=region,
+        cluster_identifier=cluster_identifier,
+        iam=iam,
+        client_id=client_id,
+        idp_tenant=idp_tenant,
+        client_secret=client_secret,
+        partner_sp_id=partner_sp_id,
+        idp_response_timeout=idp_response_timeout,
+        listen_port=listen_port,
+        login_url=login_url,
+        auto_create=auto_create,
+        db_groups=db_groups,
+        force_lowercase=force_lowercase,
+        allow_db_user_override=allow_db_user_override,
+    )
 
     return Connection(
-        user=info.user_name, host=info.host, database=info.db_name,
-        port=info.port, password=info.password,
-        source_address=info.source_address, unix_sock=info.unix_sock,
-        ssl=info.ssl, sslmode=info.sslmode, timeout=info.timeout,
+        user=info.user_name,
+        host=info.host,
+        database=info.db_name,
+        port=info.port,
+        password=info.password,
+        source_address=info.source_address,
+        unix_sock=info.unix_sock,
+        ssl=info.ssl,
+        sslmode=info.sslmode,
+        timeout=info.timeout,
         max_prepared_statements=info.max_prepared_statements,
-        tcp_keepalive=info.tcp_keepalive, application_name=info.application_name,
-        replication=info.replication)
+        tcp_keepalive=info.tcp_keepalive,
+        application_name=info.application_name,
+        replication=info.replication,
+    )
 
 
 apilevel: str = "2.0"
@@ -151,7 +199,7 @@ def connect(
 <http://www.python.org/dev/peps/pep-0249/>`_.
 """
 
-paramstyle: str = 'format'
+paramstyle: str = "format"
 
 # I have no idea what this would be used for by a client app.  Should it be
 # TEXT, VARCHAR, CHAR?  It will only compare against row_description's
@@ -162,20 +210,44 @@ def connect(
 """String type oid."""
 
 
-NUMBER: int  = 1700
+NUMBER: int = 1700
 """Numeric type oid"""
 
-DATETIME: int  = 1114
+DATETIME: int = 1114
 """Timestamp type oid"""
 
-ROWID: int  = 26
+ROWID: int = 26
 """ROWID type oid"""
 
-__all__: Any = [
-    Warning, DataError, DatabaseError, connect, InterfaceError,
-    ProgrammingError, Error, OperationalError, IntegrityError, InternalError,
-    NotSupportedError, ArrayContentNotHomogenousError,
-    ArrayDimensionsNotConsistentError, ArrayContentNotSupportedError,
-    Connection, Cursor, Binary, Date, DateFromTicks, Time, TimeFromTicks,
-    Timestamp, TimestampFromTicks, BINARY, Interval, PGEnum, PGJson, PGJsonb,
-    PGTsvector, PGText, PGVarchar]
+__all__: typing.Any = [
+    "Warning",
+    "DataError",
+    "DatabaseError",
+    "connect",
+    "InterfaceError",
+    "ProgrammingError",
+    "Error",
+    "OperationalError",
+    "IntegrityError",
+    "InternalError",
+    "NotSupportedError",
+    "ArrayContentNotHomogenousError",
+    "ArrayDimensionsNotConsistentError",
+    "ArrayContentNotSupportedError",
+    "Connection",
+    "Cursor",
+    "Binary",
+    "Date",
+    "DateFromTicks",
+    "Time",
+    "TimeFromTicks",
+    "Timestamp",
+    "TimestampFromTicks",
+    "BINARY",
+    "PGEnum",
+    "PGJson",
+    "PGJsonb",
+    "PGTsvector",
+    "PGText",
+    "PGVarchar",
+]