Handle KeyboardInterrupt exception

2016-06-09 22:34:38 +01:00 · 2016-06-09 22:34:38 +01:00 · 748005b7ea
parent d3403e7a43
commit 748005b7ea
1 changed files with 5 additions and 1 deletions
--- a/twik/run.py
+++ b/twik/run.py
@ -50,7 +50,11 @@ def main():

    util = Util(args.tag, args.chars, args.passwordtype, args.profile)

-    master_key = getpass.getpass(prompt='Master Key: ')
+    try:
+        master_key = getpass.getpass(prompt='Master Key: ')
+    except KeyboardInterrupt:
+        print "^C"
+        raise SystemExit(0)

    twik = Twik()
    password = twik.getpassword(args.tag, util.get_privatekey(), master_key,