feat: add cython

build_ext.py

@@ -0,0 +1,35 @@
+"""Build optional cython modules."""
+
+import contextlib
+import os
+from distutils.command.build_ext import build_ext
+from typing import Any
+
+
+class BuildExt(build_ext):
+    def build_extensions(self) -> None:
+        with contextlib.suppress(Exception):
+            super().build_extensions()
+
+
+def build(setup_kwargs: dict[Any, Any]) -> None:
+    if os.environ.get("SKIP_CYTHON", False):
+        return
+    try:
+        from Cython.Build import cythonize
+
+        setup_kwargs.update(
+            dict(
+                ext_modules=cythonize(
+                    [
+                        "src/kasa_crypt/kasa_crypt.py",
+                    ],
+                    compiler_directives={"language_level": "3"},  # Python 3
+                ),
+                cmdclass=dict(build_ext=BuildExt),
+            )
+        )
+    except Exception:
+        if os.environ.get("REQUIRE_CYTHON"):
+            raise
+        pass

pyproject.toml

@@ -17,6 +17,10 @@ packages = [
     { include = "kasa_crypt", from = "src" },
 ]
 
+[tool.poetry.build]
+generate-setup-file = true
+script = "build_ext.py"
+
 [tool.poetry.urls]
 "Bug Tracker" = "https://github.com/bdraco/kasa-crypt/issues"
 "Changelog" = "https://github.com/bdraco/kasa-crypt/blob/main/CHANGELOG.md"
@@ -73,5 +77,5 @@ module = "tests.*"
 allow_untyped_defs = true
 
 [build-system]
-requires = ["poetry-core>=1.0.0"]
+requires = ['setuptools>=65.4.1', 'wheel', 'Cython', "poetry-core>=1.0.0"]
 build-backend = "poetry.core.masonry.api"