Add pony stubs (#14361)

Author: donBarbos

pyrightconfig.stricter.json

@@ -70,6 +70,7 @@
         "stubs/peewee",
         "stubs/pexpect",
         "stubs/pika",
+        "stubs/pony",
         "stubs/protobuf",
         "stubs/psutil",
         "stubs/psycopg2",

stubs/pony/@tests/stubtest_allowlist.txt

@@ -0,0 +1,14 @@
+# Tests should not be part of the stubs
+pony.orm.tests.*
+
+# Modules with ImportError, cannot import third-party libraries:
+pony.flask.*
+pony.orm.dbproviders.cockroach
+pony.orm.dbproviders.mysql
+pony.orm.dbproviders.oracle
+pony.orm.dbproviders.postgres
+pony.orm.integration.bottle_plugin
+pony.orm.examples.bottle_example
+
+# TODO: Incomplete issues in examples dir:
+pony.orm.examples.*

stubs/pony/METADATA.toml

@@ -0,0 +1,3 @@
+version = "0.7.*"
+upstream_repository = "https://github.com/ponyorm/pony"
+requires = ["types-psycopg2", "types-PyMySQL"]

stubs/pony/pony/__init__.pyi

@@ -0,0 +1,14 @@
+from typing import Final, Literal
+from typing_extensions import TypeAlias
+
+_Mode: TypeAlias = Literal[
+    "GAE-LOCAL", "GAE-SERVER", "MOD_WSGI", "INTERACTIVE", "FCGI-FLUP", "UWSGI", "FLASK", "CHERRYPY", "BOTTLE", "UNKNOWN"
+]
+__version__: Final[str]
+
+def detect_mode() -> _Mode: ...
+
+MODE: Final[_Mode]
+MAIN_FILE: Final[str | None]
+MAIN_DIR: Final[str | None]
+PONY_DIR: Final[str]

stubs/pony/pony/converting.pyi

@@ -0,0 +1,48 @@
+import re
+from _typeshed import ConvertibleToInt
+from collections.abc import Callable, Sequence
+from datetime import date, datetime, time, timedelta
+from typing import Any, Literal
+
+class ValidationError(ValueError): ...
+
+def check_ip(s: str) -> str: ...
+def check_positive(s: ConvertibleToInt) -> int: ...
+def check_identifier(s: str) -> str: ...
+
+isbn_re: re.Pattern[str]
+
+def isbn10_checksum(digits: Sequence[ConvertibleToInt]) -> str: ...
+def isbn13_checksum(digits: Sequence[ConvertibleToInt]) -> str: ...
+def check_isbn(s: str, convert_to: Literal[10, 13] | None = None) -> str: ...
+def isbn10_to_isbn13(s: str) -> str: ...
+def isbn13_to_isbn10(s: str) -> str: ...
+
+email_re: re.Pattern[str]
+rfc2822_email_re: re.Pattern[str]
+
+def check_email(s: str) -> str: ...
+def check_rfc2822_email(s: str) -> str: ...
+
+date_str_list: list[str]
+date_re_list: list[re.Pattern[str]]
+time_str: str
+time_re: re.Pattern[str]
+datetime_re_list: list[re.Pattern[str]]
+month_lists: list[list[str]]
+month_list: list[str]
+i: int
+month: str
+month_dict: dict[str, int]
+
+def str2date(s: str) -> date: ...
+def str2time(s: str) -> time: ...
+def str2datetime(s: str) -> datetime: ...
+def str2timedelta(s: str) -> timedelta: ...
+def timedelta2str(td: timedelta) -> str: ...
+
+converters: dict[type | str, tuple[Callable[[str], Any], type[str], str | None]]  # Any type from types above
+
+def str2py(
+    value: str, type: str | type | tuple[Callable[[str], Any], type[str], str | None] | None
+) -> Any: ...  # Any type from types above

stubs/pony/pony/flask/__init__.pyi

@@ -0,0 +1,17 @@
+from _typeshed import Incomplete
+from types import ModuleType
+from typing import Protocol
+
+# Protocol for flask.Flask class
+class _Flask(Protocol):
+    def before_request(self, f): ...
+    def after_request(self, f): ...
+    def teardown_request(self, f): ...
+
+flask_lib: ModuleType
+request: Incomplete
+
+class Pony:
+    app: _Flask | None
+    def __init__(self, app: _Flask | None = None) -> None: ...
+    def init_app(self, app: _Flask) -> None: ...

stubs/pony/pony/flask/example/__init__.pyi

@@ -0,0 +1,8 @@
+from _typeshed import Incomplete
+
+from pony.flask import _Flask
+
+app: _Flask
+login_manager: Incomplete
+
+def load_user(user_id): ...

stubs/pony/pony/flask/example/app.pyi

@@ -0,0 +1 @@
+config: dict[str, bool | str | dict[str, str | bool]]

stubs/pony/pony/flask/example/config.pyi

@@ -0,0 +1,10 @@
+from datetime import datetime
+
+from pony.orm.core import Database, Entity
+
+db: Database
+
+class User(Entity):
+    login: str
+    password: str
+    last_login: datetime | None