def get_site_creator(registry: Registry) -> ISiteCreator:
"""Get the class implementing ISiteCreator.
:param registry: Pyramid registry.
:return: Class implementing ISiteCreator.
"""
site_creator = registry.queryUtility(ISiteCreator)
return site_creator
def get_group_class(registry: Registry) -> t.Type[IGroupModel]:
"""Get the class implementing IGroupModel.
:param registry: Pyramid registry.
:return: Class implementing IGroupModel.
"""
group_class = registry.queryUtility(IGroupModel)
return group_class
def get_activation_model(registry: Registry) -> t.Type[IActivationModel]:
"""Get the class implementing IActivationModel.
:param registry: Pyramid registry.
:return: Class implementing IActivationModel.
"""
activation_model = registry.queryUtility(IActivationModel)
return activation_model
def get_user_class(registry: Registry) -> t.Type[IUserModel]:
"""Get the class implementing IUserModel.
:param registry: Pyramid registry.
:return: Class implementing IUserModel.
"""
user_class = registry.queryUtility(IUserModel)
return user_class
def get_first_login_manager(registry: Registry) -> IFirstLoginManager:
"""Get the class implementing IFirstLoginManager.
:param registry: Pyramid registry.
:return: Class implementing IFirstLoginManager.
"""
first_login_manager = registry.queryUtility(IFirstLoginManager)
return first_login_manager
def get_social_login_mapper(registry: Registry, provider_id:str) -> ISocialLoginMapper:
"""Get a named social login mapper.
Example::
get_social_login_mapper(registry, "facebook")
"""
return registry.queryUtility(ISocialLoginMapper, name=provider_id)
def test_setup_registry_no_default_root_factory(self):
from pyramid.registry import Registry
from pyramid.interfaces import IRootFactory
reg = Registry()
config = self._makeOne(reg)
config.setup_registry()
config.commit()
self.assertEqual(reg.queryUtility(IRootFactory), None)
def get_social_login_mapper(registry: Registry, provider_id: str) -> ISocialLoginMapper:
"""Get a named social login mapper.
Example::
get_social_login_mapper(registry, "facebook")
"""
return registry.queryUtility(ISocialLoginMapper, name=provider_id)
def get_authomatic(registry: Registry) -> Authomatic:
"""Get active Authomatic instance from the registry.
This is registed in ``Initializer.configure_authomatic()``.
:param registry:
:return:
"""
instance = registry.queryUtility(IAuthomatic)
return instance
def get_authomatic(registry: Registry) -> Authomatic:
"""Get active Authomatic instance from the registry.
This is registered in ``Initializer.configure_authomatic()``.
:param registry: Pyramid registry.
:return: Instance of Authomatic.
"""
instance = registry.queryUtility(IAuthomatic)
return instance
def test_setup_registry_no_default_root_factory(self):
from pyramid.registry import Registry
from pyramid.interfaces import IRootFactory
reg = Registry()
config = self._makeOne(reg)
config.setup_registry()
config.commit()
self.assertEqual(reg.queryUtility(IRootFactory), None)
def test_setup_registry_response_factory(self):
from pyramid.registry import Registry
from pyramid.interfaces import IResponseFactory
reg = Registry()
config = self._makeOne(reg)
factory = lambda r: object()
config.setup_registry(response_factory=factory)
self.assertEqual(reg.queryUtility(IResponseFactory), None)
config.commit()
utility = reg.getUtility(IResponseFactory)
self.assertEqual(utility, factory)
def test_setup_registry_response_factory(self):
from pyramid.registry import Registry
from pyramid.interfaces import IResponseFactory
reg = Registry()
config = self._makeOne(reg)
factory = lambda r: object()
config.setup_registry(response_factory=factory)
self.assertEqual(reg.queryUtility(IResponseFactory), None)
config.commit()
utility = reg.getUtility(IResponseFactory)
self.assertEqual(utility, factory)
def test_setup_registry_locale_negotiator(self):
from pyramid.registry import Registry
from pyramid.interfaces import ILocaleNegotiator
reg = Registry()
config = self._makeOne(reg)
negotiator = object()
config.setup_registry(locale_negotiator=negotiator)
self.assertEqual(reg.queryUtility(ILocaleNegotiator), None)
config.commit()
utility = reg.getUtility(ILocaleNegotiator)
self.assertEqual(utility, negotiator)
def test_setup_registry_dottedname_root_factory(self):
from pyramid.registry import Registry
from pyramid.interfaces import IRootFactory
reg = Registry()
config = self._makeOne(reg)
import tests.test_config
config.setup_registry(root_factory='tests.test_config')
self.assertEqual(reg.queryUtility(IRootFactory), None)
config.commit()
self.assertEqual(reg.getUtility(IRootFactory), tests.test_config)
def test_setup_registry_locale_negotiator(self):
from pyramid.registry import Registry
from pyramid.interfaces import ILocaleNegotiator
reg = Registry()
config = self._makeOne(reg)
negotiator = object()
config.setup_registry(locale_negotiator=negotiator)
self.assertEqual(reg.queryUtility(ILocaleNegotiator), None)
config.commit()
utility = reg.getUtility(ILocaleNegotiator)
self.assertEqual(utility, negotiator)
def open_pyramid_request(registry: Registry, path='http://localhost') -> ContextManager[PyramidRequest]:
request_factory = registry.queryUtility(IRequestFactory, default=Request)
request = request_factory.blank(path)
request.registry = registry
apply_request_extensions(request)
get_pyramid_root(request)
context = RequestContext(request)
context.begin()
try:
yield request
finally:
context.end()
def test_setup_registry_dottedname_root_factory(self):
from pyramid.registry import Registry
from pyramid.interfaces import IRootFactory
reg = Registry()
config = self._makeOne(reg)
import tests.test_config
config.setup_registry(root_factory='tests.test_config')
self.assertEqual(reg.queryUtility(IRootFactory), None)
config.commit()
self.assertEqual(reg.getUtility(IRootFactory), tests.test_config)
def get_social_login_mapper(registry: Registry, provider_id: str) -> ISocialLoginMapper:
"""Get a named social login mapper.
Example::
get_social_login_mapper(registry, "facebook")
:param registry: Pyramid registry.
:param provider_id: Provider id of a social login mapper.
:return: Implementation of ISocialLoginMapper.
"""
return registry.queryUtility(ISocialLoginMapper, name=provider_id)
def test_setup_registry_locale_negotiator_dottedname(self):
from pyramid.registry import Registry
from pyramid.interfaces import ILocaleNegotiator
reg = Registry()
config = self._makeOne(reg)
import tests.test_config
config.setup_registry(locale_negotiator='tests.test_config')
self.assertEqual(reg.queryUtility(ILocaleNegotiator), None)
config.commit()
utility = reg.getUtility(ILocaleNegotiator)
self.assertEqual(utility, tests.test_config)
def test_setup_registry_locale_negotiator_dottedname(self):
from pyramid.registry import Registry
from pyramid.interfaces import ILocaleNegotiator
reg = Registry()
config = self._makeOne(reg)
import tests.test_config
config.setup_registry(locale_negotiator='tests.test_config')
self.assertEqual(reg.queryUtility(ILocaleNegotiator), None)
config.commit()
utility = reg.getUtility(ILocaleNegotiator)
self.assertEqual(utility, tests.test_config)
def get_activation_model(registry: Registry) -> IActivationModel:
activation_model = registry.queryUtility(IActivationModel)
return activation_model
def get_activation_model(registry: Registry) -> IActivationModel:
activation_model = registry.queryUtility(IActivationModel)
return activation_model
def get_site_creator(registry: Registry) -> ISiteCreator:
site_creator = registry.queryUtility(ISiteCreator)
return site_creator
def get_group_class(registry: Registry) -> IGroupModel:
group_class = registry.queryUtility(IGroupModel)
return group_class
def get_group_class(registry: Registry) -> IGroupModel:
group_class = registry.queryUtility(IGroupModel)
return group_class
def get_user_class(registry: Registry) -> IUserModel:
user_class = registry.queryUtility(IUserModel)
return user_class
def get_site_creator(registry: Registry) -> ISiteCreator:
site_creator = registry.queryUtility(ISiteCreator)
return site_creator
def get_user_class(registry: Registry) -> IUserModel:
user_class = registry.queryUtility(IUserModel)
return user_class
