Also remember the module a class is defined in.

1 files changed, 3 insertions, 2 deletions

diff --git a/Lib/pyclbr.py b/Lib/pyclbr.py
index c19c7a577e..f6f5bb932e 100644
--- a/Lib/pyclbr.py
+++ b/Lib/pyclbr.py
@@ -52,7 +52,8 @@ _modules = {}				# cache of modules we've seen
 # 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 @@ def readmodule(module, path = []):
 					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: