setup.py

from setuptools import setup
from setuptools.command.develop import develop
from setuptools.command.install import install
import subprocess
import os
import sys

def run_welcome_script():
    """Run src/installer/welcome.py.
    
    This is called at the top level to ensure it runs during the 
    metadata generation phase of `pip install`.
    """
    # Prevent running multiple times in the same session (pip often calls setup.py twice)
    if os.environ.get('XMEM_INSTALL_RUNNING') == '1':
        return
    os.environ['XMEM_INSTALL_RUNNING'] = '1'

    installer_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "src", "installer")
    welcome_file = os.path.join(installer_dir, "welcome.py")

    if not os.path.exists(welcome_file):
        return

    try:
        # Use stdout directly to bypass some pip output capturing
        result = subprocess.run(
            [sys.executable, welcome_file],
            check=False,
            timeout=20,
            capture_output=True,
            text=True,
        )
        # Print output so user can see it
        if result.stdout:
            print(result.stdout)
        if result.stderr:
            print(result.stderr)
    except subprocess.TimeoutExpired:
        print("[xmem] Welcome script timed out")
    except Exception as e:
        print(f"[xmem] Welcome script error: {e}")

# Run it immediately when setup.py is loaded
run_welcome_script()

class PostDevelopCommand(develop):
    def run(self):
        develop.run(self)

class PostInstallCommand(install):
    def run(self):
        install.run(self)

setup(
    cmdclass={
        "develop": PostDevelopCommand,
        "install": PostInstallCommand,
    },
)