100 lines
1.7 KiB
Python
100 lines
1.7 KiB
Python
![]() |
|
||
|
from typing import Any, Callable, Collection, Iterable, List, NoReturn, Optional, Text, TypeVar, Union, overload
|
||
|
|
||
|
from absl.flags import _flag
|
||
|
|
||
|
|
||
|
_MainArgs = TypeVar('_MainArgs')
|
||
|
_Exc = TypeVar('_Exc', bound=Exception)
|
||
|
|
||
|
|
||
|
class ExceptionHandler():
|
||
|
|
||
|
def wants(self, exc: _Exc) -> bool:
|
||
|
...
|
||
|
|
||
|
def handle(self, exc: _Exc):
|
||
|
...
|
||
|
|
||
|
|
||
|
EXCEPTION_HANDLERS: List[ExceptionHandler] = ...
|
||
|
|
||
|
|
||
|
class HelpFlag(_flag.BooleanFlag):
|
||
|
def __init__(self):
|
||
|
...
|
||
|
|
||
|
|
||
|
class HelpshortFlag(HelpFlag):
|
||
|
...
|
||
|
|
||
|
|
||
|
class HelpfullFlag(_flag.BooleanFlag):
|
||
|
def __init__(self):
|
||
|
...
|
||
|
|
||
|
|
||
|
class HelpXMLFlag(_flag.BooleanFlag):
|
||
|
def __init__(self):
|
||
|
...
|
||
|
|
||
|
|
||
|
def define_help_flags() -> None:
|
||
|
...
|
||
|
|
||
|
|
||
|
@overload
|
||
|
def usage(shorthelp: Union[bool, int] = ...,
|
||
|
writeto_stdout: Union[bool, int] = ...,
|
||
|
detailed_error: Optional[Any] = ...,
|
||
|
exitcode: None = ...) -> None:
|
||
|
...
|
||
|
|
||
|
|
||
|
@overload
|
||
|
def usage(shorthelp: Union[bool, int] = ...,
|
||
|
writeto_stdout: Union[bool, int] = ...,
|
||
|
detailed_error: Optional[Any] = ...,
|
||
|
exitcode: int = ...) -> NoReturn:
|
||
|
...
|
||
|
|
||
|
|
||
|
def install_exception_handler(handler: ExceptionHandler) -> None:
|
||
|
...
|
||
|
|
||
|
|
||
|
class Error(Exception):
|
||
|
...
|
||
|
|
||
|
|
||
|
class UsageError(Error):
|
||
|
exitcode: int
|
||
|
|
||
|
|
||
|
def parse_flags_with_usage(args: List[Text]) -> List[Text]:
|
||
|
...
|
||
|
|
||
|
|
||
|
def call_after_init(callback: Callable[[], Any]) -> None:
|
||
|
...
|
||
|
|
||
|
|
||
|
# Without the flag_parser argument, `main` should require a List[Text].
|
||
|
@overload
|
||
|
def run(
|
||
|
main: Callable[[List[Text]], Any],
|
||
|
argv: Optional[List[Text]] = ...,
|
||
|
*,
|
||
|
) -> NoReturn:
|
||
|
...
|
||
|
|
||
|
|
||
|
@overload
|
||
|
def run(
|
||
|
main: Callable[[_MainArgs], Any],
|
||
|
argv: Optional[List[Text]] = ...,
|
||
|
*,
|
||
|
flags_parser: Callable[[List[Text]], _MainArgs],
|
||
|
) -> NoReturn:
|
||
|
...
|