slick-queue-py/setup.py at main · SlickQuant/slick-queue-py · GitHub

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
from setuptools import setup, Extension
import sys
import re
from pathlib import Path

# Read version from slick_queue_py.py
def get_version():
    with open('slick_queue_py.py', 'r', encoding='utf-8') as f:
        content = f.read()
        match = re.search(r"^__version__ = ['\"]([^'\"]*)['\"]", content, re.M)
        if match:
            return match.group(1)
        raise RuntimeError("Unable to find version string.")

# Read long description from README
def get_long_description():
    readme_path = Path(__file__).parent / 'README.md'
    if readme_path.exists():
        with open(readme_path, 'r', encoding='utf-8') as f:
            return f.read()
    return ''

# Define the C++ extension module (uses std::atomic for cross-platform atomic ops)
atomic_ops_ext = Extension(
    'atomic_ops_ext',
    sources=['atomic_ops_ext.cpp'],
    include_dirs=[],
    libraries=[],
    extra_compile_args=['/std:c++17', '/O2'] if sys.platform == 'win32' else ['-std=c++17', '-O2'],
)

setup(
    name='slick-queue-py',
    version=get_version(),
    description='Lock-free MPMC queue with C++ interoperability via shared memory',
    long_description=get_long_description(),
    long_description_content_type='text/markdown',
    author='Slick Quant',
    author_email='slickquant@slickquant.com',
    url='https://github.com/SlickQuant/slick-queue-py',
    project_urls={
        'Bug Tracker': 'https://github.com/SlickQuant/slick-queue-py/issues',
        'Documentation': 'https://github.com/SlickQuant/slick-queue-py#readme',
        'Source Code': 'https://github.com/SlickQuant/slick-queue-py',
    },
    classifiers=[
        'Development Status :: 4 - Beta',
        'Intended Audience :: Developers',
        'Topic :: Software Development :: Libraries :: Python Modules',
        'Topic :: System :: Distributed Computing',
        'License :: OSI Approved :: MIT License',
        'Programming Language :: Python :: 3',
        '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',
        'Programming Language :: C++',
        'Operating System :: Microsoft :: Windows',
        'Operating System :: POSIX :: Linux',
        'Operating System :: MacOS',
    ],
    keywords='queue, lock-free, atomic, shared-memory, ipc, multiprocessing, mpmc',
    ext_modules=[atomic_ops_ext],
    py_modules=['slick_queue_py', 'atomic_ops'],
    python_requires='>=3.8',
    license='MIT',
)