Set labels when building image from Dockerfile

repo2docker/__main__.py

@@ -204,6 +204,14 @@ def get_argparser():
 
     argparser.add_argument("--appendix", type=str, help=Repo2Docker.appendix.help)
 
+    argparser.add_argument(
+        "--label",
+        dest="labels",
+        action="append",
+        help="Extra label to set on the image, in form name=value",
+        default=[],
+    )
+
     argparser.add_argument("--subdir", type=str, help=Repo2Docker.subdir.help)
 
     argparser.add_argument(
@@ -246,6 +254,13 @@ def make_r2d(argv=None):
     if args.appendix:
         r2d.appendix = args.appendix
 
+    for l in args.labels:
+        if "=" in l:
+            key, val = l.split("=", 1)
+            r2d.labels[key] = val
+        else:
+            r2d.labels[l] = ""
+
     r2d.repo = args.repo
     r2d.ref = args.ref
 

repo2docker/app.py

@@ -237,6 +237,17 @@ def _user_name_default(self):
         """,
     )
 
+    labels = Dict(
+        {},
+        help="""
+        Extra labels to set on the final image.
+
+        Each Label is a key-value pair, with the key being the name of the label
+        and the value its value.
+        """,
+        config=True,
+    )
+
     json_logs = Bool(
         False,
         help="""
@@ -747,6 +758,8 @@ def build(self):
                 picked_buildpack.labels["repo2docker.repo"] = repo_label
                 picked_buildpack.labels["repo2docker.ref"] = self.ref
 
+                picked_buildpack.labels.update(self.labels)
+
                 if self.dry_run:
                     print(picked_buildpack.render())
                 else:

repo2docker/buildpacks/docker.py

@@ -52,6 +52,7 @@ def build(
             buildargs=build_args,
             container_limits=limits,
             cache_from=cache_from,
+            labels=self.get_labels(),
         )
 
         build_kwargs.update(extra_build_kwargs)

repo2docker/docker.py

@@ -73,6 +73,7 @@ def build(
         dockerfile="",
         fileobj=None,
         path="",
+        labels=None,
         **kwargs,
     ):
         return self._apiclient.build(
@@ -87,6 +88,7 @@ def build(
             dockerfile=dockerfile,
             fileobj=fileobj,
             path=path,
+            labels=labels,
             **kwargs,
         )
 

repo2docker/engine.py

@@ -176,6 +176,7 @@ def build(
         dockerfile="",
         fileobj=None,
         path="",
+        labels=None,
         **kwargs,
     ):
         """
@@ -204,6 +205,8 @@ def build(
             A tar file-like object containing the build context
         path : str
             path to the Dockerfile
+        labels : dict
+            Dictionary of labels to set on the image
 
         Returns
         -------