def _get_version_tags()

in sagemaker_image_builder/main.py [0:0]

• 14 lines of code
• 8 McCabe index (conditional complexity)

def _get_version_tags(target_version: Version, env_out_file_name: str) -> list[str]:
    # First, add '2.6.x' as is.
    res = [str(target_version)]
    prerelease_version_suffix = f"-{target_version.prerelease}" if target_version.prerelease else ""

    # If we were to add '2.6', check if '2.6.(x+1)' is present.
    if not is_exists_dir_for_version(_get_next_version(target_version, "bump_patch"), env_out_file_name):
        res.append(f"{target_version.major}.{target_version.minor}{prerelease_version_suffix}")
    else:
        return res

    # If we were to add '2', check if '2.7' is present.
    if not is_exists_dir_for_version(_get_next_version(target_version, "bump_minor"), env_out_file_name):
        res.append(f"{target_version.major}{prerelease_version_suffix}")
    else:
        return res

    # If we were to add 'latest', check if '3.0.0' is present.
    if not is_exists_dir_for_version(_get_next_version(target_version, "bump_major"), env_out_file_name):
        res.append(f"latest{prerelease_version_suffix}")

    return res