tools/toollib.py

"""Various utilities common to IPython release and maintenance tools.
"""

# Library imports
import os
import sys

from pathlib import Path

# Useful shorthands
cd = os.chdir

# Constants

# SSH root address of the archive site
archive_user = '[e-mail]'
archive_dir = 'archive.ipython.org'
archive = '%s:%s' % (archive_user, archive_dir)

# Build commands
# Source dists
build_command = "{python} -m build".format(python=sys.executable)


# Utility functions
def sh(cmd):
    """Run system command in shell, raise SystemExit if it returns an error."""
    print("$", cmd)
    stat = os.system(cmd)
    #stat = 0  # Uncomment this and comment previous to run in debug mode
    if stat:
        raise SystemExit("Command %s failed with code: %s" % (cmd, stat))

def get_ipdir():
    """Get IPython directory from command line, or assume it's the one above."""

    # Initialize arguments and check location
    ipdir = Path(__file__).parent / os.pardir
    ipdir = ipdir.resolve()

    cd(ipdir)
    if not Path("IPython").is_dir() and Path("setup.py").is_file():
        raise SystemExit("Invalid ipython directory: %s" % ipdir)
    return ipdir

def execfile(fname, globs, locs=None):
    locs = locs or globs
    exec(compile(open(fname, encoding="utf-8").read(), fname, "exec"), globs, locs)