feat: Add mypy type checking

.pre-commit-config.yaml

@@ -40,5 +40,13 @@ repos:
       - id: poetry-check
         args: ['-C', 'api']
 
+  - repo: local
+    hooks:
+      - id: mypy-check
+        name: mypy check
+        entry: bash -c "cd api && poetry run python scripts/mypy_check.py"
+        language: system
+        types: [python]
+
 ci:
   autoupdate_commit_msg: 'ci: pre-commit autoupdate'

api/app/settings/common.py

@@ -543,7 +543,9 @@
 
 SECURE_REDIRECT_EXEMPT = env.list("DJANGO_SECURE_REDIRECT_EXEMPT", default=[])
 SECURE_REFERRER_POLICY = env.str("DJANGO_SECURE_REFERRER_POLICY", default="same-origin")
-SECURE_CROSS_ORIGIN_OPENER_POLICY = env.str("DJANGO_SECURE_CROSS_ORIGIN_OPENER_POLICY", default="same-origin")
+SECURE_CROSS_ORIGIN_OPENER_POLICY = env.str(
+    "DJANGO_SECURE_CROSS_ORIGIN_OPENER_POLICY", default="same-origin"
+)
 SECURE_SSL_HOST = env.str("DJANGO_SECURE_SSL_HOST", default=None)
 SECURE_SSL_REDIRECT = env.bool("DJANGO_SECURE_SSL_REDIRECT", default=False)
 

api/mypy.ini

@@ -0,0 +1,10 @@
+[mypy]
+exclude = ^(tests/)$
+plugins = mypy_django_plugin.main
+
+[mypy.plugins.django-stubs]
+django_settings_module = "app.settings.local"
+
+# Enable Django Rest Framework stubs.
+[mypy.plugins.drf-stubs]
+enabled = True

api/pyproject.toml

@@ -161,7 +161,7 @@ django-ordered-model = "~3.4.1"
 django-ses = "~3.5.0"
 django-axes = "~5.32.0"
 pydantic = "^2.3.0"
-pyngo = "~2.0.1"
+pyngo = "~2.2.1"
 flagsmith = "^3.6.0"
 python-gnupg = "^0.5.1"
 django-redis = "^5.4.0"
@@ -219,6 +219,10 @@ requests-mock = "^1.11.0"
 django-extensions = "^3.2.3"
 pdbpp = "^0.10.3"
 mypy-boto3-dynamodb = "^1.33.0"
+mypy = "^1.11.2"
+django-stubs = "~5.1.1"
+djangorestframework-stubs = "~3.14.0"
+boto3-stubs = "~1.35.67"
 
 [build-system]
 requires = ["poetry-core>=1.5.0"]

api/scripts/mypy_check.py

@@ -0,0 +1,52 @@
+import re
+import subprocess
+import sys
+from pathlib import Path
+
+
+def main():
+    baseline_path = Path("scripts/mypy_baseline.txt")
+    if not baseline_path.exists():
+        print("Baseline file not found. Run mypy and create mypy_baseline.txt first.")
+        sys.exit(1)
+
+    # Run mypy and capture output
+    result = subprocess.run(
+        ["poetry", "run", "mypy", "--config-file", "mypy.ini", "."],
+        capture_output=True,
+        text=True,
+    )
+    current_output = result.stdout.strip().splitlines()
+    if result.returncode != 1:
+        print("Error running mypy:", result.stderr)
+        sys.exit(1)
+
+    with open(baseline_path, "r") as f:
+        baseline = set(line.strip() for line in f if line.strip())
+
+    # Detect new errors and remove information line
+    current_errors = set(current_output)
+    new_errors = current_errors - baseline
+    pattern = r"Found (\d+) errors in (\d+) files \(checked (\d+) source files\)"
+    removal = None
+
+    for error in new_errors:
+        has_match = re.match(pattern, error)
+        if has_match:
+            removal = error
+            break
+
+    if removal:
+        new_errors.remove(removal)
+
+    if new_errors:
+        print("New mypy errors detected:")
+        print("\n".join(new_errors))
+        sys.exit(1)
+
+    print("No new mypy errors detected.")
+    sys.exit(0)
+
+
+if __name__ == "__main__":
+    main()