p4 - perforce utility

p5.py

import argparse
from P4 import P4, P4Exception


def edit_files(p4, cl):
    files = p4.run("describe", "-s", cl)[0]["depotFile"]
    change = {"Change": "new", "Description": f"Checked out from CL{cl}"}
    p4.input = change
    change_output = p4.run("change", "-i")
    new_cl = change_output[0].split()[1]
    for file in files:
        print(file)
        p4.run("edit", "-c", new_cl, file)


def main():
    parser = argparse.ArgumentParser(prog="p5", description="p4 utility")

    subparsers = parser.add_subparsers(dest="command")

    checkout_parser = subparsers.add_parser("edit")
    checkout_parser.add_argument("cl", help="Changelist to edit files from")

    args = parser.parse_args()
    
    p4 = P4()
    try:
        p4.connect()
        if args.command == "edit":
            print(f"Edit {args.cl}")
            edit_files(p4, args.cl)
    except P4Exception:
        for e in p4.errors:
            print(e)
    finally:
        p4.disconnect()


if __name__ == "__main__":
    main()