diff --git a/stubs/peewee/METADATA.toml b/stubs/peewee/METADATA.toml index 5adfe58b8..cabd85fab 100644 --- a/stubs/peewee/METADATA.toml +++ b/stubs/peewee/METADATA.toml @@ -1,2 +1,2 @@ -version = "3.16.*" +version = "3.17.*" upstream_repository = "https://github.com/coleifer/peewee" diff --git a/stubs/peewee/peewee.pyi b/stubs/peewee/peewee.pyi index a59530702..8f2b35474 100644 --- a/stubs/peewee/peewee.pyi +++ b/stubs/peewee/peewee.pyi @@ -2,9 +2,9 @@ import itertools import logging import threading from _typeshed import Incomplete, SupportsKeysAndGetItem -from collections.abc import Generator, Iterable +from collections.abc import Callable, Generator, Iterable from types import TracebackType -from typing import ClassVar, NamedTuple, TypeVar +from typing import Any, ClassVar, NamedTuple, TypeVar from typing_extensions import Self class NullHandler(logging.Handler): @@ -18,6 +18,7 @@ long = int izip_longest = itertools.zip_longest _VT = TypeVar("_VT") +_F = TypeVar("_F", bound=Callable[..., Any]) class attrdict(dict[str, _VT]): def __getattr__(self, attr: str) -> _VT: ... @@ -116,9 +117,10 @@ class Node: def unwrap(self): ... class ColumnFactory: - node: Incomplete - def __init__(self, node) -> None: ... - def __getattr__(self, attr: str): ... + node: Node + def __init__(self, node: Node) -> None: ... + def __getattr__(self, attr: str) -> Column: ... + def __getitem__(self, attr: str) -> Column: ... class _DynamicColumn: def __get__(self, instance, instance_type: Incomplete | None = ...): ... @@ -880,8 +882,8 @@ class SqliteDatabase(Database): def aggregate(self, name: Incomplete | None = ..., num_params: int = ...): ... def register_collation(self, fn, name: Incomplete | None = ...): ... def collation(self, name: Incomplete | None = ...): ... - def register_function(self, fn, name: Incomplete | None = ..., num_params: int = ...) -> None: ... - def func(self, name: Incomplete | None = ..., num_params: int = ...): ... + def register_function(self, fn, name: str | None = None, num_params: int = -1, deterministic: bool | None = None) -> None: ... + def func(self, name: str | None = None, num_params: int = -1, deterministic: bool | None = None) -> Callable[[_F], _F]: ... def register_window_function(self, klass, name: Incomplete | None = ..., num_params: int = ...) -> None: ... def window_function(self, name: Incomplete | None = ..., num_params: int = ...): ... def register_table_function(self, klass, name: Incomplete | None = ...) -> None: ... @@ -934,6 +936,7 @@ class PostgresqlDatabase(Database): def is_connection_usable(self): ... def last_insert_id(self, cursor, query_type: Incomplete | None = ...): ... def rows_affected(self, cursor): ... + def begin(self, isolation_level: str | None = None) -> None: ... def get_tables(self, schema: Incomplete | None = ...): ... def get_views(self, schema: Incomplete | None = ...): ... def get_indexes(self, table, schema: Incomplete | None = ...): ... @@ -967,6 +970,7 @@ class MySQLDatabase(Database): def init(self, database, **kwargs) -> None: ... def is_connection_usable(self): ... def default_values_insert(self, ctx): ... + def begin(self, isolation_level: str | None = None) -> None: ... def get_tables(self, schema: Incomplete | None = ...): ... def get_views(self, schema: Incomplete | None = ...): ... def get_indexes(self, table, schema: Incomplete | None = ...): ...