Изменить имя выходного файла в setup.py (distutils.extension) - PullRequest
2 голосов
/ 18 февраля 2020

Вот мои setup.py

from distutils.core import setup
from distutils.extension import Extension
from Cython.Build import cythonize

wrapper = Extension(
    name="libwrapper",
    ...
)
setup(
    name="libwrapper",
    ext_modules=cythonize([wrapper])
)

Когда я запускаю python3 setup.py build_ext, выходной файл называется libwrapper.cpython-36m-x86_64-linux-gnu.so, но я хочу назвать его только libwrapper.so, как я могу это сделать?

1 Ответ

0 голосов
/ 18 февраля 2020

Попробуйте следующий код. sysconfig.get_config_var('EXT_SUFFIX') возвращает спецификацию платформы c суффикс, который можно удалить из окончательного имени файла путем подкласса build_ext и переопределения get_ext_filename.

from distutils import sysconfig
from Cython.Distutils import build_ext
from distutils.core import setup
import os

class NoSuffixBuilder(build_ext):
    def get_ext_filename(self, ext_name):
        filename = super().get_ext_filename(ext_name)
        suffix = sysconfig.get_config_var('EXT_SUFFIX')
        ext = os.path.splitext(filename)[1]
        return filename.replace(suffix, "") + ext


setup(
    ....
    cmdclass={"build_ext": NoSuffixBuilder},
)

Окончательное имя файла будет test.so

...