diff options
Diffstat (limited to 'create/create_common.py')
-rw-r--r-- | create/create_common.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/create/create_common.py b/create/create_common.py index 1d1a5212..a5694d7b 100644 --- a/create/create_common.py +++ b/create/create_common.py @@ -146,7 +146,7 @@ def GetCvdHostPackage(package_path=None): '\n'.join(dirs_to_check)) -def FindLocalImage(path, default_name_pattern): +def FindLocalImage(path, default_name_pattern, raise_error=True): """Find an image file in the given path. Args: @@ -165,7 +165,9 @@ def FindLocalImage(path, default_name_pattern): names = [name for name in os.listdir(path) if re.fullmatch(default_name_pattern, name)] if not names: - raise errors.GetLocalImageError("No image in %s." % path) + if raise_error: + raise errors.GetLocalImageError("No image in %s." % path) + return None if len(names) != 1: raise errors.GetLocalImageError("More than one image in %s: %s" % (path, " ".join(names))) |