diff options
| author | emile <emile@crater.logilab.fr> | 2008-07-23 18:35:39 +0200 |
|---|---|---|
| committer | emile <emile@crater.logilab.fr> | 2008-07-23 18:35:39 +0200 |
| commit | 39fcdbf07baf76bed1cb359d37ae63525aa0c69e (patch) | |
| tree | e2d7d9510dafb171ea10c1aeabcd5a410341edf9 /pyreverse | |
| parent | 6f3d93b2feea0c1f9f2653f95cb7dd27d5ebc2b7 (diff) | |
| download | pylint-39fcdbf07baf76bed1cb359d37ae63525aa0c69e.tar.gz | |
add try ... / except KeyError for correct unittest_vcg.py
Diffstat (limited to 'pyreverse')
| -rw-r--r-- | pyreverse/diadefslib.py | 20 |
1 files changed, 13 insertions, 7 deletions
diff --git a/pyreverse/diadefslib.py b/pyreverse/diadefslib.py index 9948258..a85d656 100644 --- a/pyreverse/diadefslib.py +++ b/pyreverse/diadefslib.py @@ -168,10 +168,13 @@ class DefaultDiadefGenerator(LocalsVisitor): """ # cleanup locals inserted by the astng builder to mimick python # interpretor behaviour - del node.locals['__name__'] - del node.locals['__file__'] - del node.locals['__dict__'] - del node.locals['__doc__'] + try: + del node.locals['__name__'] + del node.locals['__file__'] + del node.locals['__dict__'] + del node.locals['__doc__'] + except KeyError: + pass if self.pkgdiagram: self.linker.visit(node) self.pkgdiagram.add_object(node=node, title=node.name) @@ -183,9 +186,12 @@ class DefaultDiadefGenerator(LocalsVisitor): """ # cleanup locals inserted by the astng builder to mimick python # interpretor behaviour - del node.locals['__name__'] - del node.locals['__dict__'] - del node.locals['__doc__'] + try: + del node.locals['__name__'] + del node.locals['__dict__'] + del node.locals['__doc__'] + except KeyError: + pass self.linker.visit(node) self.classdiagram.add_object(node=node, title=node.name) |