Flask中保存帖子图片始终返回None问题求助

问题描述 #

我用和保存用户头像几乎一样的方法保存帖子图片，但这个方法在帖子功能里完全失效，始终返回None。在创建帖子页面上传图片时，看起来文件已经上传成功，但点击发布按钮后图片就变成了None。我已经确认路径是正确的，甚至试过替换成已经生效的头像保存函数和目录，还是解决不了问题。

相关代码 #

Post模型 #

class Post(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    title = db.Column(db.String(100), nullable=False)
    date_posted = db.Column(db.DateTime, nullable=False, default=datetime.utcnow)
    content = db.Column(db.Text, nullable=False)
    user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
    image_file = db.Column(db.String(100), nullable=False)
    comments = db.relationship('Comment', backref='post', passive_deletes=True)
    likes = db.relationship('Like', backref='post', passive_deletes=True)
    def __repr__(self):
        return f"Post(\"{self.title}\", \"{self.date_posted}\")"

Post表单 #

class PostForm(FlaskForm):
    title = StringField('Title', validators=[DataRequired()])
    content = TextAreaField('Content', validators=[DataRequired()])
    image_file = FileField('Post picture', validators=[DataRequired(), FileAllowed(['jpg', 'png'])])
    submit = SubmitField('Post')

图片保存函数 #

def save_post_picture(form_picture):
    random_hex = secrets.token_hex(8)
    _, f_ext = os.path.splitext(form_picture.filename)
    picture_fn = random_hex + f_ext
    picture_path = os.path.join('static/post_images', picture_fn)
    output_size = (125, 125)
    i = Image.open(form_picture)
    i.thumbnail(output_size)
    i.save(picture_path)
    return picture_fn

新建帖子路由 #

@app.route("/post/new", methods=["GET", "POST"])
@login_required
def new_post():
    form = PostForm()
    if form.validate_on_submit():
        print('$ok')
        picture_file = save_post_picture(form.image_file.data)
        post = Post(title=form.title.data, content=form.content.data, user_id=current_user.id, image_file=picture_file)
        db.session.add(post)
        db.session.commit()
        print('works')
        flash('Your post has been created!', 'success')
        return redirect(url_for('home'))
    return render_template('create_post.html', title='New Post', form=form, legend='New Post')

模板表单片段 #

<div class="form-group">
    <label for="image_file" class="form-control-label">Image</label>
    {{ form.image_file(class="form-control-file") }}
    <small>Please make sure the file type is either jpg or png.</small>
    {% if form.image_file.errors %}
        {% for error in form.image_file.errors %}
            <small><span class="text-danger">{{ error }}</span></small>
        {% endfor %}
    {% endif %} 
</div>

排查及解决步骤 #

• 修复表单enctype属性：文件上传表单必须指定enctype="multipart/form-data"，否则后端无法接收文件。修改模板中的表单标签：

  <form method="POST" action="" enctype="multipart/form-data">
      <!-- 原有表单内容 -->
  </form>
  

• 添加错误捕获与绝对路径：当前保存函数没有异常处理，且使用相对路径可能导致问题。修改函数：

  import os
  from flask import current_app
  import secrets
  from PIL import Image
  
  def save_post_picture(form_picture):
      try:
          random_hex = secrets.token_hex(8)
          _, f_ext = os.path.splitext(form_picture.filename)
          picture_fn = random_hex + f_ext
          # 用current_app.root_path获取项目根目录，生成绝对路径
          post_images_dir = os.path.join(current_app.root_path, 'static/post_images')
          # 自动创建目录（如果不存在）
          if not os.path.exists(post_images_dir):
              os.makedirs(post_images_dir)
          picture_path = os.path.join(post_images_dir, picture_fn)
          
          output_size = (125, 125)
          with Image.open(form_picture) as i:
              i.thumbnail(output_size)
              i.save(picture_path)
          return picture_fn
      except Exception as e:
          print(f"图片保存失败: {str(e)}")
          return None
  

• 验证表单提交状态：在路由中打印表单错误，确认是否有验证未通过的情况：

  if form.validate_on_submit():
      print('$ok')
      print("表单错误:", form.errors)  # 新增这行
      picture_file = save_post_picture(form.image_file.data)
      # 后续代码
  

• 检查目录权限：确保服务器用户对static/post_images目录有读写权限，否则图片无法保存。

内容的提问来源于stack exchange，提问作者Eric Lee