toollib.py
55 lines
| 1.4 KiB
| text/x-python
|
PythonLexer
/ tools / toollib.py
Fernando Perez
|
r2118 | """Various utilities common to IPython release and maintenance tools. | ||
""" | ||||
Fernando Perez
|
r4450 | from __future__ import print_function | ||
Fernando Perez
|
r2118 | # Library imports | ||
import os | ||||
import sys | ||||
from distutils.dir_util import remove_tree | ||||
# Useful shorthands | ||||
pjoin = os.path.join | ||||
cd = os.chdir | ||||
Fernando Perez
|
r4452 | # Constants | ||
# SSH root address of the archive site | ||||
archive = 'ipython@archive.ipython.org:archive.ipython.org' | ||||
Fernando Perez
|
r2118 | # Utility functions | ||
Fernando Perez
|
r3197 | def sh(cmd): | ||
"""Run system command in shell, raise SystemExit if it returns an error.""" | ||||
Fernando Perez
|
r4450 | print("$", cmd) | ||
Fernando Perez
|
r2119 | stat = os.system(cmd) | ||
#stat = 0 # Uncomment this and comment previous to run in debug mode | ||||
Fernando Perez
|
r2118 | if stat: | ||
raise SystemExit("Command %s failed with code: %s" % (cmd, stat)) | ||||
Fernando Perez
|
r3197 | # Backwards compatibility | ||
c = sh | ||||
Fernando Perez
|
r2118 | |||
def get_ipdir(): | ||||
"""Get IPython directory from command line, or assume it's the one above.""" | ||||
# Initialize arguments and check location | ||||
try: | ||||
ipdir = sys.argv[1] | ||||
except IndexError: | ||||
ipdir = '..' | ||||
ipdir = os.path.abspath(ipdir) | ||||
cd(ipdir) | ||||
if not os.path.isdir('IPython') and os.path.isfile('setup.py'): | ||||
raise SystemExit('Invalid ipython directory: %s' % ipdir) | ||||
return ipdir | ||||
def compile_tree(): | ||||
"""Compile all Python files below current directory.""" | ||||
Fernando Perez
|
r3197 | stat = os.system('python -m compileall .') | ||
Fernando Perez
|
r2118 | if stat: | ||
msg = '*** ERROR: Some Python files in tree do NOT compile! ***\n' | ||||
msg += 'See messages above for the actual file that produced it.\n' | ||||
raise SystemExit(msg) | ||||