106 lines
2.9 KiB
Python
106 lines
2.9 KiB
Python
import functools
|
|
from enum import Enum
|
|
|
|
import click
|
|
from arrow import Arrow
|
|
from click import pass_context
|
|
|
|
from mobilizon_reshare.cli import safe_execution
|
|
from mobilizon_reshare.cli.commands.format.format import format_event
|
|
from mobilizon_reshare.cli.commands.inspect.inspect_event import inspect_events
|
|
from mobilizon_reshare.cli.commands.start.main import main as start_main
|
|
from mobilizon_reshare.cli.commands.recap.main import main as recap_main
|
|
from mobilizon_reshare.config.publishers import publisher_names
|
|
|
|
from mobilizon_reshare.event.event import EventPublicationStatus
|
|
|
|
status_name_to_enum = {
|
|
"waiting": EventPublicationStatus.WAITING,
|
|
"completed": EventPublicationStatus.COMPLETED,
|
|
"failed": EventPublicationStatus.FAILED,
|
|
"partial": EventPublicationStatus.PARTIAL,
|
|
"all": None,
|
|
}
|
|
|
|
settings_file_option = click.option(
|
|
"--settings-file",
|
|
type=click.Path(exists=True),
|
|
help="The path for the settings file. "
|
|
"Overrides the one specified in the environment variables.",
|
|
)
|
|
from_date_option = click.option(
|
|
"--begin",
|
|
type=click.DateTime(),
|
|
expose_value=True,
|
|
help="Include only events that begin after this datetime",
|
|
)
|
|
to_date_option = click.option(
|
|
"--end",
|
|
type=click.DateTime(),
|
|
expose_value=True,
|
|
help="Include only events that begin before this datetime",
|
|
)
|
|
|
|
|
|
class InspectTarget(Enum):
|
|
ALL = "all"
|
|
WAITING = "waiting"
|
|
|
|
def __str__(self):
|
|
return self.value
|
|
|
|
|
|
@click.group()
|
|
def mobilizon_reshare():
|
|
pass
|
|
|
|
|
|
@mobilizon_reshare.command(help="Synchronize and publish events")
|
|
@settings_file_option
|
|
def start(settings_file):
|
|
safe_execution(start_main, settings_file=settings_file)
|
|
|
|
|
|
@mobilizon_reshare.command(help="Publish a recap of already published events")
|
|
@settings_file_option
|
|
def recap(settings_file):
|
|
safe_execution(recap_main, settings_file=settings_file)
|
|
|
|
|
|
@mobilizon_reshare.command(help="Print events in the database that are in STATUS")
|
|
@from_date_option
|
|
@to_date_option
|
|
@click.argument(
|
|
"status", type=click.Choice(list(status_name_to_enum.keys())),
|
|
)
|
|
@settings_file_option
|
|
@pass_context
|
|
def inspect(ctx, status, begin, end, settings_file):
|
|
ctx.ensure_object(dict)
|
|
begin = Arrow.fromdatetime(begin) if begin else None
|
|
end = Arrow.fromdatetime(end) if end else None
|
|
|
|
safe_execution(
|
|
functools.partial(
|
|
inspect_events, status_name_to_enum[status], frm=begin, to=end,
|
|
),
|
|
settings_file,
|
|
)
|
|
|
|
|
|
@mobilizon_reshare.command(
|
|
help="Format and print event with mobilizon id EVENT-ID using the publisher's format named"
|
|
"PUBLISHER"
|
|
)
|
|
@settings_file_option
|
|
@click.argument("event-id", type=click.UUID)
|
|
@click.argument("publisher", type=click.Choice(publisher_names))
|
|
def format(settings_file, event_id, publisher):
|
|
safe_execution(
|
|
functools.partial(format_event, event_id, publisher), settings_file,
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
mobilizon_reshare()
|