check for mapper or class, raise exception otherwise [ticket:305]

This commit is contained in:
Mike Bayer
2006-09-20 00:26:38 +00:00
parent 0044c26368
commit 3e94bbdba8
+4 -2
View File
@@ -205,9 +205,11 @@ class PropertyLoader(mapper.MapperProperty):
def do_init(self):
if isinstance(self.argument, type):
self.mapper = mapper.class_mapper(self.argument, compile=False)._check_compile()
else:
elif isinstance(self.argument, mapper.Mapper):
self.mapper = self.argument._check_compile()
else:
raise exceptions.ArgumentError("relation '%s' expects a class or a mapper argument (received: %s)" % (self.key, type(self.argument)))
self.mapper = self.mapper.get_select_mapper()._check_compile()
if self.association is not None: