From 27a9d06b30aae800ccc712928813f0685ea06021 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Edgar=20Ram=C3=ADrez=20Mondrag=C3=B3n?=
 <edgarrm358@gmail.com>
Date: Tue, 29 Aug 2023 10:07:15 -0600
Subject: [PATCH] Add SQLAlchemy stubs

---
 poetry.lock    | 18 +++++++++++++++++-
 pyproject.toml |  2 ++
 2 files changed, 19 insertions(+), 1 deletion(-)

diff --git a/poetry.lock b/poetry.lock
index 0e386c5..3b2eb1b 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -1626,6 +1626,8 @@ files = [
 [package.dependencies]
 greenlet = {version = "!=0.4.17", markers = "python_version >= \"3\" and (platform_machine == \"aarch64\" or platform_machine == \"ppc64le\" or platform_machine == \"x86_64\" or platform_machine == \"amd64\" or platform_machine == \"AMD64\" or platform_machine == \"win32\" or platform_machine == \"WIN32\")"}
 importlib-metadata = {version = "*", markers = "python_version < \"3.8\""}
+mypy = {version = ">=0.910", optional = true, markers = "python_version >= \"3\" and extra == \"mypy\""}
+sqlalchemy2-stubs = {version = "*", optional = true, markers = "extra == \"mypy\""}
 
 [package.extras]
 aiomysql = ["aiomysql", "greenlet (!=0.4.17)"]
@@ -1648,6 +1650,20 @@ postgresql-psycopg2cffi = ["psycopg2cffi"]
 pymysql = ["pymysql", "pymysql (<1)"]
 sqlcipher = ["sqlcipher3-binary"]
 
+[[package]]
+name = "sqlalchemy2-stubs"
+version = "0.0.2a35"
+description = "Typing Stubs for SQLAlchemy 1.4"
+optional = false
+python-versions = ">=3.6"
+files = [
+    {file = "sqlalchemy2-stubs-0.0.2a35.tar.gz", hash = "sha256:bd5d530697d7e8c8504c7fe792ef334538392a5fb7aa7e4f670bfacdd668a19d"},
+    {file = "sqlalchemy2_stubs-0.0.2a35-py3-none-any.whl", hash = "sha256:593784ff9fc0dc2ded1895e3322591689db3be06f3ca006e3ef47640baf2d38a"},
+]
+
+[package.dependencies]
+typing-extensions = ">=3.7.4"
+
 [[package]]
 name = "tenacity"
 version = "8.2.2"
@@ -1804,4 +1820,4 @@ s3 = []
 [metadata]
 lock-version = "2.0"
 python-versions = "<3.12,>=3.7.1"
-content-hash = "5810085cc4224f60eeac33c930ce166898434e6795a7513d49305a3ded949138"
+content-hash = "8fb01c131ccd8260c4d72e57ee63c67254ea62d37db85bb3496fd1176234241f"
diff --git a/pyproject.toml b/pyproject.toml
index e973ca5..a05b2a9 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -30,6 +30,7 @@ isort = "^5.11.5"
 moto = "^4.1.6"
 coverage = "^7.2.3"
 pydocstyle = "^6.3.0"
+sqlalchemy = {version="~=1.4", extras=["mypy"]}
 
 [tool.poetry.extras]
 s3 = ["fs-s3fs"]
@@ -40,6 +41,7 @@ multi_line_output = 3 # Vertical Hanging Indent
 src_paths = "tap_dynamodb"
 
 [tool.mypy]
+plugins = ["sqlalchemy.ext.mypy.plugin"]
 python_version = "3.9"
 warn_unused_configs = true