Split out stubs.

chanicpanic · Feb 15, 2020 · 39aa08d · 39aa08d
1 parent 34d3ff9
commit 39aa08d
Show file tree

Hide file tree

Showing 10 changed files with 14 additions and 4 deletions.
diff --git a/.github/workflows/mypy.yml b/.github/workflows/mypy.yml
@@ -16,4 +16,4 @@ jobs:
           python -m pip install --upgrade pip
           pip install mypy
       - name: Lint with mypy
-        run: mypy -m lark || true
+        run: mypy -m lark-stubs || true
diff --git a/lark-stubs/__init__.pyi b/lark-stubs/__init__.pyi
@@ -0,0 +1,10 @@
+# -*- coding: utf-8 -*-
+
+from .tree import Tree
+from .visitors import Transformer, Visitor, v_args, Discard
+from .visitors import InlineTransformer, inline_args   # XXX Deprecated
+from .exceptions import ParseError, LexError, GrammarError, UnexpectedToken, UnexpectedInput, UnexpectedCharacters
+from .lexer import Token
+from .lark import Lark
+
+__version__: str
diff --git a/lark/exceptions.pyi → lark-stubs/exceptions.pyi b/lark/exceptions.pyi → lark-stubs/exceptions.pyi
diff --git a/lark/lark.pyi → lark-stubs/lark.pyi b/lark/lark.pyi → lark-stubs/lark.pyi
diff --git a/lark/lexer.pyi → lark-stubs/lexer.pyi b/lark/lexer.pyi → lark-stubs/lexer.pyi
diff --git a/lark/reconstruct.pyi → lark-stubs/reconstruct.pyi b/lark/reconstruct.pyi → lark-stubs/reconstruct.pyi
diff --git a/lark/tree.pyi → lark-stubs/tree.pyi b/lark/tree.pyi → lark-stubs/tree.pyi
diff --git a/lark/visitors.pyi → lark-stubs/visitors.pyi b/lark/visitors.pyi → lark-stubs/visitors.pyi
diff --git a/lark/py.typed b/lark/py.typed
diff --git a/setup.py b/setup.py
@@ -1,17 +1,17 @@
 import re
-from setuptools import setup
+from setuptools import find_packages, setup
 
 __version__ ,= re.findall('__version__ = "(.*)"', open('lark/__init__.py').read())
 
 setup(
     name = "lark-parser",
     version = __version__,
-    packages = ['lark', 'lark.parsers', 'lark.tools', 'lark.grammars'],
+    packages = ['lark', 'lark.parsers', 'lark.tools', 'lark.grammars', 'lark-stubs'],
 
     requires = [],
     install_requires = [],
 
-    package_data = { '': ['*.md', '*.lark', '*.pyi'], 'lark': ['py.typed'] },
+    package_data = {'': ['*.md', '*.lark'], 'lark-stubs': ['*.pyi']},
 
     test_suite = 'tests.__main__',