diff options
| -rwxr-xr-x | bin/envstore | 16 |
1 files changed, 10 insertions, 6 deletions
diff --git a/bin/envstore b/bin/envstore index e314a12..87f797a 100755 --- a/bin/envstore +++ b/bin/envstore @@ -75,13 +75,17 @@ elif command == "eval": for k, v in store.iteritems(): print "export " + k + "=" + v elif command == "save": - key = get_key() + arg = get_key() - try: - value = os.environ[key] - except KeyError: - print >> sys.stderr, key + " does not exist in environment" - sys.exit(3) + if "=" in arg: + key, value = arg.split('=', 2) + + else: + try: + value = os.environ[key] + except KeyError: + print >> sys.stderr, key + " does not exist in environment" + sys.exit(3) store[key] = value save_store() |