From 93c1bfcdd7b9fa099aa67aaa19a00b4167978720 Mon Sep 17 00:00:00 2001 From: Xavier Francisco Date: Sat, 28 May 2022 02:34:05 +0000 Subject: [PATCH] Add stubs for dj-database-url --- pyrightconfig.stricter.json | 1 + stubs/dj-database-url/METADATA.toml | 1 + stubs/dj-database-url/dj_database_url.pyi | 22 ++++++++++++++++++++++ 3 files changed, 24 insertions(+) create mode 100644 stubs/dj-database-url/METADATA.toml create mode 100644 stubs/dj-database-url/dj_database_url.pyi diff --git a/pyrightconfig.stricter.json b/pyrightconfig.stricter.json index 2f41cc1deedc..803af495aaae 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 000000000000..1883b870198a --- /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 000000000000..7c9a69a6f5c5 --- /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: ...