Adds TaskFormatter (#202)

Closes #146

Author: 9en9i

celery-stubs/app/log.pyi

@@ -1,9 +1,17 @@
-from logging import Logger
+from logging import Logger, LogRecord
 from typing import Any
 
 from celery.app.base import Celery
-from celery.utils.log import LoggingProxy
-from celery.utils.term import colored as colored_
+from celery.local import class_property
+from celery.utils.log import ColorFormatter, LoggingProxy
+from celery.utils.term import colored
+from typing_extensions import override
+
+__all__ = ("Logging", "TaskFormatter")
+
+class TaskFormatter(ColorFormatter):
+    @override
+    def format(self, record: LogRecord) -> str: ...
 
 class Logging:
     app: Celery
@@ -52,7 +60,7 @@ class Logging:
     ) -> bool: ...
     def colored(
         self, logfile: Any | None = ..., enabled: bool | None = ...
-    ) -> colored_: ...
+    ) -> colored: ...
     def setup_handlers(
         self,
         logger: Logger,
@@ -63,3 +71,7 @@ class Logging:
         **kwargs: Any,
     ) -> Logger: ...
     def get_default_logger(self, name: str = ..., **kwargs: Any) -> Logger: ...
+    @class_property
+    def already_setup(self) -> bool: ...  # pyright: ignore[reportRedeclaration]
+    @already_setup.setter  # type: ignore[no-redef]
+    def already_setup(self, was_setup: bool) -> None: ...

celery-stubs/local.pyi

@@ -0,0 +1,22 @@
+from collections.abc import Callable
+from typing import Generic, TypeAlias, TypeVar
+
+from typing_extensions import Self
+
+_T = TypeVar("_T")
+_R = TypeVar("_R")
+
+_Getter: TypeAlias = Callable[[_T], _R]
+_Setter: TypeAlias = Callable[[_T, _R], None]
+
+class class_property(Generic[_T, _R]):
+    __get: _Getter[_T, _R]
+    __set: _Setter[_T, _R] | None
+    def __init__(
+        self,
+        getter: _Getter[_T, _R],
+        setter: _Setter[_T, _R] | None = None,
+    ) -> None: ...
+    def __get__(self, obj: _T | None, type: type[_T] | None = None) -> _R: ...
+    def __set__(self, obj: _T | None, value: _R) -> Self: ...
+    def setter(self, setter: _Setter[_T, _R]) -> Self: ...

celery-stubs/utils/log.pyi

@@ -1,7 +1,8 @@
-from logging import Formatter, Logger
+from logging import Formatter, Logger, LogRecord, _SysExcInfoType
 from typing import ClassVar
 
 from celery.utils.term import colored
+from typing_extensions import override
 
 class LoggingProxy: ...
 
@@ -17,3 +18,7 @@ class ColorFormatter(Formatter):
     COLORS: ClassVar[dict[str, colored]]
     colors: ClassVar[dict[str, colored]]
     def __init__(self, fmt: str | None = ..., use_color: bool = ...) -> None: ...
+    @override
+    def formatException(self, ei: _SysExcInfoType) -> str: ...
+    @override
+    def format(self, record: LogRecord) -> str: ...