from contextlib import redirect_stdout as r_stdout
from contextlib import redirect_stderr as r_stderr
from io import StringIO

import pip


def pip_exec(*args):
    with StringIO() as stdout, StringIO() as stderr:
        with r_stdout(stdout), r_stderr(stderr):
            rcode = pip.main(list(args))
        return rcode, stdout.getvalue(), stderr.getvalue()


if __name__ == "__main__":
    import sys
    rcode, stdout, stderr = pip_exec(*sys.argv[1:])
    if rcode:
        raise RuntimeError("pip failed", rcode, stdout, stderr)
    else:
        for line in stdout.splitlines() + stderr.splitlines():
            print(line.rstrip())