catacomb/pwsafe.py: Make `PW' be a context manager, and use it.

diff --git a/catacomb/pwsafe.py b/catacomb/pwsafe.py
index 1e8e4dbfeb41ce1097be5868c2a1b985922a658d..82f2a46f1df6952c202e18974b471cc8130b5408 100644 (file)
--- a/catacomb/pwsafe.py
+++ b/catacomb/pwsafe.py
@@ -299,4 +299,11 @@ class PW (object):
       if k[0] == '$': yield me.unpack(me.db[k])[0]
       k = me.db.nextkey(k)
 
+  ## Context protocol.
+
+  def __enter__(me):
+    return me
+  def __exit__(me, excty, excval, exctb):
+    me.db.close()
+
 ###----- That's all, folks --------------------------------------------------

diff --git a/pwsafe b/pwsafe
index 8a51bd7243bdc51b12ff1ac28b7c3a190721fcf2..5f2e75d1e7ae22f488c6d14f772ec9491c01e9df 100644 (file)
--- a/pwsafe
+++ b/pwsafe
@@ -89,65 +89,63 @@ def cmd_create(av):
 
 def cmd_changepp(av):
   if len(av) != 0: return 1
-  pw = PW(file, writep = True)
-  pw.changepp()
+  with PW(file, writep = True) as pw: pw.changepp()
 
 def cmd_find(av):
   if len(av) != 1: return 1
-  pw = PW(file)
-  try: print pw[av[0]]
-  except KeyError, exc: die("Password `%s' not found." % exc.args[0])
+  with PW(file) as pw:
+    try: print pw[av[0]]
+    except KeyError, exc: die("Password `%s' not found" % exc.args[0])
 
 def cmd_store(av):
-  if len(av) < 1 or len(av) > 2:
-    return 1
+  if len(av) < 1 or len(av) > 2: return 1
   tag = av[0]
-  pw = PW(file, writep = True)
-  if len(av) < 2:
-    pp = C.getpass("Enter passphrase `%s': " % tag)
-    vpp = C.getpass("Confirm passphrase `%s': " % tag)
-    if pp != vpp: die("passphrases don't match")
-  elif av[1] == '-':
-    pp = stdin.readline()
-  else:
-    pp = av[1]
-  pw[av[0]] = chomp(pp)
+  with PW(file, writep = True) as pw:
+    if len(av) < 2:
+      pp = C.getpass("Enter passphrase `%s': " % tag)
+      vpp = C.getpass("Confirm passphrase `%s': " % tag)
+      if pp != vpp: die("passphrases don't match")
+    elif av[1] == '-':
+      pp = stdin.readline()
+    else:
+      pp = av[1]
+    pw[av[0]] = chomp(pp)
 
 def cmd_copy(av):
   if len(av) < 1 or len(av) > 2: return 1
-  pw_in = PW(file)
-  pw_out = PW(av[0], 'w')
-  if len(av) >= 3: pat = av[1]
-  else: pat = None
-  for k in pw_in:
-    if pat is None or fnmatch(k, pat): pw_out[k] = pw_in[k]
+  with PW(file) as pw_in:
+    with PW(av[0], writep = True) as pw_out:
+      if len(av) >= 3: pat = av[1]
+      else: pat = None
+      for k in pw_in:
+        if pat is None or fnmatch(k, pat): pw_out[k] = pw_in[k]
 
 def cmd_list(av):
   if len(av) > 1: return 1
-  pw = PW(file)
-  if len(av) >= 1: pat = av[0]
-  else: pat = None
-  for k in pw:
-    if pat is None or fnmatch(k, pat): print k
+  with PW(file) as pw:
+    if len(av) >= 1: pat = av[0]
+    else: pat = None
+    for k in pw:
+      if pat is None or fnmatch(k, pat): print k
 
 def cmd_topixie(av):
   if len(av) > 2: return 1
-  pw = PW(file)
-  pix = C.Pixie()
-  if len(av) == 0:
-    for tag in pw: pix.set(tag, pw[tag])
-  else:
-    tag = av[0]
-    if len(av) >= 2: pptag = av[1]
-    else: pptag = av[0]
-    pix.set(pptag, pw[tag])
+  with PW(file) as pw:
+    pix = C.Pixie()
+    if len(av) == 0:
+      for tag in pw: pix.set(tag, pw[tag])
+    else:
+      tag = av[0]
+      if len(av) >= 2: pptag = av[1]
+      else: pptag = av[0]
+      pix.set(pptag, pw[tag])
 
 def cmd_del(av):
   if len(av) != 1: return 1
-  pw = PW(file, writep = True)
-  tag = av[0]
-  try: del pw[tag]
-  except KeyError, exc: die("Password `%s' not found." % exc.args[0])
+  with PW(file, writep = True) as pw:
+    tag = av[0]
+    try: del pw[tag]
+    except KeyError, exc: die("Password `%s' not found" % exc.args[0])
 
 commands = { 'create': [cmd_create,
                         '[-c CIPHER] [-h HASH] [-m MAC] [PP-TAG]'],