Merge pull request #6 from someengineering/lloesche/pgdump

Add sleep option

fixbackup/__main__.py

@@ -1,5 +1,6 @@
 import sys
 import os
+import time
 from pathlib import Path
 from typing import List
 from .logger import add_args as logging_add_args, log
@@ -12,11 +13,24 @@
 def main() -> None:
     args = parse_args([logging_add_args, s3_add_args, *backup_add_args])
     exit_code = 0
-    log.info("Starting FIX Databases Backup System")
+    log.info("Starting Fix Databases Backup System")
 
     if not verify_binaries():
         sys.exit(1)
 
+    if args.sleep:
+        # This option is used to keep the container running for debugging purposes.
+        # It allows you to connect to it inside of e.g. a K8s environment
+        # and manually test the backup process. Alternatively, you could
+        # override the entrypoint of the container and sleep indefinitely.
+        log.info("Sleeping forever")
+        try:
+            while True:
+                time.sleep(300)
+        finally:
+            log.info("Shutdown complete")
+            sys.exit(0)
+
     backup_directory = Path(args.backup_directory)
     rmdir_backup_directory = True
     if backup_directory.exists() and not backup_directory.is_dir():

fixbackup/args.py

@@ -19,6 +19,13 @@ def parse_args(add_args: List[Callable[[ArgumentParser], None]]) -> Namespace:
         help="Name of the environment",
         default=os.getenv("FIX_ENVIRONMENT", "dev"),
     )
+    arg_parser.add_argument(
+        "--sleep",
+        help="Don't do anything, just sleep forever",
+        dest="sleep",
+        action="store_true",
+        default=False,
+    )
 
     for add_arg in add_args:
         add_arg(arg_parser)

fixbackup/backup/__init__.py

@@ -8,7 +8,7 @@
 from .arangodb import backup as arangodb_backup, add_args as arangodb_add_args
 from ..utils import valid_hostname, valid_ip, valid_dbname
 
-add_args = [redis_add_args, mysql_add_args, arangodb_add_args, postgresql_add_args]
+add_args = [redis_add_args, mysql_add_args, postgresql_add_args, arangodb_add_args]
 
 
 def backup(args: Namespace, backup_directory: Path) -> Tuple[List[Path], bool]: