FilesExpand file tree

 master

/

setup.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

77 lines (65 loc) · 2.04 KB

 master

/

setup.py

Copy path

Top

File metadata and controls

• Code
• Blame

77 lines (65 loc) · 2.04 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

from os import path

from platform import system

from sysconfig import get_config_var

from setuptools import Extension, find_packages, setup

from setuptools.command.build import build

from setuptools.command.egg_info import egg_info

from wheel.bdist_wheel import bdist_wheel

sources = [

"bindings/python/tree_sitter_javascript/binding.c",

"src/parser.c",

]

if path.exists("src/scanner.c"):

sources.append("src/scanner.c")

macros: list[tuple[str, str | None]] = [

("PY_SSIZE_T_CLEAN", None),

("TREE_SITTER_HIDE_SYMBOLS", None),

]

if limited_api := not get_config_var("Py_GIL_DISABLED"):

macros.append(("Py_LIMITED_API", "0x030A0000"))

if system() != "Windows":

cflags = ["-std=c11", "-fvisibility=hidden"]

else:

cflags = ["/std:c11", "/utf-8"]

class Build(build):

def run(self):

if path.isdir("queries"):

dest = path.join(self.build_lib, "tree_sitter_javascript", "queries")

self.copy_tree("queries", dest)

super().run()

class BdistWheel(bdist_wheel):

def get_tag(self):

python, abi, platform = super().get_tag()

if python.startswith("cp"):

python, abi = "cp310", "abi3"

return python, abi, platform

class EggInfo(egg_info):

def find_sources(self):

super().find_sources()

self.filelist.recursive_include("queries", "*.scm")

self.filelist.include("src/tree_sitter/*.h")

setup(

packages=find_packages("bindings/python"),

package_dir={"": "bindings/python"},

package_data={

"tree_sitter_javascript": ["*.pyi", "py.typed"],

"tree_sitter_javascript.queries": ["*.scm"],

},

ext_package="tree_sitter_javascript",

ext_modules=[

Extension(

name="_binding",

sources=sources,

extra_compile_args=cflags,

define_macros=macros,

include_dirs=["src"],

py_limited_api=limited_api,

)

],

cmdclass={

"build": Build,

"bdist_wheel": BdistWheel,

"egg_info": EggInfo,

},

zip_safe=False

)