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)