setup.py

import os
import subprocess
from setuptools import setup, find_packages, Extension


# hum... borrowed from psycopg2
def get_pg_config(kind, pg_config="pg_config"):
    p = subprocess.Popen([pg_config, '--%s' % kind], stdout=subprocess.PIPE)
    r = p.communicate()
    r = r[0].strip().decode('utf8')
    if not r:
        raise Warning(p[2].readline())
    return r


include_dirs = [get_pg_config(d) for d in ("includedir", "pkgincludedir", "includedir-server")]

multicorn_utils_module = Extension('multicorn._utils',
                                   include_dirs=include_dirs,
                                   extra_compile_args=['-shared'],
                                   sources=['src/utils.c'])


def get_version():
    script_dir = os.path.dirname(os.path.realpath(__file__))
    with open(os.path.join(script_dir, 'multicorn.control')) as f:
        for line in f:
            if line.startswith('default_version'):
                return line.strip().split("'")[-2]
        raise RuntimeError('default_version not found in multicorn.control')


setup(
    version=get_version(),
    ext_modules=[multicorn_utils_module]
)