Add StrPromise and StrOrPromise aliases to django_stubs_ext.

We make StrPromise and StrOrPromise available via django_stubs_ext so
that conditional imports with TYPE_CHECKING is not required.
These aliases fall back to Promise or Union[str, Promise]
when not TYPE_CHECKING.

Signed-off-by: Zixuan James Li <[email protected]>

Author: PIG208

django_stubs_ext/django_stubs_ext/__init__.py

@@ -1,7 +1,16 @@
+from .aliases import StrOrPromise, StrPromise
 from .aliases import ValuesQuerySet as ValuesQuerySet
 from .annotations import Annotations as Annotations
 from .annotations import WithAnnotations as WithAnnotations
 from .patch import monkeypatch as monkeypatch
 from .types import AnyAttrAllowed as AnyAttrAllowed
 
-__all__ = ["monkeypatch", "ValuesQuerySet", "WithAnnotations", "Annotations", "AnyAttrAllowed"]
+__all__ = [
+    "monkeypatch",
+    "ValuesQuerySet",
+    "WithAnnotations",
+    "Annotations",
+    "AnyAttrAllowed",
+    "StrPromise",
+    "StrOrPromise",
+]

django_stubs_ext/django_stubs_ext/aliases.py

@@ -2,9 +2,15 @@
 
 if typing.TYPE_CHECKING:
     from django.db.models.query import _T, _QuerySet, _Row
+    from django.utils.functional import _StrOrPromise as StrOrPromise
+    from django.utils.functional import _StrPromise as StrPromise
 
     ValuesQuerySet = _QuerySet[_T, _Row]
 else:
     from django.db.models.query import QuerySet
+    from django.utils.functional import Promise as StrPromise
 
     ValuesQuerySet = QuerySet
+    StrOrPromise = typing.Union[str, StrPromise]
+
+__all__ = ["StrOrPromise", "StrPromise"]