Fix issues with generators.

This commit is contained in:
Dave Halter
2015-01-05 19:11:09 +01:00
parent 1d2704fb68
commit 9cd8fabf2c
3 changed files with 32 additions and 2 deletions
+12 -1
View File
@@ -185,13 +185,24 @@ class FakeImport(pr.Import):
class FakeName(pr.Name):
def __init__(self, name_str, parent=None, start_pos=(0, 0)):
def __init__(self, name_str, parent=None, start_pos=(0, 0), is_definition=None):
"""
In case is_definition is defined (not None), that bool value will be
returned.
"""
super(FakeName, self).__init__(name_str, start_pos)
self.parent = parent
self._is_definition = is_definition
def get_definition(self):
return self.parent
def is_definition(self):
if self._is_definition is None:
return super(FakeName, self).is_definition()
else:
return self._is_definition
class LazyName(FakeName):
def __init__(self, name, parent_callback):