def modify_repository_name_for_context()

in src/image_builder.py [0:0]

• 12 lines of code
• 3 McCabe index (conditional complexity)

def modify_repository_name_for_context(image_repo_uri, build_context):
    """
    Modifies the repository name for mainline vs nightly builds.

    Args:
        image_repo_uri: the initial repository name.

    Returns:
        The modified repository name.
    """
    repo_uri_values = image_repo_uri.split("/")
    repo_name = repo_uri_values[-1]
    if build_context == "MAINLINE":
        repo_uri_values[-1] = repo_name.replace(
            constants.PR_REPO_PREFIX, constants.MAINLINE_REPO_PREFIX
        )
    elif build_context == "NIGHTLY":
        repo_uri_values[-1] = repo_name.replace(
            constants.PR_REPO_PREFIX, constants.NIGHTLY_REPO_PREFIX
        )
    return "/".join(repo_uri_values)