mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Giacomo Vercesi
f626dece53
Add the `revng2 project dump-pipeline` command which allows emitting the pipeline description from the command-line.
72 lines
1.9 KiB
Python
72 lines
1.9 KiB
Python
#
|
|
# This file is distributed under the MIT License. See LICENSE.md for details.
|
|
#
|
|
|
|
import os
|
|
|
|
import click
|
|
|
|
from revng.pypeline.cli.context import ClickContext, pass_context
|
|
from revng.pypeline.cli.utils import EagerParsedPath, PypeGroup, StorageProviderUrl
|
|
from revng.pypeline.pipeline import Pipeline
|
|
from revng.pypeline.pipeline_parser import load_pipeline_yaml_file
|
|
from revng.pypeline.utils import cache_directory
|
|
|
|
from .analyze import analyze
|
|
from .artifact import artifact
|
|
from .daemon import run_daemon
|
|
from .dump_pipeline import dump_pipeline
|
|
|
|
|
|
@click.group(
|
|
cls=PypeGroup,
|
|
help="Project commands (porcelain)",
|
|
)
|
|
@click.option(
|
|
"--pipeline",
|
|
"pipeline",
|
|
type=EagerParsedPath(
|
|
name="pipeline",
|
|
parser=lambda path, _ctx: load_pipeline_yaml_file(path),
|
|
),
|
|
help='Path to the pipeline file. Defaults to the "PYPELINE_PIPELINE" environment if set',
|
|
default="pipeline.yml",
|
|
envvar="PYPELINE_PIPELINE",
|
|
show_default=True,
|
|
)
|
|
@click.option(
|
|
"--storage-provider",
|
|
"storage_provider",
|
|
type=StorageProviderUrl(),
|
|
help="The URL of the storage provider to use.",
|
|
default="local://",
|
|
envvar="PYPELINE_STORAGE_PROVIDER",
|
|
show_default=True,
|
|
)
|
|
@click.option(
|
|
"--cache-dir",
|
|
"cache_dir",
|
|
type=click.Path(exists=False, file_okay=False, dir_okay=True, writable=True),
|
|
help="The directory to use for caching.",
|
|
default=str(cache_directory()),
|
|
show_default=True,
|
|
)
|
|
@pass_context
|
|
def project(
|
|
ctx: ClickContext,
|
|
pipeline: Pipeline,
|
|
storage_provider: str,
|
|
cache_dir: str,
|
|
) -> None:
|
|
os.makedirs(ctx.params["cache_dir"], exist_ok=True)
|
|
# Store the params so the subcommands can access them
|
|
ctx.obj.cache_dir = cache_dir
|
|
ctx.obj.storage_provider_url = storage_provider
|
|
ctx.obj.pipeline = pipeline
|
|
|
|
|
|
project.add_command(analyze)
|
|
project.add_command(artifact)
|
|
project.add_command(run_daemon, name="daemon")
|
|
project.add_command(dump_pipeline, name="dump-pipeline")
|