X-Git-Url: https://www.chiark.greenend.org.uk/ucgi/~mdw/git/catacomb-python/blobdiff_plain/f39dc0c877f1e53559b3767a1e3d4bc08c38ab3d..5bf6e9f5b6a247c906d51d86317755b99c3d731a:/catacomb/pwsafe.py

diff --git a/catacomb/pwsafe.py b/catacomb/pwsafe.py
index 1e8e4db..82f2a46 100644
--- 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 --------------------------------------------------