From 3a22bf2411c7357bffb2fe1dc675486a8da97177 Mon Sep 17 00:00:00 2001
From: kasium <15907922+kasium@users.noreply.github.com>
Date: Thu, 20 Jan 2022 22:10:26 +0100
Subject: [PATCH] Add stubs for Flask-SQLAlchemy (#6946)

---
 pyrightconfig.stricter.json                   |  1 +
 stubs/Flask-SQLAlchemy/METADATA.toml          |  1 +
 .../flask_sqlalchemy/__init__.pyi             | 88 +++++++++++++++++++
 .../flask_sqlalchemy/model.pyi                | 22 +++++
 .../flask_sqlalchemy/utils.pyi                |  3 +
 5 files changed, 115 insertions(+)
 create mode 100644 stubs/Flask-SQLAlchemy/METADATA.toml
 create mode 100644 stubs/Flask-SQLAlchemy/flask_sqlalchemy/__init__.pyi
 create mode 100644 stubs/Flask-SQLAlchemy/flask_sqlalchemy/model.pyi
 create mode 100644 stubs/Flask-SQLAlchemy/flask_sqlalchemy/utils.pyi

diff --git a/pyrightconfig.stricter.json b/pyrightconfig.stricter.json
index d890f5be2..eb179359a 100644
--- a/pyrightconfig.stricter.json
+++ b/pyrightconfig.stricter.json
@@ -31,6 +31,7 @@
         "stubs/cryptography",
         "stubs/dateparser",
         "stubs/docutils",
+        "stubs/Flask-SQLAlchemy",
         "stubs/fpdf2",
         "stubs/html5lib",
         "stubs/httplib2",
diff --git a/stubs/Flask-SQLAlchemy/METADATA.toml b/stubs/Flask-SQLAlchemy/METADATA.toml
new file mode 100644
index 000000000..43ea0392c
--- /dev/null
+++ b/stubs/Flask-SQLAlchemy/METADATA.toml
@@ -0,0 +1 @@
+version = "2.5.*"
diff --git a/stubs/Flask-SQLAlchemy/flask_sqlalchemy/__init__.pyi b/stubs/Flask-SQLAlchemy/flask_sqlalchemy/__init__.pyi
new file mode 100644
index 000000000..32cac5d92
--- /dev/null
+++ b/stubs/Flask-SQLAlchemy/flask_sqlalchemy/__init__.pyi
@@ -0,0 +1,88 @@
+from typing import Any
+
+from . import utils as utils
+from .model import DefaultMeta as DefaultMeta, Model as Model
+
+# SQLAlchemy is not part of typeshed
+_Query = Any
+_SessionBase = Any
+
+models_committed: Any
+before_models_committed: Any
+
+class SignallingSession(_SessionBase):
+    app: Any
+    def __init__(self, db, autocommit: bool = ..., autoflush: bool = ..., **options) -> None: ...
+    def get_bind(self, mapper: Any | None = ..., clause: Any | None = ...): ...
+
+def get_debug_queries(): ...
+
+class Pagination:
+    query: Any
+    page: Any
+    per_page: Any
+    total: Any
+    items: Any
+    def __init__(self, query, page, per_page, total, items) -> None: ...
+    @property
+    def pages(self): ...
+    def prev(self, error_out: bool = ...): ...
+    @property
+    def prev_num(self): ...
+    @property
+    def has_prev(self): ...
+    def next(self, error_out: bool = ...): ...
+    @property
+    def has_next(self): ...
+    @property
+    def next_num(self): ...
+    def iter_pages(
+        self, left_edge: int = ..., left_current: int = ..., right_current: int = ..., right_edge: int = ...
+    ) -> None: ...
+
+class BaseQuery(_Query):
+    def get_or_404(self, ident, description: Any | None = ...): ...
+    def first_or_404(self, description: Any | None = ...): ...
+    def paginate(
+        self, page: Any | None = ..., per_page: Any | None = ..., error_out: bool = ..., max_per_page: Any | None = ...
+    ): ...
+
+def get_state(app): ...
+
+class SQLAlchemy:
+    Query: Any
+    use_native_unicode: Any
+    session: Any
+    Model: Any
+    app: Any
+    def __init__(
+        self,
+        app: Any | None = ...,
+        use_native_unicode: bool = ...,
+        session_options: Any | None = ...,
+        metadata: Any | None = ...,
+        query_class=...,
+        model_class=...,
+        engine_options: Any | None = ...,
+    ) -> None: ...
+    @property
+    def metadata(self): ...
+    def create_scoped_session(self, options: Any | None = ...): ...
+    def create_session(self, options): ...
+    def make_declarative_base(self, model, metadata: Any | None = ...): ...
+    def init_app(self, app): ...
+    def apply_pool_defaults(self, app, options): ...
+    def apply_driver_hacks(self, app, sa_url, options): ...
+    @property
+    def engine(self): ...
+    def make_connector(self, app: Any | None = ..., bind: Any | None = ...): ...
+    def get_engine(self, app: Any | None = ..., bind: Any | None = ...): ...
+    def create_engine(self, sa_url, engine_opts): ...
+    def get_app(self, reference_app: Any | None = ...): ...
+    def get_tables_for_bind(self, bind: Any | None = ...): ...
+    def get_binds(self, app: Any | None = ...): ...
+    def create_all(self, bind: str = ..., app: Any | None = ...) -> None: ...
+    def drop_all(self, bind: str = ..., app: Any | None = ...) -> None: ...
+    def reflect(self, bind: str = ..., app: Any | None = ...) -> None: ...
+
+class FSADeprecationWarning(DeprecationWarning): ...
diff --git a/stubs/Flask-SQLAlchemy/flask_sqlalchemy/model.pyi b/stubs/Flask-SQLAlchemy/flask_sqlalchemy/model.pyi
new file mode 100644
index 000000000..7909be376
--- /dev/null
+++ b/stubs/Flask-SQLAlchemy/flask_sqlalchemy/model.pyi
@@ -0,0 +1,22 @@
+from typing import Any, Pattern
+
+_DeclarativeMeta = Any  # SQLAlchemy is not part of typeshed
+
+def should_set_tablename(cls: type) -> bool: ...
+
+camelcase_re: Pattern[str]
+
+def camel_to_snake_case(name: str) -> str: ...
+
+class NameMetaMixin(type):
+    def __init__(cls, name, bases, d) -> None: ...
+    def __table_cls__(cls, *args, **kwargs): ...
+
+class BindMetaMixin(type):
+    def __init__(cls, name, bases, d) -> None: ...
+
+class DefaultMeta(NameMetaMixin, BindMetaMixin, _DeclarativeMeta): ...
+
+class Model:
+    query_class: Any | None
+    query: Any | None
diff --git a/stubs/Flask-SQLAlchemy/flask_sqlalchemy/utils.pyi b/stubs/Flask-SQLAlchemy/flask_sqlalchemy/utils.pyi
new file mode 100644
index 000000000..190d72483
--- /dev/null
+++ b/stubs/Flask-SQLAlchemy/flask_sqlalchemy/utils.pyi
@@ -0,0 +1,3 @@
+def parse_version(v: str) -> tuple[int, int, int]: ...
+def sqlalchemy_version(op: str, val: str) -> bool: ...
+def engine_config_warning(config, version: str, deprecated_config_key: str, engine_option) -> None: ...