Skip to content

Commit 02caae2

Browse files
committed
change name to ALLOWED_COMMANDS
1 parent 38fd876 commit 02caae2

1 file changed

Lines changed: 10 additions & 10 deletions

File tree

lightbeam/__main__.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -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
3030
handler = 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

Comments
 (0)