From b40e9b6eed80416a9e19fc8e1873beb2a9f4a09b Mon Sep 17 00:00:00 2001 From: Iain Learmonth Date: Fri, 22 Apr 2022 14:30:53 +0100 Subject: [PATCH] cli: import/export more tables --- app/cli/db.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/app/cli/db.py b/app/cli/db.py index 3d03437..d118d83 100644 --- a/app/cli/db.py +++ b/app/cli/db.py @@ -62,9 +62,9 @@ class DbCliHandler: @classmethod def add_subparser_to(cls, subparsers: argparse._SubParsersAction) -> None: parser = subparsers.add_parser("db", help="database operations") - parser.add_argument("--export", choices=["group", "origin", "proxy", "bridge"], + parser.add_argument("--export", choices=sorted(models.keys()), help="export data to CSV format") - parser.add_argument("--import", choices=["group", "origin", "proxy", "bridge"], + parser.add_argument("--import", choices=sorted(models.keys()), help="import data from CSV format", dest="impot") parser.set_defaults(cls=cls)