1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
from pathlib import Path
from platform import system
from setuptools import Extension, find_packages, setup
from setuptools.command.build import build
from wheel.bdist_wheel import bdist_wheel
class Build(build):
def run(self):
if (block_queries := Path("tree-sitter-markdown", "queries")).is_dir():
dest = Path(self.build_lib, "tree_sitter_markdown", "queries", "markdown")
self.copy_tree(str(block_queries), str(dest))
if (inline_queries := Path("tree-sitter-markdown-inline", "queries")).is_dir():
dest = Path(self.build_lib, "tree_sitter_markdown", "queries", "markdown_inline")
self.copy_tree(str(inline_queries), str(dest))
super().run()
class BdistWheel(bdist_wheel):
def get_tag(self):
python, abi, platform = super().get_tag()
if python.startswith("cp"):
python, abi = "cp39", "abi3"
return python, abi, platform
setup(
packages=find_packages("bindings/python"),
package_dir={"": "bindings/python"},
package_data={
"tree_sitter_markdown": ["*.pyi", "py.typed"],
"tree_sitter_markdown.queries": ["*.scm"],
},
ext_package="tree_sitter_markdown",
ext_modules=[
Extension(
name="_binding",
sources=[
"bindings/python/tree_sitter_markdown/binding.c",
"tree-sitter-markdown/src/parser.c",
"tree-sitter-markdown/src/scanner.c",
"tree-sitter-markdown-inline/src/parser.c",
"tree-sitter-markdown-inline/src/scanner.c",
],
extra_compile_args=(
["-std=c11"] if system() != "Windows" else []
),
define_macros=[
("Py_LIMITED_API", "0x03090000"),
("PY_SSIZE_T_CLEAN", None)
],
include_dirs=["tree-sitter-markdown/src"],
py_limited_api=True,
)
],
cmdclass={
"build": Build,
"bdist_wheel": BdistWheel
},
zip_safe=False
)
|
