add mypy.ini pyproject.toml (#64)

* add mypy.ini pyproject.toml

* fix contextmanager type

Author: koxudaxi

mypy.ini

@@ -0,0 +1,19 @@
+[mypy]
+plugins = pydantic.mypy
+
+ignore_missing_imports = True
+follow_imports = silent
+strict_optional = True
+warn_redundant_casts = True
+warn_unused_ignores = True
+disallow_any_generics = True
+check_untyped_defs = True
+no_implicit_reexport = True
+
+disallow_untyped_defs = True
+
+[pydantic-mypy]
+init_forbid_extra = True
+init_typed = True
+warn_required_dynamic_aliases = True
+warn_untyped_fields = True

pydataapi/dbapi.py

@@ -25,7 +25,7 @@
 """
 
 # https://docs.oracle.com/javase/8/docs/api/constant-values.html#java.sql.Types
-JDBC_TYPES: Dict[int, Type] = {
+JDBC_TYPES: Dict[int, Type[Any]] = {
     # 2003: list # ARRAY  2003
     -5: int,  # BIGINT -5
     # BINARY  -2
@@ -68,7 +68,7 @@
 }
 
 
-def get_description(column_metadata: List[Dict[str, Any]]) -> Tuple:
+def get_description(column_metadata: List[Dict[str, Any]]) -> Tuple[Any, ...]:
     return tuple(
         (
             meta['label'],  # name
@@ -170,9 +170,9 @@ def __init__(self, data_api: DataAPI) -> None:
 
         self.closed = False
 
-        self.description: Optional[List] = None
+        self.description: Optional[List[Any]] = None
 
-        self._rows: List[List] = []
+        self._rows: List[List[Any]] = []
         self._rowcount: int = -1
         self._lastrowid: Optional[int] = None
 
@@ -195,7 +195,7 @@ def execute(
         self.description = get_description(  # type: ignore
             getattr(result, '_column_metadata')
         )
-        rows: List[List] = getattr(result, '_rows')
+        rows: List[List[Any]] = getattr(result, '_rows')
         self._rows = rows
         self._rowcount = len(rows) or result.number_of_records_updated
         self._lastrowid = result.generated_fields_first  # type: ignore
@@ -209,23 +209,23 @@ def executemany(
         self._rows = [result.generated_fields for result in results]
         self._rowcount = len(self._rows)
         self.description = []
-        self._lastrowid = (  # type: ignore
+        self._lastrowid = (
             results[-1].generated_fields_first if results else None  # type: ignore
         )
         return self
 
-    def fetchone(self) -> Optional[List]:
+    def fetchone(self) -> Optional[List[Any]]:
         try:
             return self._rows.pop(0)
         except IndexError:
             return None
 
-    def fetchmany(self, size: Optional[int] = None) -> List[List]:
+    def fetchmany(self, size: Optional[int] = None) -> List[List[Any]]:
         size = size or self.arraysize
         result, self._rows = self._rows[:size], self._rows[size:]
         return result
 
-    def fetchall(self) -> List[List]:
+    def fetchall(self) -> List[List[Any]]:
         rows = self._rows
         self._rows = []
         return rows
@@ -236,7 +236,7 @@ def setinputsizes(self, sizes: Any) -> None:  # pragma: no cover
     def setoutputsizes(self, sizes: Any) -> None:  # pragma: no cover
         pass
 
-    def __iter__(self) -> Iterator[List]:
+    def __iter__(self) -> Iterator[List[Any]]:
         return iter(self._rows)
 
 

pydataapi/dialect/base.py

@@ -4,12 +4,13 @@
 from typing import Any, Callable, List, Optional, Pattern, Type, TypeVar, Union
 
 from botocore.exceptions import ClientError
-from pydataapi.dbapi import Connection
 from sqlalchemy import cast
 from sqlalchemy.engine.default import DefaultDialect
 from sqlalchemy.sql import sqltypes
 from sqlalchemy.sql.type_api import TypeEngine
 
+from pydataapi.dbapi import Connection
+
 
 class DataAPIDialect(DefaultDialect, ABC):
     driver: str = 'dataapi'
@@ -41,12 +42,12 @@ def dbapi(cls) -> Type[Connection]:
 
 DatetimeProtocol = Union[datetime.date, datetime.datetime, datetime.time]
 
-DATE_PATTERN: Pattern = re.compile(r'^\d{4}-[0-1]\d-[0-3]\d$')
+DATE_PATTERN: Pattern[str] = re.compile(r'^\d{4}-[0-1]\d-[0-3]\d$')
 
-DATETIME_PATTERN: Pattern = re.compile(
+DATETIME_PATTERN: Pattern[str] = re.compile(
     r'^\d{4}-[0-1]\d-[0-3]\d [0-2]\d:[0-6]\d:[0-6]\d$'
 )
-DATETIME_MICROSECOND_PATTERN: Pattern = re.compile(
+DATETIME_MICROSECOND_PATTERN: Pattern[str] = re.compile(
     r'^\d{4}-[0-1]\d-[0-3]\d [0-2]\d:[0-6]\d:[0-6]\d\.\d{1,6}$'
 )
 
@@ -75,15 +76,15 @@ class DataAPIDatetimeBase:
     def bind_expression(self, value: Any) -> Any:
         return cast(value, self.db_type)
 
-    def bind_processor(self, dialect: DataAPIDialect) -> Callable:
+    def bind_processor(self, dialect: DataAPIDialect) -> Callable[..., Any]:
         def process_bind_value(value: Any) -> Any:
             if isinstance(value, self.python_type):
                 return value.strftime(DATETIME_MICROSECOND_FORMAT)
             return value
 
         return process_bind_value
 
-    def result_processor(self, dialect: DataAPIDialect, coltype: List) -> Any:
+    def result_processor(self, dialect: DataAPIDialect, coltype: List[Any]) -> Any:
         def process_result_value(value: Any) -> Any:
             parsed_datetime = _parse_datetime(value)
             if parsed_datetime: