Fix warning in mustclose decorator

2013-02-18 17:56:59 -05:00 · 2013-02-18 17:56:59 -05:00 · 52e674a192
commit 52e674a192
parent e241c13bf1
1 changed files with 2 additions and 1 deletions
--- a/nilmdb/utils/mustclose.py
+++ b/nilmdb/utils/mustclose.py
@ -38,7 +38,8 @@ def must_close(errorfile = sys.stderr, wrap_verify = False):

        @wrap_class_method
        def close(orig, self, *args, **kwargs):
-            del self._must_close
+            if "_must_close" in self.__dict__:
+                del self._must_close
            return orig(self, *args, **kwargs)

        # Optionally wrap all other functions