From 7294d85c090786430be1a78ddb9147598e6d42dd Mon Sep 17 00:00:00 2001
From: einarf <eforselv@gmail.com>
Date: Wed, 8 Mar 2023 23:59:27 +0100
Subject: [PATCH] Dynamically load commands
---
src/restic_compose_backup/cli.py | 40 ++++++++++++++++++++------------
1 file changed, 25 insertions(+), 15 deletions(-)
diff --git a/src/restic_compose_backup/cli.py b/src/restic_compose_backup/cli.py
index fb8cbce..5119836 100644
--- a/src/restic_compose_backup/cli.py
+++ b/src/restic_compose_backup/cli.py
@@ -1,6 +1,9 @@
import argparse
import os
import logging
+from importlib import import_module
+from pkgutil import iter_modules
+from typing import Dict, List
from restic_compose_backup import (
alerts,
@@ -11,13 +14,29 @@ from restic_compose_backup import (
from restic_compose_backup.config import Config
from restic_compose_backup.containers import RunningContainers
from restic_compose_backup import cron, utils
-
+from restic_compose_backup import commands
+from restic_compose_backup.commands.base import BaseCommand
logger = logging.getLogger(__name__)
+def get_commands() -> Dict[str, BaseCommand]:
+ """Return the list of available command classes"""
+ _commands = {}
+ for module_info in iter_modules(commands.__path__):
+ module = import_module(f'restic_compose_backup.commands.{module_info.name}')
+ if hasattr(module, 'Command'):
+ _commands[module_info.name] = module.Command
+ return _commands
+
+
def main():
"""CLI entrypoint"""
- args = parse_args()
+ commands = get_commands()
+ args = parse_args(sorted(commands.keys()))
+ command = commands[args.action]()
+ command.run()
+ return
+
config = Config()
log.setup(level=args.log_level or config.log_level)
containers = RunningContainers()
@@ -101,7 +120,8 @@ def status(config, containers):
if container.database_backup_enabled:
instance = container.instance
- ping = instance.ping()
+ # ping = instance.ping()
+ ping = 0
logger.info(
' - %s (is_ready=%s) -> %s',
instance.container_type,
@@ -290,21 +310,11 @@ def crontab(config):
print(cron.generate_crontab(config))
-def parse_args():
+def parse_args(choices: List[str]):
parser = argparse.ArgumentParser(prog='restic_compose_backup')
parser.add_argument(
'action',
- choices=[
- 'status',
- 'snapshots',
- 'backup',
- 'start-backup-process',
- 'alert',
- 'cleanup',
- 'version',
- 'crontab',
- 'test',
- ],
+ choices=choices,
)
parser.add_argument(
'--log-level',