Add stubs for invoke (#6938)

Author: kasium

pyrightconfig.stricter.json

@@ -36,6 +36,7 @@
         "stubs/html5lib",
         "stubs/httplib2",
         "stubs/humanfriendly",
+        "stubs/invoke",
         "stubs/jmespath",
         "stubs/jsonschema",
         "stubs/ldap3",

stubs/invoke/METADATA.toml

@@ -0,0 +1 @@
+version = "1.6.*"

stubs/invoke/invoke/__init__.pyi

@@ -0,0 +1,37 @@
+from typing import Any
+
+from .collection import Collection as Collection
+from .config import Config as Config
+from .context import Context as Context, MockContext as MockContext
+from .exceptions import (
+    AmbiguousEnvVar as AmbiguousEnvVar,
+    AuthFailure as AuthFailure,
+    CollectionNotFound as CollectionNotFound,
+    CommandTimedOut as CommandTimedOut,
+    Exit as Exit,
+    Failure as Failure,
+    ParseError as ParseError,
+    PlatformError as PlatformError,
+    ResponseNotAccepted as ResponseNotAccepted,
+    SubprocessPipeError as SubprocessPipeError,
+    ThreadException as ThreadException,
+    UncastableEnvVar as UncastableEnvVar,
+    UnexpectedExit as UnexpectedExit,
+    UnknownFileType as UnknownFileType,
+    UnpicklableConfigMember as UnpicklableConfigMember,
+    WatcherError as WatcherError,
+)
+from .executor import Executor as Executor
+from .loader import FilesystemLoader as FilesystemLoader
+from .parser import Argument as Argument, Parser as Parser, ParserContext as ParserContext, ParseResult as ParseResult
+from .program import Program as Program
+from .runners import Local as Local, Promise as Promise, Result as Result, Runner as Runner
+from .tasks import Call as Call, Task as Task, call as call, task as task
+from .terminals import pty_size as pty_size
+from .watchers import FailingResponder as FailingResponder, Responder as Responder, StreamWatcher as StreamWatcher
+
+__version_info__: tuple[int, int, int]
+__version__: str
+
+def run(command: str, **kwargs: Any) -> Result: ...
+def sudo(command: str, **kwargs: Any) -> Result: ...

stubs/invoke/invoke/collection.pyi

@@ -0,0 +1,24 @@
+from typing import Any
+
+class Collection:
+    tasks: Any
+    collections: Any
+    default: str | None
+    name: str | None
+    loaded_from: Any
+    auto_dash_names: bool
+    def __init__(self, *args, **kwargs) -> None: ...
+    @classmethod
+    def from_module(cls, module, name=..., config=..., loaded_from=..., auto_dash_names=...): ...
+    def add_task(self, task, name=..., aliases=..., default=...) -> None: ...
+    def add_collection(self, coll, name=..., default=...) -> None: ...
+    def subcollection_from_path(self, path): ...
+    def task_with_config(self, name): ...
+    def to_contexts(self): ...
+    def subtask_name(self, collection_name, task_name): ...
+    def transform(self, name): ...
+    @property
+    def task_names(self): ...
+    def configuration(self, taskpath=...): ...
+    def configure(self, options) -> None: ...
+    def serialized(self): ...

stubs/invoke/invoke/completion/__init__.pyi

@@ -0,0 +1,8 @@
+from typing import Iterable, NoReturn, Sequence
+
+from ..collection import Collection
+from ..parser import ParserContext, ParseResult
+
+def complete(names: Iterable[str], core: ParseResult, initial_context: ParserContext, collection: Collection) -> NoReturn: ...
+def print_task_names(collection: Collection) -> None: ...
+def print_completion_script(shell: str, names: Sequence[str]) -> None: ...

stubs/invoke/invoke/completion/complete.pyi

@@ -0,0 +1,60 @@
+from typing import Any
+
+def load_source(name: str, path: str) -> dict[str, Any]: ...
+
+class DataProxy:
+    @classmethod
+    def from_data(cls, data, root=..., keypath=...): ...
+    def __getattr__(self, key): ...
+    def __setattr__(self, key, value) -> None: ...
+    def __iter__(self): ...
+    def __eq__(self, other): ...
+    __hash__: Any
+    def __len__(self): ...
+    def __setitem__(self, key, value) -> None: ...
+    def __getitem__(self, key): ...
+    def __contains__(self, key): ...
+    def __delitem__(self, key) -> None: ...
+    def __delattr__(self, name) -> None: ...
+    def clear(self) -> None: ...
+    def pop(self, *args): ...
+    def popitem(self): ...
+    def setdefault(self, *args): ...
+    def update(self, *args, **kwargs) -> None: ...
+
+class Config(DataProxy):
+    prefix: str
+    file_prefix: Any
+    env_prefix: Any
+    @staticmethod
+    def global_defaults(): ...
+    def __init__(
+        self,
+        overrides=...,
+        defaults=...,
+        system_prefix=...,
+        user_prefix=...,
+        project_location=...,
+        runtime_path=...,
+        lazy: bool = ...,
+    ) -> None: ...
+    def load_base_conf_files(self) -> None: ...
+    def load_defaults(self, data, merge: bool = ...) -> None: ...
+    def load_overrides(self, data, merge: bool = ...) -> None: ...
+    def load_system(self, merge: bool = ...) -> None: ...
+    def load_user(self, merge: bool = ...) -> None: ...
+    def load_project(self, merge: bool = ...) -> None: ...
+    def set_runtime_path(self, path) -> None: ...
+    def load_runtime(self, merge: bool = ...) -> None: ...
+    def load_shell_env(self) -> None: ...
+    def load_collection(self, data, merge: bool = ...) -> None: ...
+    def set_project_location(self, path) -> None: ...
+    def merge(self) -> None: ...
+    def clone(self, into=...): ...
+
+class AmbiguousMergeError(ValueError): ...
+
+def merge_dicts(base, updates): ...
+def copy_dict(source): ...
+def excise(dict_, keypath) -> None: ...
+def obliterate(base, deletions) -> None: ...

stubs/invoke/invoke/config.pyi

@@ -0,0 +1,22 @@
+from contextlib import AbstractContextManager
+
+from .config import Config, DataProxy
+
+class Context(DataProxy):
+    def __init__(self, config: Config | None = ...) -> None: ...
+    @property
+    def config(self) -> Config: ...
+    @config.setter
+    def config(self, value: Config) -> None: ...
+    def run(self, command: str, **kwargs): ...
+    def sudo(self, command: str, *, password: str = ..., user: str = ..., **kwargs): ...
+    def prefix(self, command: str) -> AbstractContextManager[None]: ...
+    @property
+    def cwd(self) -> str: ...
+    def cd(self, path: str) -> AbstractContextManager[None]: ...
+
+class MockContext(Context):
+    def __init__(self, config: Config | None = ..., **kwargs) -> None: ...
+    def run(self, command: str, *args, **kwargs): ...
+    def sudo(self, command: str, *args, **kwargs): ...
+    def set_result_for(self, attname, command, result) -> None: ...

stubs/invoke/invoke/context.pyi

@@ -0,0 +1,6 @@
+from typing import Any
+
+class Environment:
+    data: Any
+    def __init__(self, config, prefix) -> None: ...
+    def load(self): ...

stubs/invoke/invoke/env.pyi

@@ -0,0 +1,47 @@
+from typing import Any
+
+class CollectionNotFound(Exception):
+    name: Any
+    start: Any
+    def __init__(self, name, start) -> None: ...
+
+class Failure(Exception):
+    result: Any
+    reason: Any
+    def __init__(self, result, reason=...) -> None: ...
+    def streams_for_display(self): ...
+
+class UnexpectedExit(Failure): ...
+
+class CommandTimedOut(Failure):
+    timeout: Any
+    def __init__(self, result, timeout) -> None: ...
+
+class AuthFailure(Failure):
+    result: Any
+    prompt: Any
+    def __init__(self, result, prompt) -> None: ...
+
+class ParseError(Exception):
+    context: Any
+    def __init__(self, msg, context=...) -> None: ...
+
+class Exit(Exception):
+    message: Any
+    def __init__(self, message=..., code=...) -> None: ...
+    @property
+    def code(self): ...
+
+class PlatformError(Exception): ...
+class AmbiguousEnvVar(Exception): ...
+class UncastableEnvVar(Exception): ...
+class UnknownFileType(Exception): ...
+class UnpicklableConfigMember(Exception): ...
+
+class ThreadException(Exception):
+    exceptions: Any
+    def __init__(self, exceptions) -> None: ...
+
+class WatcherError(Exception): ...
+class ResponseNotAccepted(WatcherError): ...
+class SubprocessPipeError(Exception): ...