Include docker image sha in hash (openproblems-bio#505)

* include docker image sha in hash

* bugfixes

Author: scottgigante-immunai

openproblems/api/hash.py

@@ -1,4 +1,3 @@
-from ..tools.utils import check_version
 from . import utils
 
 import hashlib
@@ -31,6 +30,21 @@ def git_hash(file):
         return p.stdout.decode()
 
 
+def docker_hash(image_name):
+    """Get the docker image hash associated with an image."""
+    subprocess.run(["docker", "pull", "-q", image_name])
+    p = subprocess.run(
+        ["docker", "inspect", "--format='{{index .RepoDigests 0}}'", image_name],
+        stdout=subprocess.PIPE,
+        stderr=subprocess.PIPE,
+        cwd=os.path.dirname(__file__),
+    )
+    if p.returncode != 0:
+        raise RuntimeError(p.stderr.decode())
+    else:
+        return p.stdout.decode()
+
+
 def get_context(obj, context=None):
     """Return a dictionary describing the context of an object.
 
@@ -51,25 +65,19 @@ def get_context(obj, context=None):
     if isinstance(obj, _MODULE):
         module_name = obj.__name__
     else:
-        if isinstance(obj, scprep.run.RFunction):
-            try:
+        if isinstance(obj, scprep.run.RFunction) and hasattr(obj, "__r_file__"):
+            if obj.__r_file__ not in context:
                 context[obj.__r_file__] = git_hash(obj.__r_file__)
-            except AttributeError:
-                pass
+        if hasattr(obj, "metadata") and "image" in obj.metadata:
+            image_name = f"singlecellopenproblems/{obj.metadata['image']}"
+            if image_name not in context:
+                context[image_name] = docker_hash(image_name)
         try:
             module_name = get_module(obj)
         except AttributeError:
             # doesn't belong to a module
             return context
-    if module_name in context:
-        # already done
-        pass
-    elif not module_name.startswith("openproblems"):
-        # use version number of external libs
-        version = check_version(module_name.split(".")[0])
-        if version != "ModuleNotFound":
-            context[module_name] = version
-    else:
+    if module_name.startswith("openproblems") and module_name not in context:
         module = importlib.import_module(module_name)
         context[module_name] = git_hash(module.__file__)
         for member_name in dir(module):