Commit 8d30aec

committed

improving the exception handling and couple other things

1 parent 0ac49b7 commit 8d30aecCopy full SHA for 8d30aec

File tree

-8

lines changed

-8

lines changed

Lines changed: 9 additions & 8 deletions

Original file line number	Diff line number	Diff line change
`@@ -3,7 +3,7 @@`
`3`	`3`	`import sys`
`4`	`4`	`sys.path.insert(0, '/opt/python/.vendor')`
`5`	`5`
`6`		`-from ssm_cache import SSMParameter`
	`6`	`+from ssm_cache import SSMParameter, InvalidParameterError`
`7`	`7`	`import time`
`8`	`8`	`import random`
`9`	`9`	`import json`
`@@ -22,12 +22,15 @@ def get_config():`
`22`	`22`	`return -1`
`23`	`23`	`else:`
`24`	`24`	`return 0`
`25`		`- except Exception as e:`
`26`		`- print(e)`
	`25`	`+ except InvalidParameterError as e:`
	`26`	`+ print("{} does not exist in SSM".format(e))`
	`27`	`+ return 0`
	`28`	`+ except KeyError as e:`
	`29`	`+ print("{} is not a valid Key in the SSM configuration".format(e))`
`27`	`30`	`return 0`
`28`	`31`
`29`	`32`
`30`		`-def delayit(method):`
	`33`	`+def delayit(func):`
`31`	`34`	`def latency(args, *kw):`
`32`	`35`	`delay = get_config()`
`33`	`36`	`start = time.time()`
`@@ -40,14 +43,12 @@ def latency(args, *kw):`
`40`	`43`	`if random.random() > 0.5:`
`41`	`44`	`# random sleep time between 1 and 10 seconds`
`42`	`45`	`time.sleep(random.randint(1, 10))`
`43`		`- else:`
`44`		`- pass`
`45`	`46`
`46`		`- result = method(args, *kw)`
	`47`	`+ result = func(args, *kw)`
`47`	`48`	`end = time.time()`
`48`	`49`
`49`	`50`	`print('Added {1:.2f}ms to {0:s}'.format(`
`50`		`- method.__name__,`
	`51`	`+ func.__name__,`
`51`	`52`	`(end - start) * 1000`
`52`	`53`	`))`
`53`	`54`

Comments

(0)