fix: type check on image object passed to get_image_digest

blamebutton · Jul 29, 2020 · 8b0d020 · 8b0d020
1 parent f993c45
commit 8b0d020
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/main.py b/main.py
@@ -153,6 +153,8 @@ def get_image_digest(image: Image) -> Optional[str]:
     :param image: the image object to get the digest out of
     :return: the repository digest for the given image or None if it's not present
     """
+    if not isinstance(image, Image):
+        return None
     digests = image.attrs.get('RepoDigests')
     if digests:
         tag, digest = split_image(digests[0])