feat: add Cython as build dependency (#122)

Author: cdce8p

CHANGES.rst

@@ -11,6 +11,7 @@ CHANGES
 * Rebuild Cython wrapper with Cython 3.1.2 (#119).
 * Moved static project metadata to ``pyproject.toml`` (#120).
 * Updated metadata license to include the bundled one from marisa-trie as well (#120).
+* Add Cython as build dependency (#122).
 
 1.2.1 (2024-10-12)
 ------------------

MANIFEST.in

@@ -1,5 +1,3 @@
-include update_cpp.sh
-
 recursive-include src *.cpp *.pxd *.pyx
 recursive-include marisa-trie/lib/marisa *.h *.cc
 recursive-include marisa-trie/include/marisa *.h

pyproject.toml

@@ -1,6 +1,6 @@
 [build-system]
 build-backend = "setuptools.build_meta"
-requires = ["setuptools>=77.0"]
+requires = ["setuptools>=77.0", "cython>=3.1.2"]
 
 [project]
 name = "marisa-trie"

setup.py

@@ -1,13 +1,13 @@
-import glob
 import itertools
-import os.path
+from pathlib import Path
 
+from Cython.Build import cythonize
 from setuptools import setup, Extension
 
 
-MARISA_ROOT_DIR = "marisa-trie"
-MARISA_SOURCE_DIR = os.path.join(MARISA_ROOT_DIR, "lib")
-MARISA_INCLUDE_DIR = os.path.join(MARISA_ROOT_DIR, "include")
+MARISA_ROOT_DIR = Path("marisa-trie")
+MARISA_SOURCE_DIR = MARISA_ROOT_DIR / "lib"
+MARISA_INCLUDE_DIR = MARISA_ROOT_DIR / "include"
 MARISA_FILES = [
     "marisa/*.cc",
     "marisa/grimoire.cc",
@@ -16,36 +16,29 @@
     "marisa/grimoire/vector/*.cc",
 ]
 
-MARISA_FILES[:] = itertools.chain(
-    *(glob.glob(os.path.join(MARISA_SOURCE_DIR, path)) for path in MARISA_FILES)
+MARISA_FILES[:] = map(str, itertools.chain(
+        *(MARISA_SOURCE_DIR.glob(path) for path in MARISA_FILES)
+    )
 )
 
+extensions = [
+    Extension(
+        "marisa_trie",
+        sources=["src/*.pyx"],
+        language="c++",
+        include_dirs=[str(MARISA_INCLUDE_DIR)],
+    ),
+]
 
 setup(
     libraries=[
         (
             "libmarisa-trie",
             {
                 "sources": MARISA_FILES,
-                "include_dirs": [MARISA_SOURCE_DIR, MARISA_INCLUDE_DIR],
+                "include_dirs": [str(MARISA_SOURCE_DIR), str(MARISA_INCLUDE_DIR)],
             },
         )
     ],
-    ext_modules=[
-        Extension(
-            "marisa_trie",
-            [
-                "src/agent.cpp",
-                "src/base.cpp",
-                "src/iostream.cpp",
-                "src/key.cpp",
-                "src/keyset.cpp",
-                "src/marisa_trie.cpp",
-                "src/query.cpp",
-                "src/std_iostream.cpp",
-                "src/trie.cpp",
-            ],
-            include_dirs=[MARISA_INCLUDE_DIR],
-        )
-    ],
+    ext_modules=cythonize(extensions),
 )