Fixed Mypy errors

Author: agronholm

.pre-commit-config.yaml

@@ -37,8 +37,7 @@ repos:
       - id: mypy
         additional_dependencies:
           - pytest
-          - sqlalchemy
-          - sqlalchemy2-stubs
+          - sqlalchemy[mypy]
 
   - repo: https://github.com/pre-commit/pygrep-hooks
     rev: v1.9.0

pyproject.toml

@@ -74,6 +74,7 @@ max_line_length = 99
 python_version = "3.7"
 strict = true
 ignore_missing_imports = true
+plugins = ["sqlalchemy.ext.mypy.plugin"]
 
 [tool.pytest.ini_options]
 addopts = "-rsx --tb=short"

src/sqlacodegen/cli.py

@@ -3,6 +3,7 @@
 import argparse
 import sys
 from contextlib import ExitStack
+from typing import TextIO
 
 from sqlalchemy.engine import create_engine
 from sqlalchemy.schema import MetaData
@@ -63,6 +64,7 @@ def main() -> None:
 
     # Open the target file (if given)
     with ExitStack() as stack:
+        outfile: TextIO
         if args.outfile:
             outfile = open(args.outfile, "w", encoding="utf-8")
             stack.enter_context(outfile)

src/sqlacodegen/utils.py

@@ -77,9 +77,9 @@ def uses_default_name(constraint: Constraint | Index) -> bool:
             values.update(
                 {
                     "column_0_name": constraint.columns[0].name,  # type: ignore[index]
-                    "column_0_label": constraint.columns[0]
+                    "column_0_label": constraint.columns[0]  # type: ignore[index]
                     .label(constraint.columns[0].name)  # type: ignore[index]
-                    .name,  # type: ignore[index]
+                    .name,
                     "column_0_key": constraint.columns[0].key,  # type: ignore[index]
                 }
             )