openpmix-src 0.1.0

Vendored build of OpenPMIx, developed for use by the Lamellar runtime.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135

#!/usr/bin/env python3
#
# Copyright (c) 2022      Nanook Consulting. All rights reserved

# Note: as of Dec 2022, we still use setup.py with the Python
# setuptools package, even though setuptools docs now recommend
# upgrading to the setup.cfg or pyproject.toml methods.  Perhaps
# someday we'll migrate to the newer methods, but until then, setup.py
# seems to be working.

import os, sys
import warnings
import setuptools

from setuptools import Extension, setup
from Cython.Build import cythonize

from Cython.Compiler.Main import default_options
default_options['emit_linenums'] = True

def get_include():
    dirs = []
    for key in ['BUILDDIR', 'SRCDIR']:
        name = f'PMIX_BINDINGS_TOP_{key}'
        if name in os.environ:
            dirs.append(f'{os.environ[name]}/include')

    return dirs

def getVersion():
    dir = os.path.dirname(__file__)

    vers_path = os.path.join(dir, '../../include', 'pmix_version.h')
    if not os.path.exists(vers_path):
        include_dirs = get_include()
        vers_path = None
        for dir in include_dirs:
            tmp_path = os.path.join(dir, 'pmix_version.h')
            if os.path.exists(tmp_path):
                vers_path = tmp_path
                break
        if vers_path is None:
            print("Error: pmix_version.h does not exist at path: ",vers_path)
            sys.exit(1)

    with open(vers_path) as verFile:
        lines = verFile.readlines()
        for l in lines:
            if 'MAJOR' in l:
                major = l.split()[2]
                major = major[:-1]
            elif 'MINOR' in l:
                minor = l.split()[2]
                minor = minor[:-1]
            elif 'RELEASE' in l:
                release = l.split()[2]
                release = release[:-1]
        vers = [major, minor, release]
        version = ".".join(vers)
        return version

def package_setup(package_name, package_vers):
    '''
    Package setup routine.
    '''

    # Find the Cython source file.  Note that we always look in the
    # build tree.  The Makefile.am will ensure to always set this env
    # variable, but if someone invokes this script without setting
    # that env variable, just assume that pmix.pyx is in the same
    # directory as this script.
    key = 'PMIX_BINDINGS_TOP_BUILDDIR'
    if key in os.environ:
        dir = os.environ[key]
        src_filename = os.path.join(dir, "bindings", "python", "pmix.pyx")
    else:
        dir = os.path.dirname(sys.argv[0])
        src_filename = os.path.join(dir, "pmix.pyx")

    # Per comment at the top of this file, we know that setup.py is
    # deprecated/discouraged.  Until we switch away from it, remove
    # the setuptools "deprecated" warnings that are emitted to stderr
    # so that we do not scare the users.
    if hasattr(setuptools, "SetuptoolsDeprecationWarning"):
        warnings.filterwarnings("ignore",
                                category=setuptools.SetuptoolsDeprecationWarning)

    setup(
        name = 'pypmix',
        version = getVersion(),
        url = 'https://pmix.org',
        license = '3-clause BSD',
        author = 'Ralph H. Castain',
        author_email = 'rhc@pmix.org',
        description = 'Python bindings for PMIx',
        classifiers = [
                'Development Status :: 5 - Production/Stable',
                'Intended Audience :: Developers',
                'Topic :: HPC :: Parallel Programming :: System Management',
                'License :: OSI Approved :: BSD License',
                'Programming Language :: Python :: 3.4',
                'Programming Language :: Python :: 3.5',
                'Programming Language :: Python :: 3.6',
                'Programming Language :: Python :: 3.7',
                'Programming Language :: Python :: 3.8',
                'Programming Language :: Python :: 3.9',
                'Programming Language :: Python :: 3.10',
                'Programming Language :: Python :: 3.11',
                'Programming Language :: Python :: 3.12'],
        keywords = ['PMI', 'PMIx', 'HPC', 'MPI', 'SHMEM' ],
        platforms = 'any',
        install_requires = ["cython"],
        zip_safe = False,
        ext_modules = cythonize([Extension("pmix",
                                           [src_filename],
                                           libraries=["pmix"],
                                           depends=[], )],
                                compiler_directives={'language_level': 3}),
        include_dirs = get_include()
    )

def main():
    '''
    The main entry point for this program.
    '''
    package_name = 'pypmix'
    package_vers = getVersion()

    package_setup(package_name, package_vers)

    return os.EX_OK


if __name__ == '__main__':
    sys.exit(main())