Accept sqlalchemy.engine.Engine

Author: wakabame

pandas-stubs/core/generic.pyi

@@ -160,7 +160,7 @@ class NDFrame(PandasObject, indexing.IndexingMixin):
     def to_sql(
         self,
         name: _str,
-        con: str | sqlalchemy.engine.Connection | sqlite3.Connection,
+        con: str | sqlalchemy.engine.Connectable | sqlite3.Connection,
         schema: _str | None = ...,
         if_exists: Literal["fail", "replace", "append"] = ...,
         index: _bool = ...,

pandas-stubs/io/sql.pyi

@@ -23,7 +23,7 @@ class DatabaseError(IOError): ...
 @overload
 def read_sql_table(
     table_name: str,
-    con: str | sqlalchemy.engine.Connection | sqlite3.Connection,
+    con: str | sqlalchemy.engine.Connectable | sqlite3.Connection,
     schema: str | None = ...,
     index_col: str | list[str] | None = ...,
     coerce_float: bool = ...,
@@ -35,7 +35,7 @@ def read_sql_table(
 @overload
 def read_sql_table(
     table_name: str,
-    con: str | sqlalchemy.engine.Connection | sqlite3.Connection,
+    con: str | sqlalchemy.engine.Connectable | sqlite3.Connection,
     schema: str | None = ...,
     index_col: str | list[str] | None = ...,
     coerce_float: bool = ...,
@@ -46,7 +46,7 @@ def read_sql_table(
 @overload
 def read_sql_query(
     sql: str,
-    con: str | sqlalchemy.engine.Connection | sqlite3.Connection,
+    con: str | sqlalchemy.engine.Connectable | sqlite3.Connection,
     index_col: str | list[str] | None = ...,
     coerce_float: bool = ...,
     params: list[str] | tuple[str, ...] | dict[str, str] | None = ...,
@@ -58,7 +58,7 @@ def read_sql_query(
 @overload
 def read_sql_query(
     sql: str,
-    con: str | sqlalchemy.engine.Connection | sqlite3.Connection,
+    con: str | sqlalchemy.engine.Connectable | sqlite3.Connection,
     index_col: str | list[str] | None = ...,
     coerce_float: bool = ...,
     params: list[str] | tuple[str, ...] | dict[str, str] | None = ...,
@@ -69,7 +69,7 @@ def read_sql_query(
 @overload
 def read_sql(
     sql: str,
-    con: str | sqlalchemy.engine.Connection | sqlite3.Connection,
+    con: str | sqlalchemy.engine.Connectable | sqlite3.Connection,
     index_col: str | list[str] | None = ...,
     coerce_float: bool = ...,
     params: list[str] | tuple[str, ...] | dict[str, str] | None = ...,
@@ -81,7 +81,7 @@ def read_sql(
 @overload
 def read_sql(
     sql: str,
-    con: str | sqlalchemy.engine.Connection | sqlite3.Connection,
+    con: str | sqlalchemy.engine.Connectable | sqlite3.Connection,
     index_col: str | list[str] | None = ...,
     coerce_float: bool = ...,
     params: list[str] | tuple[str, ...] | dict[str, str] | None = ...,