def preview_edit(id): form = PreviewForm() form.logo.validators = [] form.title.validators = [] preview = Preview.query.get_or_404(int(id)) if form.is_submitted(): data = form.data preview_count = Preview.query.filter_by(title=data['title']).count() if preview_count == 1 and preview.title != data['title']: flash("标题已存在", "err") return redirect(url_for('admin.preview_edit')) if not os.path.exists(app.config['UP_DIR']): os.makedirs(app.config['UP_DIR']) os.chmod(app.config['UP_DIR'], 'rw') if form.logo.data != '': file_logo = secure_filename(form.logo.data.filename) preview.logo = change_filename(file_logo) form.logo.data.save(app.config['UP_DIR'] + preview.logo) preview.title = data['title'] db.session.add(preview) db.session.commit() flash("修改成功!", 'ok') return redirect(url_for('admin.preview_list', page=1)) return render_template("admin/preview_edit.html", form=form, preview=preview)
def preview_add(): form = PreviewForm() if form.is_submitted(): data = form.data preview_count = Preview.query.filter_by(title=data['title']).count() if preview_count == 1: flash("标题已存在", "err") return redirect(url_for('admin.preview_add')) if not os.path.exists(app.config['UP_DIR']): os.makedirs(app.config['UP_DIR']) os.chmod(app.config['UP_DIR'], 'rw') file_logo = secure_filename(form.logo.data.filename) logo = change_filename(file_logo) form.logo.data.save(app.config['UP_DIR'] + logo) preview = Preview(title=data['title'], logo=logo) db.session.add(preview) db.session.commit() flash("添加成功!", 'ok') return render_template("admin/preview_add.html", form=form)