[peewee] Model.select() type contains the model being selected

Author: Psycojoker

stubs/peewee/peewee.pyi

@@ -5,7 +5,7 @@ from collections.abc import Callable, Generator, Iterable, Iterator
 from datetime import datetime
 from decimal import Decimal
 from types import TracebackType
-from typing import Any, ClassVar, Final, Literal, NamedTuple, NoReturn, TypeVar, overload, type_check_only
+from typing import Any, ClassVar, Final, Literal, Generic, NamedTuple, NoReturn, TypeVar, overload, type_check_only
 from typing_extensions import Self, TypeIs
 from uuid import UUID
 
@@ -1660,7 +1660,7 @@ class Model(metaclass=ModelBase):
     @classmethod
     def alias(cls, alias=None) -> ModelAlias: ...
     @classmethod
-    def select(cls, *fields) -> ModelSelect: ...
+    def select(cls: type[_TModel], *fields) -> ModelSelect[_TModel]: ...
     @classmethod
     def update(cls, data=None, /, **update) -> ModelUpdate: ...
     @classmethod
@@ -1782,11 +1782,11 @@ class ModelCompoundSelectQuery(BaseModelSelect, CompoundSelectQuery):  # type: i
     model: Incomplete
     def __init__(self, model, *args, **kwargs) -> None: ...
 
-class ModelSelect(BaseModelSelect, Select):  # type: ignore[misc]
-    model: Incomplete
+class ModelSelect(BaseModelSelect, Select, Generic[_TModel]):  # type: ignore[misc]
+    model: _TModel
     def __init__(self, model, fields_or_models, is_default: bool = False) -> None: ...
     def clone(self) -> Self: ...
-    def select(self, *fields_or_models): ...
+    def select(self, *fields_or_models) -> ModelSelect[_TModel]: ...
     def select_extend(self, *columns): ...
     def switch(self, ctx=None) -> Self: ...
     def join(self, dest, join_type="INNER JOIN", on=None, src=None, attr=None) -> Self: ...  # type: ignore[override]