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
