Differential D5036 Diff 18361 swh/clearlydefined/cli.py

Changeset View

Standalone View

swh/clearlydefined/cli.py

				# Copyright (C) 2021 The Software Heritage developers
				# See the AUTHORS file at the top-level directory of this distribution
				# License: GNU Affero General Public License version 3, or any later version
				# See top-level LICENSE file for more information

				import os

	import click			import click

				from swh.clearlydefined.orchestrator import orchestrator
	from swh.core.cli import CONTEXT_SETTINGS			from swh.core.cli import CONTEXT_SETTINGS
	from swh.core.cli import swh as swh_cli_group			from swh.core.cli import swh as swh_cli_group
				from swh.storage import get_storage


	@swh_cli_group.group(name="clearlydefined", context_settings=CONTEXT_SETTINGS)			@swh_cli_group.group(name="clearlydefined", context_settings=CONTEXT_SETTINGS)
				@click.option(
				"--config-file",
				"-C",
				default=None,
				type=click.Path(
				exists=True,
				dir_okay=False,
				),
				help="SWH storage config.",
				)
				@click.option("--clearcode-dsn", default=None, type=click.STRING, help="Clearcode DSN.")
	@click.pass_context			@click.pass_context
	def clearlydefined_cli_group(ctx):			def clearlydefined(ctx, config_file, clearcode_dsn):
	"""Foo main command.			"""Software Heritage Clearlydefined Metadata Fetcher"""
	"""			from swh.core import config

				if config_file:
				if not os.path.exists(config_file):
				raise ValueError("%s does not exist" % config_file)
				conf = config.read(config_file)
				else:
				conf = {}

				if "storage" not in conf:
				ctx.fail("You must have a storage configured in your config file.")

				ctx.ensure_object(dict)
				ctx.obj["config"] = conf
				ctx.obj["dsn"] = clearcode_dsn


	@clearlydefined_cli_group.command()			@clearlydefined.command(name="fill_storage")
	@click.option("--bar", help="Something")
	@click.pass_context			@click.pass_context
	def bar(ctx, bar):			def run_orchestration(ctx):
	"""Do something."""			print(ctx.obj["config"]["storage"])
				vlorentzUnsubmitted Done Inline Actions don't pass the config as JSON on the CLI. See how other packages do it. vlorentz: don't pass the config as JSON on the CLI. See how other packages do it.
	click.echo("bar")			storage = get_storage(**ctx.obj["config"]["storage"])
				clearcode_dsn = ctx.obj["dsn"]
				orchestrator(storage=storage, clearcode_dsn=clearcode_dsn)
				vlorentzUnsubmitted Done Inline Actions that's a CLI group, it shouldn't run the code itself. vlorentz: that's a CLI group, it shouldn't run the code itself.