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)