@@ -15,7 +15,7 @@ def emit(self, record):
1515
1616# use a dictionary here so that command strings can be accessed with a lookup.
1717# This helps enforce usage of this structure
18- ALL_COMMANDS = {
18+ ALLOWED_COMMANDS = {
1919 "validate" : "validate" ,
2020 "send" : "send" ,
2121 "validate+send" : "validate+send" ,
@@ -24,7 +24,7 @@ def emit(self, record):
2424 "count" : "count" ,
2525 "fetch" : "fetch" ,
2626}
27- command_list = ', ' .join (f"'{ c } '" for c in ALL_COMMANDS .values ())
27+ command_list = ', ' .join (f"'{ c } '" for c in ALLOWED_COMMANDS .values ())
2828
2929# Set up logging
3030handler = ExitOnExceptionHandler ()
@@ -117,7 +117,7 @@ def main(argv=None):
117117 print (f"unknown arguments { unknown_args_str } passed, use -h flag for help" )
118118 exit (1 )
119119
120- if args .command not in ALL_COMMANDS .values ():
120+ if args .command not in ALLOWED_COMMANDS .values ():
121121 if args .command is None :
122122 logger .error (f"no command provided. Use one of ({ command_list } ), see -h flag for help" )
123123 else :
@@ -161,15 +161,15 @@ def main(argv=None):
161161 )
162162 try :
163163 logger .info ("starting..." )
164- if args .command == ALL_COMMANDS ['count' ]: lb .counter .count ()
165- elif args .command == ALL_COMMANDS ['fetch' ]: lb .fetcher .fetch ()
166- elif args .command == ALL_COMMANDS ['validate' ]: lb .validator .validate ()
167- elif args .command == ALL_COMMANDS ['send' ]: lb .sender .send ()
168- elif args .command == ALL_COMMANDS ['validate+send' ]:
164+ if args .command == ALLOWED_COMMANDS ['count' ]: lb .counter .count ()
165+ elif args .command == ALLOWED_COMMANDS ['fetch' ]: lb .fetcher .fetch ()
166+ elif args .command == ALLOWED_COMMANDS ['validate' ]: lb .validator .validate ()
167+ elif args .command == ALLOWED_COMMANDS ['send' ]: lb .sender .send ()
168+ elif args .command == ALLOWED_COMMANDS ['validate+send' ]:
169169 lb .validator .validate ()
170170 lb .sender .send ()
171- elif args .command == ALL_COMMANDS ['delete' ]: lb .deleter .delete ()
172- elif args .command == ALL_COMMANDS ['truncate' ]: lb .truncator .truncate ()
171+ elif args .command == ALLOWED_COMMANDS ['delete' ]: lb .deleter .delete ()
172+ elif args .command == ALLOWED_COMMANDS ['truncate' ]: lb .truncator .truncate ()
173173 lb .logger .info ("done!" )
174174 except Exception as e :
175175 logger .exception (e , exc_info = lb .config ["show_stacktrace" ])
0 commit comments