def test_import_submodules(): from muffin import import_submodules result = import_submodules('muffin') assert len(result) == 7 result = import_submodules('muffin', 'plugins', 'manage') assert len(result) == 2
def test_import_submodules(): from muffin import import_submodules result = import_submodules('muffin') assert len(result) == 8 result = import_submodules('muffin', 'plugins', 'manage') assert len(result) == 2
"""API points.""" from muffin import import_submodules from muffin_rest import Api, PWRESTHandler, HTTPForbidden from salesmeter import app v1 = Api(app, "/api/v1") class AuthorizedResource(PWRESTHandler): """Allow access only for logged users.""" def authorize(self, request): """Ensure that current user is logged.""" user = yield from app.ps.session.load_user(request) if user is None: raise HTTPForbidden(reason="Access denied.") return user # Initialize others handlers import_submodules(__name__)
app.on_response_prepare.append(on_prepare) from salesmeter.models import Operator @app.ps.session.user_loader def load_operator(pk): """Load current user from saved session PK.""" try: return Operator.get(Operator.id == pk) except Operator.DoesNotExist: return None @app.ps.babel.locale_selector def set_locale(request): """Return locale from GET lang-param or automatically.""" return request.GET.get( 'lang', # Get locale based on user settings app.ps.babel.select_locale_by_request(request) ) muffin.import_submodules(__name__) # pylama:ignore=W0401,E402