|
4 | 4 |
|
5 | 5 | from .. import log |
6 | 6 |
|
| 7 | +if t.TYPE_CHECKING: |
| 8 | + from typing_extensions import TypeAlias |
| 9 | + |
| 10 | + CLIColour: TypeAlias = t.Union[int, t.Tuple[int, int, int], str] |
| 11 | + |
| 12 | + |
7 | 13 | logger = logging.getLogger(__name__) |
8 | 14 |
|
9 | 15 |
|
@@ -184,14 +190,14 @@ def _interpret_color( |
184 | 190 |
|
185 | 191 |
|
186 | 192 | class UnknownStyleColor(Exception): |
187 | | - def __init__(self, color: str, *args: object, **kwargs: object) -> None: |
| 193 | + def __init__(self, color: "CLIColour", *args: object, **kwargs: object) -> None: |
188 | 194 | return super().__init__(f"Unknown color {color!r}", *args, **kwargs) |
189 | 195 |
|
190 | 196 |
|
191 | 197 | def style( |
192 | 198 | text: t.Any, |
193 | | - fg: t.Optional[t.Union[int, t.Tuple[int, int, int], str]] = None, |
194 | | - bg: t.Optional[t.Union[int, t.Tuple[int, int, int], str]] = None, |
| 199 | + fg: t.Optional["CLIColour"] = None, |
| 200 | + bg: t.Optional["CLIColour"] = None, |
195 | 201 | bold: t.Optional[bool] = None, |
196 | 202 | dim: t.Optional[bool] = None, |
197 | 203 | underline: t.Optional[bool] = None, |
|
0 commit comments