def get_object_or_404(query_or_model, *query):
if not isinstance(query_or_model, SelectQuery):
query_or_model = query_or_model.select()
try:
return query_or_model.where(*query).get()
except DoesNotExist:
abort(404)
def inner(*args, **kwargs):
if self.auth.autogc:
with self.auth.db:
if self.auth.check:
user = self.auth.get_logged_in_user()
if not user:
login_url = url_for('auth.login', next=get_next())
return redirect(login_url)
if not self.check_user_permission(user, perm_key):
return abort(403)
return func(*args, **kwargs)
else:
return func(*args, **kwargs)
else:
if self.auth.check:
user = self.auth.get_logged_in_user()
if not user:
login_url = url_for('auth.login', next=get_next())
return redirect(login_url)
if not self.check_user_permission(user, perm_key):
return abort(403)
return func(*args, **kwargs)
else:
return func(*args, **kwargs)
def inner(*args, **kwargs):
user = self.auth.get_logged_in_user()
if not user:
login_url = url_for('auth.login', next=get_next())
return redirect(login_url)
if not self.check_user_permission(user):
abort(403)
return func(*args, **kwargs)
def echo():
if not self.check_signature(
request.params.get("timestamp"),
request.params.get("nonce"),
request.params.get("signature")
):
return abort(403)
return request.params.get("echostr")#request.query.echostr
def edit(self, pk):
try:
instance = self.get_object(pk)
except self.model.DoesNotExist:
abort(404)
Form = self.get_edit_form(instance)
if request.method == 'POST':
form = Form(request.forms, obj=instance)
if form.validate():
self.save_model(instance, form, False)
flash('Changes to %s saved successfully' % self.get_display_name(), 'success')
return self.dispatch_save_redirect(instance)
else:
form = Form(obj=instance)
return render_template(self.templates['edit'],
model_admin=self,
instance=instance,
form=form,
**self.get_extra_context()
)
def handle():
if not self.check_signature(
request.params.get("timestamp"),
request.params.get("nonce"),
request.params.get("signature")
):
return abort(403)
body = request.body.read()
message = parse_user_msg(body)
logging.info("Receive message %s" % message)
reply = self.get_reply(message)
if not reply:
self.logger.warning("No handler responded message %s"
% message)
return ''
response.content_type = 'application/xml'
return create_reply(reply, message=message)
