setup.py

from __future__ import annotations

import platform
import sys

from setuptools import Extension
from setuptools import setup

if sys.platform != 'win32' and platform.python_implementation() == 'CPython':
    try:
        import wheel.bdist_wheel
    except ImportError:
        cmdclass = {}
    else:
        class bdist_wheel(wheel.bdist_wheel.bdist_wheel):
            def finalize_options(self) -> None:
                self.py_limited_api = f'cp3{sys.version_info[1]}'
                super().finalize_options()

        cmdclass = {'bdist_wheel': bdist_wheel}
else:
    cmdclass = {}

setup(
    ext_modules=[
        Extension(
            'c_module', ['c_module_proj/c_module.c'],
            py_limited_api=True, define_macros=[('Py_LIMITED_API', None)],
        ),
        Extension(
            'hello_lib', ['hello_lib_proj/hello_lib.go'],
            py_limited_api=True, define_macros=[('Py_LIMITED_API', None)],
        ),
        Extension(
            'red', ['red_proj/red.go'],
            py_limited_api=True, define_macros=[('Py_LIMITED_API', None)],
        ),
        Extension(
            'sum_go', ['sum_go_proj/sum_go.go'],
            py_limited_api=True, define_macros=[('Py_LIMITED_API', None)],
        ),
        Extension(
            'sum_pure_go', ['sum_pure_go_proj/sum_pure_go.go'],
            py_limited_api=True, define_macros=[('Py_LIMITED_API', None)],
        ),
    ],
    cmdclass=cmdclass,
    build_golang={'root': 'github.com/asottile/setuptools-golang-examples'},
)