dtc/pylibfdt/setup.py

#!/usr/bin/env python2

"""
setup.py file for SWIG libfdt
Copyright (C) 2017 Google, Inc.
Written by Simon Glass <sjg@chromium.org>
"""

from distutils.core import setup, Extension
import os
import re
import sys


VERSION_PATTERN = '^#define DTC_VERSION "DTC ([^"]*)"$'


def get_version():
    version_file = "version_gen.h"
    f = open(version_file, 'rt')
    m = re.match(VERSION_PATTERN, f.readline())
    return m.group(1)


libfdt_module = Extension(
    '_libfdt',
    sources = ['pylibfdt/libfdt.i'],
    include_dirs = ['libfdt'],
    libraries = ['fdt'],
    library_dirs = ['libfdt'],
)

setup(
    name='libfdt',
    version= get_version(),
    author='Simon Glass <sjg@chromium.org>',
    description='Python binding for libfdt',
    ext_modules=[libfdt_module],
    py_modules=['pylibfdt/libfdt'],
)