def add(ctx, project, plugin_type, plugin_name, **flags):
if flags["custom"]:
if plugin_type in ("transformer", "transform", "orchestrator"):
click.secho(
f"--custom is not supported for {ctx.invoked_subcommand}")
raise click.Abort()
add_service = ProjectAddCustomService(project)
else:
add_service = ProjectAddService(project)
add_plugin(
add_service,
project,
PluginType(f"{plugin_type}s"),
plugin_name,
include_related=flags["include_related"],
)
tracker = GoogleAnalyticsTracker(project)
tracker.track_meltano_add(plugin_type=plugin_type, plugin_name=plugin_name)
def add_plugin(
project: Project,
plugin_type: PluginType,
plugin_name: str,
add_service: ProjectAddService,
):
try:
plugin = add_service.add(plugin_type, plugin_name)
if plugin.should_add_to_file(project):
click.secho(
f"Added {plugin_type.descriptor} '{plugin_name}' to your Meltano project",
fg="green",
)
else:
click.secho(
f"Adding {plugin_type.descriptor} '{plugin_name}' to your Meltano project...",
fg="green",
)
except PluginAlreadyAddedException as err:
click.secho(
f"{plugin_type.descriptor.capitalize()} '{plugin_name}' is already in your Meltano project",
fg="yellow",
err=True,
)
plugin = err.plugin
except (PluginNotSupportedException, PluginNotFoundError) as err:
click.secho(
f"Error: {plugin_type.descriptor} '{plugin_name}' is not known to Meltano",
fg="red",
)
raise click.Abort()
tracker = GoogleAnalyticsTracker(project)
tracker.track_meltano_add(plugin_type=plugin_type, plugin_name=plugin_name)
return plugin
def orchestrator(ctx, project, plugin_name):
add_plugin(ctx.obj["add_service"], project, PluginType.ORCHESTRATORS, plugin_name)
tracker = GoogleAnalyticsTracker(project)
tracker.track_meltano_add(plugin_type="orchestrator", plugin_name=plugin_name)
def transformer(ctx, project, plugin_name):
add_plugin(ctx.obj["add_service"], project, PluginType.TRANSFORMERS, plugin_name)
tracker = GoogleAnalyticsTracker(project)
tracker.track_meltano_add(plugin_type="transformer", plugin_name=plugin_name)
def loader(ctx, project, plugin_name):
add_plugin(ctx.obj["add_service"], project, PluginType.LOADERS, plugin_name)
tracker = GoogleAnalyticsTracker(project)
tracker.track_meltano_add(plugin_type="loader", plugin_name=plugin_name)
def model(ctx, project, plugin_name):
add_plugin(ctx.obj["add_service"], project, PluginType.MODELS, plugin_name)
tracker = GoogleAnalyticsTracker(project)
tracker.track_meltano_add(plugin_type="model", plugin_name=plugin_name)
def extractor(ctx, project, plugin_name):
add_plugin(ctx.obj["add_service"], project, PluginType.EXTRACTORS, plugin_name)
tracker = GoogleAnalyticsTracker(project)
tracker.track_meltano_add(plugin_type="extractor", plugin_name=plugin_name)
def transform(project, plugin_name):
add_transform(project, plugin_name)
tracker = GoogleAnalyticsTracker(project)
tracker.track_meltano_add(plugin_type="transform", plugin_name=plugin_name)