saltstack/salt
1
0
Fork 0
mirror of https://github.com/saltstack/salt.git synced 2025-04-15 17:20:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Nicer handling of user keyboard interruption

Browse source
This commit is contained in:
Pedro Algarvio 2015-07-12 12:37:56 +01:00
parent 0add87e7b9
commit 6294c67b12
1 changed files with 13 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
tests/modparser.py
View file

@ -78,6 +78,7 @@ def mod_data(opts, full):
def scan(opts):
'''
Scan the provided root for python source files
'''
ret = {}
for root, dirs, files in os.walk(opts['root']):
@ -92,10 +93,15 @@ if __name__ == '__main__':
if not HAS_ARGPARSE:
print('The argparse python module is required')
opts = parse()
scand = scan(opts)
if opts['format'] == 'yaml':
print(yaml.dump(scand))
if opts['format'] == 'json':
print(json.dumps(scand))
else:
pprint.pprint(scand)
try:
scand = scan(opts)
if opts['format'] == 'yaml':
print(yaml.dump(scand))
if opts['format'] == 'json':
print(json.dumps(scand))
else:
pprint.pprint(scand)
exit(0)
except KeyboardInterrupt:
print('\nInterrupted on user request', file=sys.stderr)
exit(1)