Fix registration bug

This commit is contained in:
Matthew Tretter 2013-01-31 04:07:57 -05:00
parent 04aa72c1f9
commit 01fad6e4c6

View file

@ -44,7 +44,9 @@ class GeneratorRegistry(object):
def before_access_receiver(self, sender, file, **kwargs):
generator = file.generator
if generator in self._generators.values():
# FIXME: I guess this means you can't register functions?
if generator.__class__ in self._generators.values():
# Only invoke the strategy method for registered generators.
call_strategy_method(generator, 'before_access', file=file)