def lock()

in tasks.py [0:0]

• 32 lines of code
• 7 McCabe index (conditional complexity)

def lock(c, upgrade=False, upgrade_package="", quiet=True):
    """Lock and synchronise environments."""

    REQ_ALL_IN = os.path.join(BASE_DIR, "reqs", "requirements_all.in")

    opt_quiet = "-q" if quiet else ""

    opt_upgrade = ""
    opt_upgrade_package = ""
    if upgrade:
        opt_upgrade = "--upgrade "
    if upgrade_package:
        opt_upgrade_package = f"--upgrade-package {upgrade_package}"

    # Lock the consolidated requirements
    c.run(
        f'"{find_uv_bin()}" pip compile '
        f"{opt_quiet} {opt_upgrade} {opt_upgrade_package} "
        f'"{os.path.relpath(REQ_ALL_IN, os.getcwd())}" -o "{os.path.relpath(REQS_ALL, os.getcwd())}"'
    )

    # Generate a constraints file
    CONSTRAINTS = os.path.join(BASE_DIR, "reqs", "constraints.txt")
    c.run(f'cat "{REQS_ALL}" | grep -v "^-e " > "{CONSTRAINTS}"')

    # Gather all requirements
    reqs_in = list(
        glob.glob(os.path.join(BASE_DIR, "**", "pyproject.toml"), recursive=True)
    )
    reqs_in += list(
        glob.glob(os.path.join(BASE_DIR, "**", "requirements.in"), recursive=True)
    )

    # Re-generate the requirements.txt for specific deployments
    # (honouring consolidated requirements)
    REL_CONSTRAINTS = os.path.relpath(CONSTRAINTS, os.getcwd())
    for req_in in reqs_in:
        req_txt = os.path.join(os.path.dirname(req_in), "requirements.txt")
        if os.path.exists(req_txt):
            c.run(
                f'"{find_uv_bin()}" pip compile {opt_quiet} --generate-hashes '
                f'-c "{REL_CONSTRAINTS}" "{os.path.relpath(req_in, os.getcwd())}" | '
                'grep -v "\\${PROJECT_ROOT}" >'
                f'"{req_txt}"'
            )