setup.py

"""
Infrastructure Agent: Setup configuration
Copyright (C) 2003-2026 ITRS Group Ltd. All rights reserved
"""

import os
import sys

TOX = 'TOX_WORK_DIR' in os.environ
SCRIPT_DIR = os.path.dirname(__file__)

if TOX:
    from setuptools import setup
else:
    from cx_Freeze import Executable, setup

try:
    with open(os.path.join(SCRIPT_DIR, 'VERSION'), 'r') as f_in:
        VERSION = f_in.read().strip()
except FileNotFoundError:
    VERSION = '0.0.0'  # Probably from `make test` or similar

EXECUTABLE_CONFIG = {
    "copyright": "Copyright (C) 2003-2026 ITRS Group Ltd.",
    "icon": "icon.ico"
}

build_exe_options = {
    'excludes': ['asyncio', 'unittest', 'tkinter', 'test', 'mock'],
}

if TOX:
    executables = []
elif sys.platform.startswith('win32'):
    executables = [
        # Service Executable
        Executable(script="win_svce_config.py", target_name="infra-svce.exe", base="service", **EXECUTABLE_CONFIG),
        # Main Executable
        Executable(script="main.py", target_name="infra-agent.exe", **EXECUTABLE_CONFIG)
    ]
    build_exe_options = build_exe_options | {
        "include_files": ["icon.ico"],
        "includes": ["win_svce_handler"],
        # pywin32 packages required by cx_Freeze "service" base for --install/--uninstall and service runtime
        "packages": ["win32serviceutil", "win32service", "win32event", "servicemanager", "win32timezone"],
        # Not setting include_msvcr as we bundle the eventual MSI with the Visual C++ redistributable installer
    }
else:
    executables = [
        Executable(script='main.py', target_name="infrastructure-agent", base='Console', **EXECUTABLE_CONFIG)
    ]

setup_kwargs = {
    "name": "infrastructure-agent",
    "version": VERSION,
    "description": "Infrastructure Agent",
    "author": "ITRS Group Ltd",
    "author_email": "",
    "url": "https://itrsgroup.com/",
    "options": {"build_exe": build_exe_options},
    "executables": executables,
    "py_modules": [],
}
if TOX:
    setup_kwargs["tests_require"] = [
        "coverage",
        "flake8",
        "pytest",
        "pytest-cov",
        "pytest-mock",
    ]
setup(**setup_kwargs)