build.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
# @Author: José Sánchez-Gallego (gallegoj@uw.edu)
# @Date: 2019-12-17
# @Filename: build.py
# @License: BSD 3-clause (http://www.opensource.org/licenses/BSD-3-Clause)

# Extension build system using poetry, see
# https://github.com/python-poetry/poetry/issues/11.

import glob
import os
import shutil
import sys

from setuptools import Distribution, Extension
from setuptools.command.build_ext import build_ext


LIBFLI_PATH = os.path.join(os.path.dirname(__file__), "cextern/libfli-1.999.1-180223")

RTD = os.environ.get("READTHEDOCS", False)


def get_directories():
    dirs = [LIBFLI_PATH]

    if sys.platform in ["linux", "darwin", "unix"]:
        dirs.append(os.path.join(LIBFLI_PATH, "unix"))
        if not RTD:  # Add the libusb directory except when in RTD.
            dirs.append(os.path.join(LIBFLI_PATH, "unix", "libusb"))

    return dirs


def get_sources():
    dirs = get_directories()

    sources = []
    for dir_ in dirs:
        sources += glob.glob(dir_ + "/*.c")

    return sources


extra_compile_args = ["-O3", "-fPIC", "-g"]
extra_link_args = ["-nostartfiles"]

# Do not use libusb on RTD because it makes the build fail.
# This still creates a usable library and we are mocking the device anyway.
if RTD:
    libraries = ["m"]
else:
    libraries = ["m", "usb-1.0"]


ext_modules = [
    Extension(
        "flicamera.libfli",
        sources=get_sources(),
        include_dirs=get_directories(),
        libraries=libraries,
        define_macros=[("__LIBUSB__", "1")],
        extra_compile_args=extra_compile_args,
        extra_link_args=extra_link_args,
        language="c",
        optional=False,
    ),
]


# The *args is needed because the autogenerated setup.py sends arguments
# (this is a leftover from the old Poetry build system).
def build(*args, **setup_kwargs):
    distribution = Distribution({"name": "extended", "ext_modules": ext_modules})

    cmd = build_ext(distribution)
    cmd.ensure_finalized()
    cmd.run()

    # Copy built extensions back to the project
    for output in cmd.get_outputs():
        relative_extension = os.path.relpath(output, cmd.build_lib)
        shutil.copyfile(output, relative_extension)  # type: ignore
        mode = os.stat(relative_extension).st_mode
        mode |= (mode & 0o444) >> 2
        os.chmod(relative_extension, mode)


if __name__ == "__main__":
    build()