diff options
-rw-r--r-- | buildstream/_frontend/main.py | 31 | ||||
-rw-r--r-- | buildstream/_pipeline.py | 38 |
2 files changed, 69 insertions, 0 deletions
diff --git a/buildstream/_frontend/main.py b/buildstream/_frontend/main.py index 79adf11d7..43cdc431f 100644 --- a/buildstream/_frontend/main.py +++ b/buildstream/_frontend/main.py @@ -209,6 +209,37 @@ def track(app, target, variant, deps, except_): ################################################################## +# Push Command # +################################################################## +@cli.command(short_help="Push a built artifact") +@click.option('--variant', + help='A variant of the specified target') +@click.option('--deps', '-d', default='none', + type=click.Choice(['none', 'all']), + help='The dependencies to fetch (default: none)') +@click.argument('target') +@click.pass_obj +def push(app, target, variant, deps): + """Push a built artifact to the configured remote artifact cache. + + Specify `--deps` to control which artifacts to push: + + \b + none: No dependencies, just the element itself + all: All dependencies + """ + app.initialize(target, variant) + try: + to_push = app.pipeline.deps_elements(deps) + app.pipeline.push(app.scheduler, to_push) + click.echo("") + except _BstError as e: + click.echo("") + click.echo("ERROR: {}".format(e)) + sys.exit(-1) + + +################################################################## # Show Command # ################################################################## @cli.command(short_help="Show elements in the pipeline") diff --git a/buildstream/_pipeline.py b/buildstream/_pipeline.py index 64ce4d679..8092340fc 100644 --- a/buildstream/_pipeline.py +++ b/buildstream/_pipeline.py @@ -631,6 +631,44 @@ class Pipeline(): self.open_workspace(scheduler, workspace_dir, source_index, no_checkout, track, False) + # push() + # + # Pushes elements in the pipeline + # + # Args: + # scheduler (Scheduler): The scheduler to run this pipeline on + # elements (list): List of elements to push + # + def push(self, scheduler, elements): + + if not self.artifacts.can_push(): + self.message(self.target, MessageType.FAIL, "Not configured for pushing artifacts") + + plan = elements + self.assert_consistent(plan) + self.session_elements = len(plan) + + push = PushQueue() + push.enqueue(plan) + queues = [push] + + self.message(self.target, MessageType.START, "Pushing {} artifacts".format(len(plan))) + elapsed, status = scheduler.run(queues) + pushed = len(push.processed_elements) + + if status == SchedStatus.ERROR: + self.message(self.target, MessageType.FAIL, "Push failed", elapsed=elapsed) + raise PipelineError() + elif status == SchedStatus.TERMINATED: + self.message(self.target, MessageType.WARN, + "Terminated after pushing {} elements".format(pushed), + elapsed=elapsed) + raise PipelineError() + else: + self.message(self.target, MessageType.SUCCESS, + "Pushed {} complete".format(pushed), + elapsed=elapsed) + # remove_elements(): # # Internal function |