From cad4e3c917017d43c382df213a1c045523e247b8 Mon Sep 17 00:00:00 2001 From: Gael Pasgrimaud Date: Wed, 12 Jan 2011 19:07:14 +0100 Subject: add Configurator.extend feature --- pyramid/config.py | 27 ++++++++++++++++++++ pyramid/tests/__init__.py | 4 ++- pyramid/tests/test_config.py | 59 +++++++++++++++++++++++++++++++++++++++++++- 3 files changed, 88 insertions(+), 2 deletions(-) diff --git a/pyramid/config.py b/pyramid/config.py index 9604833f3..60197a4c6 100644 --- a/pyramid/config.py +++ b/pyramid/config.py @@ -558,6 +558,33 @@ class Configurator(object): config = self.__class__.with_context(context) c(config) + def extend(self, *callables): + _context = self._ctx + if _context is None: + _context = self._ctx = self._make_context(self.autocommit) + + klass = self.__class__ + + for c in callables: + c = self.maybe_dotted(c) + name = c.__name__ + sourcefile = inspect.getsourcefile(c) + module = inspect.getmodule(c) + spec = module.__name__ + ':' + name + if _context.processSpec(spec): + if hasattr(klass, name): + raise ConfigurationError( + "Configurator already have a method named %s" % name) + context = GroupingContextDecorator(_context) + context.basepath = os.path.dirname(sourcefile) + context.includepath = _context.includepath + (spec,) + context.package = package_of(module) + config = klass.with_context(context) + def extend_wrapper(*args, **kwargs): + c(config, *args, **kwargs) + extend_wrapper.__name__ = name + setattr(klass, name, staticmethod(extend_wrapper)) + @classmethod def with_context(cls, context): """A classmethod used by ZCML directives, diff --git a/pyramid/tests/__init__.py b/pyramid/tests/__init__.py index 5bb534f79..a62c29f47 100644 --- a/pyramid/tests/__init__.py +++ b/pyramid/tests/__init__.py @@ -1 +1,3 @@ -# package + +def dummy_extend(*args): + """used to test Configurator.extend""" diff --git a/pyramid/tests/test_config.py b/pyramid/tests/test_config.py index ac459d7e3..a16323313 100644 --- a/pyramid/tests/test_config.py +++ b/pyramid/tests/test_config.py @@ -3216,6 +3216,60 @@ class ConfiguratorTests(unittest.TestCase): for confinst in conflict: yield confinst[2] +class TestConfiguratorExtender(unittest.TestCase): + + def setUp(self): + from pyramid.config import Configurator + class Config(Configurator): pass + self.config = Config() + + def test_extend_with_dotted_name(self): + from pyramid import tests + config = self.config + context_before = config._make_context() + config._ctx = context_before + config.extend('pyramid.tests.test_config.dummy_extend') + self.assert_(hasattr(config, 'dummy_extend')) + config.dummy_extend('discrim') + context_after = config._ctx + actions = context_after.actions + self.assertEqual(len(actions), 1) + self.assertEqual( + context_after.actions[0][:3], + ('discrim', None, tests), + ) + self.assertEqual(context_after.basepath, None) + self.assertEqual(context_after.includepath, ()) + self.failUnless(context_after is context_before) + + def test_extend_with_python_callable(self): + from pyramid import tests + config = self.config + context_before = config._make_context() + config._ctx = context_before + config.extend(dummy_extend) + self.assert_(hasattr(config, 'dummy_extend')) + config.dummy_extend('discrim') + context_after = config._ctx + actions = context_after.actions + self.assertEqual(len(actions), 1) + self.assertEqual( + context_after.actions[0][:3], + ('discrim', None, tests), + ) + self.assertEqual(context_after.basepath, None) + self.assertEqual(context_after.includepath, ()) + self.failUnless(context_after is context_before) + + def test_extend_conflict(self): + from pyramid import tests + from pyramid.exceptions import ConfigurationError + config = self.config + context_before = config._make_context() + config._ctx = context_before + config.extend(dummy_extend) + self.assertRaises(ConfigurationError, config.extend, 'pyramid.tests.dummy_extend') + class TestViewDeriver(unittest.TestCase): def setUp(self): self.config = testing.setUp() @@ -4943,4 +4997,7 @@ class DummyHandler(object): # pragma: no cover def dummy_include(config): config.action('discrim', None, config.package) - + +def dummy_extend(config, discrim): + config.action(discrim, None, config.package) + -- cgit v1.2.3 From 2422ceae0750c65cc1e4ea097143f4c71fbba348 Mon Sep 17 00:00:00 2001 From: Gael Pasgrimaud Date: Wed, 12 Jan 2011 21:38:06 +0100 Subject: store extends in context and add attributes to instance --- pyramid/config.py | 19 +++++++++++++------ pyramid/tests/test_config.py | 16 +++++++++++++--- 2 files changed, 26 insertions(+), 9 deletions(-) diff --git a/pyramid/config.py b/pyramid/config.py index 60197a4c6..8e709b607 100644 --- a/pyramid/config.py +++ b/pyramid/config.py @@ -275,6 +275,7 @@ class Configurator(object): default_permission=None, session_factory=None, default_view_mapper=None, + extends = None, autocommit=False, ): if package is None: @@ -302,6 +303,8 @@ class Configurator(object): session_factory=session_factory, default_view_mapper=default_view_mapper, ) + if extends: + self.extend(*extends) def _set_settings(self, mapping): settings = Settings(mapping or {}) @@ -425,6 +428,7 @@ class Configurator(object): context = PyramidConfigurationMachine() registerCommonDirectives(context) context.registry = self.registry + context.extends = [] context.autocommit = autocommit return context @@ -572,7 +576,7 @@ class Configurator(object): module = inspect.getmodule(c) spec = module.__name__ + ':' + name if _context.processSpec(spec): - if hasattr(klass, name): + if hasattr(self, name): raise ConfigurationError( "Configurator already have a method named %s" % name) context = GroupingContextDecorator(_context) @@ -580,10 +584,13 @@ class Configurator(object): context.includepath = _context.includepath + (spec,) context.package = package_of(module) config = klass.with_context(context) - def extend_wrapper(*args, **kwargs): - c(config, *args, **kwargs) - extend_wrapper.__name__ = name - setattr(klass, name, staticmethod(extend_wrapper)) + wrapped = action_method(c) + def wrapper(*args, **kwargs): + return wrapped(config, *args, **kwargs) + wrapper.__name__ = name + wrapper.__doc__ = c.__doc__ + self.__dict__[name] = wrapper + context.extends.append(c) @classmethod def with_context(cls, context): @@ -592,7 +599,7 @@ class Configurator(object): :meth:`pyramid.config.Configurator.include` to obtain a configurator with 'the right' context. Returns a new Configurator instance.""" configurator = cls(registry=context.registry, package=context.package, - autocommit=context.autocommit) + extends=context.extends, autocommit=context.autocommit) configurator._ctx = context return configurator diff --git a/pyramid/tests/test_config.py b/pyramid/tests/test_config.py index a16323313..6b5bfbf94 100644 --- a/pyramid/tests/test_config.py +++ b/pyramid/tests/test_config.py @@ -3220,8 +3220,7 @@ class TestConfiguratorExtender(unittest.TestCase): def setUp(self): from pyramid.config import Configurator - class Config(Configurator): pass - self.config = Config() + self.config = Configurator() def test_extend_with_dotted_name(self): from pyramid import tests @@ -3241,6 +3240,8 @@ class TestConfiguratorExtender(unittest.TestCase): self.assertEqual(context_after.basepath, None) self.assertEqual(context_after.includepath, ()) self.failUnless(context_after is context_before) + self.assertEqual(len(context_before.extends), 1) + self.assertEqual(context_before.extends, context_after.extends) def test_extend_with_python_callable(self): from pyramid import tests @@ -3260,9 +3261,10 @@ class TestConfiguratorExtender(unittest.TestCase): self.assertEqual(context_after.basepath, None) self.assertEqual(context_after.includepath, ()) self.failUnless(context_after is context_before) + self.assertEqual(len(context_before.extends), 1) + self.assertEqual(context_before.extends, context_after.extends) def test_extend_conflict(self): - from pyramid import tests from pyramid.exceptions import ConfigurationError config = self.config context_before = config._make_context() @@ -3270,6 +3272,14 @@ class TestConfiguratorExtender(unittest.TestCase): config.extend(dummy_extend) self.assertRaises(ConfigurationError, config.extend, 'pyramid.tests.dummy_extend') + def test_extend_no_conflict_with_two_instance(self): + from pyramid.config import Configurator + config = self.config + config.extend(dummy_extend) + config2 = Configurator() + config2.extend('pyramid.tests.dummy_extend') + self.failUnless(config._ctx.extends != config2._ctx.extends) + class TestViewDeriver(unittest.TestCase): def setUp(self): self.config = testing.setUp() -- cgit v1.2.3 From 035d033169f48d8250afafedd0a078d3e4f62922 Mon Sep 17 00:00:00 2001 From: Gael Pasgrimaud Date: Wed, 12 Jan 2011 22:01:16 +0100 Subject: improve code lines --- pyramid/config.py | 4 +--- pyramid/tests/test_config.py | 15 +++++++++++++-- 2 files changed, 14 insertions(+), 5 deletions(-) diff --git a/pyramid/config.py b/pyramid/config.py index 8e709b607..468618c92 100644 --- a/pyramid/config.py +++ b/pyramid/config.py @@ -567,8 +567,6 @@ class Configurator(object): if _context is None: _context = self._ctx = self._make_context(self.autocommit) - klass = self.__class__ - for c in callables: c = self.maybe_dotted(c) name = c.__name__ @@ -583,7 +581,7 @@ class Configurator(object): context.basepath = os.path.dirname(sourcefile) context.includepath = _context.includepath + (spec,) context.package = package_of(module) - config = klass.with_context(context) + config = self.__class__.with_context(context) wrapped = action_method(c) def wrapper(*args, **kwargs): return wrapped(config, *args, **kwargs) diff --git a/pyramid/tests/test_config.py b/pyramid/tests/test_config.py index 6b5bfbf94..3fa9b954c 100644 --- a/pyramid/tests/test_config.py +++ b/pyramid/tests/test_config.py @@ -3277,8 +3277,19 @@ class TestConfiguratorExtender(unittest.TestCase): config = self.config config.extend(dummy_extend) config2 = Configurator() - config2.extend('pyramid.tests.dummy_extend') - self.failUnless(config._ctx.extends != config2._ctx.extends) + config2.extend(dummy_extend) + self.assertEqual(config._ctx.extends, config2._ctx.extends) + + def test_extend_after_with_package(self): + from pyramid import tests + config = self.config + context_before = config._make_context() + config._ctx = context_before + config.extend(dummy_extend) + config2 = config.with_package('pyramid.tests') + self.assert_(hasattr(config2, 'dummy_extend')) + self.assertEqual(len(config._ctx.extends), 1) + self.assertEqual(config._ctx.extends, config2._ctx.extends) class TestViewDeriver(unittest.TestCase): def setUp(self): -- cgit v1.2.3