Fix term.text type export

Author: peace-maker

pwnlib/term/text.py

@@ -6,16 +6,16 @@
 
 from pwnlib.term import termcap
 
-# Accept any module attribute for type checking purposes.
-# The actual implementation of __getattr__ is below, and
-# only accepts a defined text decorator pattern, but generating
-# all possible combinations of text decorators is not feasible.
-# No IDE auto-complete but at least no type errors.
 _WhenSetter: TypeAlias = Literal["always", "never", "auto"] | TextIO
 class _TextDecorator(Protocol):
     def __call__(self, desc: str, when: _WhenSetter | None = None) -> str: ...
 
 if TYPE_CHECKING:
+    # Accept any module attribute for type checking purposes.
+    # The actual implementation of __getattr__ is below, and
+    # only accepts a defined text decorator pattern, but generating
+    # all possible combinations of text decorators is not feasible.
+    # No IDE auto-complete but at least no type errors.
     def __getattr__(name: str) -> _TextDecorator: ...
     def get(desc: str) -> _TextDecorator: ...
     when: bool | _WhenSetter
@@ -108,6 +108,8 @@ def f(self, s, when = None):
         return functools.partial(f, self)
 
     def __getattr__(self, desc: str) -> _TextDecorator:
+        if desc in ('_TextDecorator', '_WhenSetter'):
+            return {'_TextDecorator': _TextDecorator, '_WhenSetter': _WhenSetter}[desc]
         if desc.startswith('_'):
             raise AttributeError(desc)