def _delete_all_files_except_additional_packages_input_files()

in sagemaker_image_builder/main.py [0:0]

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

def _delete_all_files_except_additional_packages_input_files(base_version_dir, image_config):
    additional_package_env_in_files = [
        image_generator_config["additional_packages_env_in_file"] for image_generator_config in image_config
    ]
    for filename in os.listdir(base_version_dir):
        if filename not in additional_package_env_in_files:
            file_path = os.path.join(base_version_dir, filename)
            try:
                if os.path.isfile(file_path) or os.path.islink(file_path):
                    os.unlink(file_path)
                elif os.path.isdir(file_path):
                    shutil.rmtree(file_path)
            except Exception as e:
                print("Failed to delete %s. Reason: %s" % (file_path, e))