def preview_add():
preview_form = PreviewForm()
if preview_form.validate_on_submit():
data = preview_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(preview_form.logo.data.filename)
logo = change_filename(file_logo)
preview_form.logo.data.save(app.config['UP_DIR'] + logo)
preview = Preview()
preview.title = data['title']
preview.logo = logo
db.session.add(preview)
db.session.commit()
flash('增加预告成功', 'ok')
return redirect(url_for('admin.preview_add'))
return render_template('admin/preview-add.html', preview_form=preview_form)
def preview_add():
form = PreviewForm()
if form.validate_on_submit():
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'))
file_logo = secure_filename(form.logo.data.filename)
if not os.path.exists(app.config["UP_DIR"]):
os.makedirs(app.config["UP_DIR"])
os.chmod(app.config["UP_DIR"], "rw")
logo = change_filename(file_logo)
form.logo.data.save(app.config["UP_DIR"] + logo)
preview = Preview()
preview.title = data["title"]
preview.logo = logo
db.session.add(preview)
db.session.commit()
flash("提交成功", "ok")
return redirect(url_for('admin.preview_add'))
return render_template("admin/preview_add.html", form=form)
