catacomb/pwsafe.py: Commentary fix.

[catacomb-python] / pwsafe

diff --git a/pwsafe b/pwsafe
index 4bdc83ef9ac0ef39f365ca155fdc66370af4cb1f..50a58780f6e1bb565cffa0805adde996a1a07c15 100644 (file)
--- a/pwsafe
+++ b/pwsafe
@@ -53,12 +53,6 @@ def die(msg):
   moan(msg)
   exit(1)
 
-def chomp(pp):
-  """Return the string PP, without its trailing newline if it has one."""
-  if len(pp) > 0 and pp[-1] == '\n':
-    pp = pp[:-1]
-  return pp
-
 ###--------------------------------------------------------------------------
 ### Subcommand implementations.
 
@@ -89,65 +83,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]
-  if len(av) < 2:
-    pp = C.getpass("Enter passphrase `%s': " % tag)
-    vpp = C.getpass("Confirm passphrase `%s': " % tag)
-    if pp != vpp: raise ValueError, "passphrases don't match"
-  elif av[1] == '-':
-    pp = stdin.readline()
-  else:
-    pp = av[1]
-  pw = PW(file, writep = True)
-  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().rstrip('\n')
+    else:
+      pp = av[1]
+    pw[av[0]] = 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]'],