Allow specifying registry with build-compose (#13)

mozilla · Apr 18, 2019 · b1c9ca4 · b1c9ca4
1 parent b47e9e3
commit b1c9ca4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/taskboot/build.py b/taskboot/build.py
@@ -100,6 +100,8 @@ def build_compose(target, args):
         context = os.path.realpath(os.path.join(root, build.get('context', '.')))
         dockerfile = os.path.realpath(os.path.join(context, build.get('dockerfile', 'Dockerfile')))
         tag = service.get('image', name)
+        if args.registry:
+            tag = '{}/{}'.format(args.registry, tag)
         docker.build(context, dockerfile, tag)
 
         # Write the produced image

diff --git a/taskboot/cli.py b/taskboot/cli.py
@@ -67,6 +67,7 @@ def main():
         help='Path to docker-compose.yml to use',
         default='docker-compose.yml',
     )
+    compose.add_argument('--registry', type=str, help='Docker registry to use in images tags')
     compose.add_argument('--write', type=str, help='Directory to write the docker images')
     compose.set_defaults(func=build_compose)