From 85e74f097ebf2ed79d6467203f15327fc540424a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Cyrill=20K=C3=BCttel?= <cyrill.kuettel@seantis.ch>
Date: Tue, 14 May 2024 23:00:39 +0200
Subject: [PATCH] use click in upgrade

---
 src/privatim/cli/upgrade.py | 20 +++-----------------
 1 file changed, 3 insertions(+), 17 deletions(-)

diff --git a/src/privatim/cli/upgrade.py b/src/privatim/cli/upgrade.py
index fc189db..2a43765 100644
--- a/src/privatim/cli/upgrade.py
+++ b/src/privatim/cli/upgrade.py
@@ -124,7 +124,9 @@ def commit(self) -> None:
         transaction.commit()
 
 
-
+@click.command()
+@click.argument('config_uri')
+@click.option('--dry', is_flag=True, default=False)
 def upgrade(args: argparse.Namespace) -> None:
 
     # Extract settings from INI config file.
@@ -154,19 +156,3 @@ def upgrade(args: argparse.Namespace) -> None:
 
     if not args.dry:
         context.commit()
-
-
-def main() -> None:
-    parser = argparse.ArgumentParser(description='Runs all upgrade steps')
-    parser.add_argument('config_uri', help='Config file')
-    parser.add_argument('-d', '--dry', help='Dry run', action='store_true')
-    args = parser.parse_args()
-
-    if args.dry:
-        print('Dry run')
-
-    upgrade(args)
-
-
-if __name__ == '__main__':
-    main()