From e6c0ce38e5a6928ab5c082b08e4e0ec3657dc97d Mon Sep 17 00:00:00 2001 From: Felix Yan Date: Tue, 17 Nov 2020 06:34:53 +0800 Subject: [PATCH] Add a command line option to run nvchecker only on a specific entry This is useful for debugging purpose, when your nvchecker.toml is very large. --- nvchecker/__main__.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/nvchecker/__main__.py b/nvchecker/__main__.py index b05b722..a98a85a 100755 --- a/nvchecker/__main__.py +++ b/nvchecker/__main__.py @@ -25,6 +25,8 @@ def main() -> None: help='use specified keyfile (override the one in configuration file)') parser.add_argument('-t', '--tries', default=1, type=int, metavar='N', help='try N times when network errors occur') + parser.add_argument('-e', '--entry', type=str, + help='only execute on specified entry (useful for debugging)') core.add_common_arguments(parser) args = parser.parse_args() if core.process_common_arguments(args): @@ -34,6 +36,11 @@ def main() -> None: entries, options = core.load_file( args.file, use_keymanager=not bool(args.keyfile)) + if args.entry: + if args.entry not in entries: + sys.exit('Specified entry not found in config') + entries = {args.entry: entries[args.entry]} + if args.keyfile: keymanager = KeyManager(Path(args.keyfile)) else: