diff --git a/stubs/tree-sitter-languages/@tests/stubtest_allowlist.txt b/stubs/tree-sitter-languages/@tests/stubtest_allowlist.txt new file mode 100644 index 000000000..ba0665953 --- /dev/null +++ b/stubs/tree-sitter-languages/@tests/stubtest_allowlist.txt @@ -0,0 +1,5 @@ +# No idea what this is, seems to be auto-generated +tree_sitter_languages.core.__test__ + +# Fails to import and is not supposed to be imported, but stubtest finds it somehow +tree_sitter_languages.languages diff --git a/stubs/tree-sitter-languages/METADATA.toml b/stubs/tree-sitter-languages/METADATA.toml new file mode 100644 index 000000000..29308d425 --- /dev/null +++ b/stubs/tree-sitter-languages/METADATA.toml @@ -0,0 +1,5 @@ +version = "1.3.*" +requires = ["types-tree-sitter"] + +[tool.stubtest] +ignore_missing_stub = false diff --git a/stubs/tree-sitter-languages/tree_sitter_languages/__init__.pyi b/stubs/tree-sitter-languages/tree_sitter_languages/__init__.pyi new file mode 100644 index 000000000..59a083877 --- /dev/null +++ b/stubs/tree-sitter-languages/tree_sitter_languages/__init__.pyi @@ -0,0 +1,7 @@ +from .core import get_language as get_language, get_parser as get_parser + +__version__: str +__title__: str +__author__: str +__license__: str +__copyright__: str diff --git a/stubs/tree-sitter-languages/tree_sitter_languages/core.pyi b/stubs/tree-sitter-languages/tree_sitter_languages/core.pyi new file mode 100644 index 000000000..b59c21d38 --- /dev/null +++ b/stubs/tree-sitter-languages/tree_sitter_languages/core.pyi @@ -0,0 +1,4 @@ +from tree_sitter import Language, Parser + +def get_language(language: str) -> Language: ... +def get_parser(language: str) -> Parser: ...