From 0b28177268c962d633addada6d9921496080462f Mon Sep 17 00:00:00 2001
From: Akuli <akuviljanen17@gmail.com>
Date: Wed, 17 Aug 2022 23:35:44 +0300
Subject: [PATCH] Add stubs for tree-sitter-languages (#8548)

Co-authored-by: Alex Waygood <Alex.Waygood@Gmail.com>
---
 stubs/tree-sitter-languages/@tests/stubtest_allowlist.txt  | 5 +++++
 stubs/tree-sitter-languages/METADATA.toml                  | 5 +++++
 .../tree_sitter_languages/__init__.pyi                     | 7 +++++++
 stubs/tree-sitter-languages/tree_sitter_languages/core.pyi | 4 ++++
 4 files changed, 21 insertions(+)
 create mode 100644 stubs/tree-sitter-languages/@tests/stubtest_allowlist.txt
 create mode 100644 stubs/tree-sitter-languages/METADATA.toml
 create mode 100644 stubs/tree-sitter-languages/tree_sitter_languages/__init__.pyi
 create mode 100644 stubs/tree-sitter-languages/tree_sitter_languages/core.pyi

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: ...