X-Git-Url: https://www.chiark.greenend.org.uk/ucgi/~mdw/git/catacomb-python/blobdiff_plain/43c0985125678d39a384362c7f0349bab812aa94..2e6a3fdaabb158cddb8fd1ac14737d6228274b37:/catacomb/pwsafe.py diff --git a/catacomb/pwsafe.py b/catacomb/pwsafe.py index 5503491..823b47f 100644 --- a/catacomb/pwsafe.py +++ b/catacomb/pwsafe.py @@ -114,11 +114,17 @@ class PW (object): value = buf.getstring() return key, value def __getitem__(me, key): - return me.unpack(me.db[me.keyxform(key)])[1] + try: + return me.unpack(me.db[me.keyxform(key)])[1] + except KeyError: + raise KeyError, key def __setitem__(me, key, value): me.db[me.keyxform(key)] = me.pack(key, value) def __delitem__(me, key): - del me.db[me.keyxform(key)] + try: + del me.db[me.keyxform(key)] + except KeyError: + raise KeyError, key def __iter__(me): return PWIter(me)