From 74232e860b6267d772ac987b5e29ce68f9723928 Mon Sep 17 00:00:00 2001 From: Joakim Soderlund Date: Tue, 28 Jan 2025 10:41:43 +0000 Subject: [PATCH] Mark database parameters in Peewee as optional Peewee injects database arguments using a decorator whenever a model or query has a bound connection. Passing the argument is therefore not required, even for some function parameters without default values. --- stubs/peewee/peewee.pyi | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/stubs/peewee/peewee.pyi b/stubs/peewee/peewee.pyi index f9066c393709..07ef1a52eb6d 100644 --- a/stubs/peewee/peewee.pyi +++ b/stubs/peewee/peewee.pyi @@ -526,7 +526,7 @@ class BaseQuery(Node): def objects(self, constructor: Incomplete | None = ...) -> Self: ... def __sql__(self, ctx) -> None: ... def sql(self): ... - def execute(self, database): ... + def execute(self, database: Incomplete | None = ...): ... def iterator(self, database: Incomplete | None = ...): ... def __iter__(self): ... def __getitem__(self, value): ... @@ -571,20 +571,20 @@ class SelectQuery(Query): def select_from(self, *columns): ... class SelectBase(_HashableSource, Source, SelectQuery): # type: ignore[misc] - def peek(self, database, n: int = ...): ... - def first(self, database, n: int = ...): ... - def scalar(self, database, as_tuple: bool = ..., as_dict: bool = ...): ... - def scalars(self, database) -> Generator[Incomplete, None, None]: ... - def count(self, database, clear_limit: bool = ...): ... - def exists(self, database): ... - def get(self, database): ... + def peek(self, database: Incomplete | None = ..., n: int = ...): ... + def first(self, database: Incomplete | None = ..., n: int = ...): ... + def scalar(self, database: Incomplete | None = ..., as_tuple: bool = ..., as_dict: bool = ...): ... + def scalars(self, database: Incomplete | None = ...) -> Generator[Incomplete, None, None]: ... + def count(self, database: Incomplete | None = ..., clear_limit: bool = ...): ... + def exists(self, database: Incomplete | None = ...): ... + def get(self, database: Incomplete | None = ...): ... class CompoundSelectQuery(SelectBase): lhs: Incomplete op: Incomplete rhs: Incomplete def __init__(self, lhs, op, rhs) -> None: ... - def exists(self, database): ... + def exists(self, database: Incomplete | None = ...): ... def __sql__(self, ctx): ... class Select(SelectBase):