Differential D2453 Diff 8667 swh/icinga_plugins/cli.py

Changeset View

Standalone View

swh/icinga_plugins/cli.py

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

				import sys

	import click			import click

	from swh.core.cli import CONTEXT_SETTINGS			from swh.core.cli import CONTEXT_SETTINGS

				from .vault import VaultCheck


	@click.group(name='foo', context_settings=CONTEXT_SETTINGS)			@click.group(name='icinga_plugins', context_settings=CONTEXT_SETTINGS)
				@click.option('--swh-storage-url', type=str,
				help='URL to an swh-storage HTTP API')
				@click.option('--swh-web-url', type=str,
				help='URL to an swh-web instance')
	@click.pass_context			@click.pass_context
	def cli(ctx):			def cli(ctx, swh_storage_url, swh_web_url):
	"""Foo main command.			"""Main command for Icinga plugins
	"""			"""
				ctx.ensure_object(dict)
				ctx.obj['swh_storage_url'] = swh_storage_url
				ctx.obj['swh_web_url'] = swh_web_url


				@cli.group(name='check_vault')
				olasdUnsubmitted Not Done Inline Actions I think we've usually used dashes rather than underscores here. olasd: I think we've usually used dashes rather than underscores here.
				vlorentzAuthorUnsubmitted Done Inline Actions I was mirroring Icinga's doc instead of our conventions, but you're right. vlorentz: I was mirroring Icinga's doc instead of our conventions, but you're right.
				@click.option('--poll-interval', type=int, default=10,
				help='Interval (in seconds) between two polls to the API, '
				'to check for cooking status.')
				@click.pass_context
				def check_vault(ctx, poll_interval):
				ctx.obj['poll_interval'] = poll_interval


	@cli.command()			@check_vault.command(name='directory')
	@click.option('--bar', help='Something')
	@click.pass_context			@click.pass_context
	def bar(ctx, bar):			def check_vault_directory(ctx):
	'''Do something.'''			"""Picks a random directory, requests its cooking via swh-web,
	click.echo('bar')			and waits for completion."""
				sys.exit(VaultCheck(ctx.obj).main())