From 9a76e5a66938e8a1f784116dcf32fa17a5e61ba1 Mon Sep 17 00:00:00 2001
From: Alex Waygood <Alex.Waygood@Gmail.com>
Date: Fri, 13 Jan 2023 00:24:13 +0000
Subject: [PATCH] `Flask-Migrate`: Add `Flask` as a dependency (#9517)

---
 stubs/Flask-Migrate/METADATA.toml              | 3 ++-
 stubs/Flask-Migrate/flask_migrate/__init__.pyi | 6 +++---
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/stubs/Flask-Migrate/METADATA.toml b/stubs/Flask-Migrate/METADATA.toml
index 481bd5554..ebdf90e18 100644
--- a/stubs/Flask-Migrate/METADATA.toml
+++ b/stubs/Flask-Migrate/METADATA.toml
@@ -1,2 +1,3 @@
 version = "4.0.*"
-requires = ["types-Flask-SQLAlchemy"]
+# Requires a version of flask with a `py.typed` file
+requires = ["Flask>=2.0.0", "types-Flask-SQLAlchemy"]
diff --git a/stubs/Flask-Migrate/flask_migrate/__init__.pyi b/stubs/Flask-Migrate/flask_migrate/__init__.pyi
index b52bb29e8..6a68c7c6f 100644
--- a/stubs/Flask-Migrate/flask_migrate/__init__.pyi
+++ b/stubs/Flask-Migrate/flask_migrate/__init__.pyi
@@ -3,11 +3,11 @@ from logging import Logger
 from typing import Any, TypeVar
 from typing_extensions import ParamSpec, TypeAlias
 
+import flask
 from flask_sqlalchemy import SQLAlchemy
 
 _T = TypeVar("_T")
 _P = ParamSpec("_P")
-_App: TypeAlias = Any  # flask.Flask is not possible as a dependency yet
 _ConfigureCallback: TypeAlias = Callable[[Config], Config]
 
 alembic_version: tuple[int, int, int]
@@ -25,7 +25,7 @@ class Migrate:
     alembic_ctx_kwargs: dict[str, Any]
     def __init__(
         self,
-        app: _App | None = ...,
+        app: flask.Flask | None = ...,
         db: SQLAlchemy | None = ...,
         directory: str = ...,
         command: str = ...,
@@ -35,7 +35,7 @@ class Migrate:
     ) -> None: ...
     def init_app(
         self,
-        app: _App,
+        app: flask.Flask,
         db: SQLAlchemy | None = ...,
         directory: str | None = ...,
         command: str | None = ...,