def post(self, obj_id=None, parent_id=None): self.set_object(obj_id, parent_id) self.extra() if self.error: flash(self.error_msg, self.error_level) return redirect(url_for(**self.redirect)) current_app.logger.debug('POSTED========================================') if self.form.validate_on_submit(): current_app.logger.debug('VALIDATED========================================') if self.action == 'Edit': log_orig = log_change(self.obj) self.pre_post() self.form.populate_obj(self.obj) self.post_post() log_change(log_orig, self.obj, self.log_msg) else: self.pre_post() self.form.populate_obj(self.obj) self.post_post() db.session.add(self.obj) log_new(self.obj, self.log_msg) db.session.commit() self.post_submit() flash(self.success_msg, 'success') if self.redirect: return redirect(url_for(**self.redirect)) log_form(self.form) flash_form_errors(self.form) return render_template(self.template, **self.context)
def delete_comment(obj_id=None): from app.models import Comment if obj_id: comment = Comment.query.filter_by(id=obj_id).first() log_new(comment, 'deleted a comment') db.session.delete(comment) db.session.commit() flash('Comment deleted successfully.', 'success') return redirect(request.referrer) form = DeleteCommentForm() if form.validate_on_submit(): comment = Comment.query.filter_by(id=form.comment_id.data).first() log_new(comment, 'deleted a comment') db.session.delete(comment) db.session.commit() flash('Comment deleted successfully.', 'success') return redirect(form.redirect.data) return redirect(url_for('main.home'))
def sign_up(checkout=''): form = SignUpForm() if form.validate_on_submit(): user = User() form.populate_obj(obj=user) user.set_password(form.new_password.data) customer = Group.query.filter_by(name='customer').first() if customer: user.groups = [customer] log_new(user, f'User added for email: {form.email.data}') db.session.add(user) db.session.commit() login_user(user, remember=False) flash('Thanks for creating an account!', 'success') if checkout == 'checkout': return redirect(url_for('shop.shipping')) return redirect(url_for('shop.index')) form.subscribed.data = True return render_template('auth/login.html', title='Sign Up', form=form, user='', checkout=checkout)
def submit_comment(): form = CommentForm() if form.validate_on_submit(): comment = Comment( text=comment_form.comment.data, user_id=current_user.id, punch_id=form.punch_id.data, event_id=form.event_id.data, ) db.session.add(comment) db.session.commit() if form.punch_id.data: comment.notify( edited_user=user, url=url_for( 'punch.edit', username=user.username, punch_id=punch_id, ), item=f"{user.display_name_short()}'s punch card", ) log_new(comment, 'commented on a punch card') if form.event_id: comment.notify( edited_user=user, url=url_for( 'schedule.leave_edit', username=user.username, leave_id=leave_id, ), item=f"{user.display_name_short()}'s leave request", ) log_new(comment, 'commented on an event') flash('Comment added successfully!', 'success') return redirect(request.referer)
def post(self, parent_id=None): self.extra() self.form = self.form() if self.form.validate_on_submit(): self.obj = self.model.query.filter_by(id=self.form.obj_id.data).first() log_new(self.obj, self.log_msg) self.pre_post() db.session.delete(self.obj) self.post_post() db.session.commit() self.post_submit() flash(self.success_msg, 'success') else: flash(f'Failed to delete {self.model.__name__}!', 'danger') log_form(self.form) #flash_form_errors(self.form) try: if self.form.redirect.data: return redirect(self.form.redirect.data) except Exception as e: current_app.logger.debug(e) if parent_id: return redirect(url_for(**self.redirect, obj_id=parent_id)) return redirect(url_for(**self.redirect))