diff --git a/pyrightconfig.stricter.json b/pyrightconfig.stricter.json index 2f41cc1de..803af495a 100644 --- a/pyrightconfig.stricter.json +++ b/pyrightconfig.stricter.json @@ -30,6 +30,7 @@ "stubs/commonmark", "stubs/cryptography", "stubs/dateparser", + "stubs/dj-database-url", "stubs/docutils", "stubs/Flask-SQLAlchemy", "stubs/fpdf2", diff --git a/stubs/dj-database-url/METADATA.toml b/stubs/dj-database-url/METADATA.toml new file mode 100644 index 000000000..1883b8701 --- /dev/null +++ b/stubs/dj-database-url/METADATA.toml @@ -0,0 +1 @@ +version = "0.5.*" diff --git a/stubs/dj-database-url/dj_database_url.pyi b/stubs/dj-database-url/dj_database_url.pyi new file mode 100644 index 000000000..7c9a69a6f --- /dev/null +++ b/stubs/dj-database-url/dj_database_url.pyi @@ -0,0 +1,22 @@ +from typing import Any +from typing_extensions import TypedDict + +DEFAULT_ENV: str +SCHEMES: dict[str, str] + +class _DBConfigBase(TypedDict): + NAME: str + +class _DBConfig(_DBConfigBase, total=False): + USER: str + PASSWORD: str + HOST: str + PORT: str + CONN_MAX_AGE: int + OPTIONS: dict[str, Any] + ENGINE: str + +def parse(url: str, engine: str | None = ..., conn_max_age: int = ..., ssl_require: bool = ...) -> _DBConfig: ... +def config( + env: str = ..., default: str | None = ..., engine: str | None = ..., conn_max_age: int = ..., ssl_require: bool = ... +) -> _DBConfig: ...