Add naming for docker images

restful_tango/server.py

@@ -144,7 +144,7 @@ def post(self, key):
         """post - Handles the post request to build."""
         name = self.tempfile.name
         self.tempfile.close()
-        self.write(tangoREST.build(key, name))
+        self.write(tangoREST.build(key, name, self.request.headers["imageName"]))
 
 
 async def main(port: int):

restful_tango/tangoREST.py

@@ -443,7 +443,7 @@ def prealloc(self, key, image, num, vmStr):
             self.log.info("Key not recognized: %s" % key)
             return self.status.wrong_key
 
-    def build(self, key, tempfile):
+    def build(self, key, tempfile, imageName):
         self.log.debug("Received build request(%s)" % (key))
         if self.validateKey(key):
             if Config.VMMS_NAME != "localDocker":
@@ -454,13 +454,22 @@ def build(self, key, tempfile):
                 client = docker.from_env()
                 imageTarStr = open(tempfile, "rb").read()
                 images = client.images.load(imageTarStr)
-                id_list = ", ".join(image.short_id for image in images)
+                if len(images) != 1:
+                    raise Exception(
+                        "Wrong number of images built:  %s" % str(len(images))
+                    )
+                image = images[0]
+                imageName = imageName + ":latest"
+                image.tag(imageName)
+                for tag in image.tags:
+                    if tag != imageName:
+                        client.images.remove(tag)
             except Exception as e:
                 self.log.error("Image build failed: " + str(e))
                 os.unlink(tempfile)
                 return self.status.image_build_failed
 
-            self.log.info("Successfully loaded images: %s" % (id_list))
+            self.log.info("Successfully loaded image: %s" % (imageName))
             os.unlink(tempfile)
             return self.status.image_built
         else: