feat: reduce size of binaries

bdraco · bdraco · commit 16f3564188eb · 2025-03-04T15:22:56.000-10:00
diff --git a/build_ext.py b/build_ext.py
@@ -5,8 +5,44 @@
 from distutils.command.build_ext import build_ext
 from typing import Any
 
+try:
+    from setuptools import Extension
+except ImportError:
+    from distutils.core import Extension
+
 _LOGGER = logging.getLogger(__name__)
 
+TO_CYTHONIZE = [
+    "src/zeroconf/_dns.py",
+    "src/zeroconf/_cache.py",
+    "src/zeroconf/_history.py",
+    "src/zeroconf/_record_update.py",
+    "src/zeroconf/_listener.py",
+    "src/zeroconf/_protocol/incoming.py",
+    "src/zeroconf/_protocol/outgoing.py",
+    "src/zeroconf/_handlers/answers.py",
+    "src/zeroconf/_handlers/record_manager.py",
+    "src/zeroconf/_handlers/multicast_outgoing_queue.py",
+    "src/zeroconf/_handlers/query_handler.py",
+    "src/zeroconf/_services/__init__.py",
+    "src/zeroconf/_services/browser.py",
+    "src/zeroconf/_services/info.py",
+    "src/zeroconf/_services/registry.py",
+    "src/zeroconf/_updates.py",
+    "src/zeroconf/_utils/ipaddress.py",
+    "src/zeroconf/_utils/time.py",
+]
+
+EXTENSIONS = [
+    Extension(
+        ext.removeprefix("src/").removesuffix(".py").replace("/", "."),
+        [ext],
+        language="c",
+        extra_compile_args=["-O3", "-g0"],
+    )
+    for ext in TO_CYTHONIZE
+]
+
 
 class BuildExt(build_ext):
     def build_extensions(self) -> None:
@@ -25,26 +61,7 @@ def build(setup_kwargs: Any) -> None:
         setup_kwargs.update(
             {
                 "ext_modules": cythonize(
-                    [
-                        "src/zeroconf/_dns.py",
-                        "src/zeroconf/_cache.py",
-                        "src/zeroconf/_history.py",
-                        "src/zeroconf/_record_update.py",
-                        "src/zeroconf/_listener.py",
-                        "src/zeroconf/_protocol/incoming.py",
-                        "src/zeroconf/_protocol/outgoing.py",
-                        "src/zeroconf/_handlers/answers.py",
-                        "src/zeroconf/_handlers/record_manager.py",
-                        "src/zeroconf/_handlers/multicast_outgoing_queue.py",
-                        "src/zeroconf/_handlers/query_handler.py",
-                        "src/zeroconf/_services/__init__.py",
-                        "src/zeroconf/_services/browser.py",
-                        "src/zeroconf/_services/info.py",
-                        "src/zeroconf/_services/registry.py",
-                        "src/zeroconf/_updates.py",
-                        "src/zeroconf/_utils/ipaddress.py",
-                        "src/zeroconf/_utils/time.py",
-                    ],
+                    EXTENSIONS,
                     compiler_directives={"language_level": "3"},  # Python 3
                 ),
                 "cmdclass": {"build_ext": BuildExt},
