From fc47cd993d9cf6927b13aa0b5b6d740d7a5e88c7 Mon Sep 17 00:00:00 2001
From: Einar Forselv <eforselv@gmail.com>
Date: Mon, 15 Apr 2019 18:31:08 +0200
Subject: [PATCH] Basic argparser

---
 restic_volume_backup/cli.py | 25 ++++++++++++++++---------
 1 file changed, 16 insertions(+), 9 deletions(-)

diff --git a/restic_volume_backup/cli.py b/restic_volume_backup/cli.py
index 791b38a..0cc2a73 100644
--- a/restic_volume_backup/cli.py
+++ b/restic_volume_backup/cli.py
@@ -1,20 +1,18 @@
+import argparse
+import sys
+
 from restic_volume_backup.config import Config
 from restic_volume_backup.containers import RunningContainers
 
 
 def main():
-    if len(sys.argv) < 2:
-        raise ValueError("Missing argument: {}".format(cmds))
-
-    mode = sys.argv[1]
-    if mode not in cmds:
-        raise ValueError("Valid arguments: {}".format(cmds))
+    args = parse_args()
 
     Config.check()
 
     containers = RunningContainers()
 
-    if mode == 'status':
+    if args.action == 'status':
         containers.print_services()
         # volumes = containers.volume_mounts()
         # for vol in volumes:
@@ -26,7 +24,7 @@ def main():
         #     print(binds)
         #     print(vol.mount_string())
 
-    if mode == 'backup':
+    elif args.mode == 'backup':
         print("Starting backup ..")
         # TODO: Errors when repo already exists
         # restic.init_repo(Config.repository)
@@ -34,9 +32,18 @@ def main():
         # for vol in containers.backup_volumes():
         #     restic.backup_volume(Config.repository, vol)
 
-    if mode == 'snapshots':
+    elif args.mode == 'snapshots':
         restic.snapshots(Config.repository)
 
 
+def parse_args():
+    parser = argparse.ArgumentParser(prog='restic_volume_backup')
+    parser.add_argument(
+        'action',
+        choices=['status', 'backup'],
+    )
+    return parser.parse_args()
+
+
 if __name__ == '__main__':
     main()