70 lines
1.9 KiB
Python
70 lines
1.9 KiB
Python
from pathlib import Path
|
|
from typing import Optional
|
|
|
|
import click
|
|
from libretime_shared.app import AbstractApp
|
|
from libretime_shared.cli import cli_config_options, cli_logging_options
|
|
from libretime_shared.config import DEFAULT_ENV_PREFIX
|
|
|
|
from .config_file import read_config_file
|
|
from .message_listener import MessageListener
|
|
from .status_reporter import StatusReporter
|
|
|
|
VERSION = "1.0"
|
|
|
|
DEFAULT_LIBRETIME_CONFIG_FILEPATH = Path("/etc/airtime/airtime.conf")
|
|
DEFAULT_RETRY_QUEUE_FILEPATH = Path("retry_queue")
|
|
|
|
|
|
@click.command()
|
|
@cli_logging_options()
|
|
@cli_config_options()
|
|
@click.option(
|
|
"--retry-queue-filepath",
|
|
envvar=f"{DEFAULT_ENV_PREFIX}_RETRY_QUEUE_FILEPATH",
|
|
type=click.Path(path_type=Path),
|
|
help="Path to the retry queue file.",
|
|
default=DEFAULT_RETRY_QUEUE_FILEPATH,
|
|
)
|
|
def cli(
|
|
log_level: str,
|
|
log_filepath: Optional[Path],
|
|
config_filepath: Optional[Path],
|
|
retry_queue_filepath: Path,
|
|
):
|
|
"""
|
|
Run analyzer.
|
|
"""
|
|
Analyzer(
|
|
log_level=log_level,
|
|
log_filepath=log_filepath,
|
|
# Handle default until the config file can be optional
|
|
config_filepath=config_filepath or DEFAULT_LIBRETIME_CONFIG_FILEPATH,
|
|
retry_queue_filepath=retry_queue_filepath,
|
|
)
|
|
|
|
|
|
class Analyzer(AbstractApp):
|
|
name = "analyzer"
|
|
|
|
def __init__(
|
|
self,
|
|
*,
|
|
config_filepath: Optional[Path],
|
|
retry_queue_filepath: Path,
|
|
**kwargs,
|
|
):
|
|
super().__init__(**kwargs)
|
|
|
|
# Read our rmq config file
|
|
rmq_config = read_config_file(config_filepath)
|
|
|
|
# Start up the StatusReporter process
|
|
StatusReporter.start_thread(retry_queue_filepath)
|
|
|
|
# Start listening for RabbitMQ messages telling us about newly
|
|
# uploaded files. This blocks until we receive a shutdown signal.
|
|
self._msg_listener = MessageListener(rmq_config)
|
|
|
|
StatusReporter.stop_thread()
|