Also remember the module a class is defined in.
diff --git a/Lib/pyclbr.py b/Lib/pyclbr.py
index c19c7a5..f6f5bb9 100644
--- a/Lib/pyclbr.py
+++ b/Lib/pyclbr.py
@@ -52,7 +52,8 @@
# each Python class is represented by an instance of this class
class Class:
'''Class to represent a Python class.'''
- def __init__(self, name, super, file, lineno):
+ def __init__(self, module, name, super, file, lineno):
+ self.module = module
self.name = name
if super is None:
super = []
@@ -187,7 +188,7 @@
names.append(n)
inherit = names
# remember this class
- cur_class = Class(class_name, inherit, file, lineno)
+ cur_class = Class(module, class_name, inherit, file, lineno)
dict[class_name] = cur_class
continue
if is_method.match(line) >= 0: