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)
