#!/usr/bin/env python
import os
import shutil
from contextlib import contextmanager
from subprocess import check_call
_dname = os.path.dirname
REPO_ROOT = _dname(_dname(_dname(os.path.abspath(__file__))))
@contextmanager
def cd(path):
"""Change directory while inside context manager."""
cwd = os.getcwd()
try:
os.chdir(path)
yield
finally:
os.chdir(cwd)
def run(command):
return check_call(command, shell=True)
if __name__ == "__main__":
with cd(REPO_ROOT):
run("pip install -r requirements.txt")
run("python scripts/ci/install-dev-deps")
if os.path.isdir("dist") and os.listdir("dist"):
shutil.rmtree("dist")
run("python setup.py bdist_wheel")
wheel_dist = os.listdir("dist")[0]
run("pip install %s" % (os.path.join("dist", wheel_dist)))
