将上传的图像保存到 Flask 上的数据库

Posted 2023-03-23

【中文标题】将上传的图像保存到 Flask 上的数据库

【英文标题】：Save uploaded image to database on Flask

【发布时间】：2020-09-01 14:53:51

【问题描述】：

我正在学习 Python 并使用 Flask 开发应用程序，其中一个功能是用户可以上传个人资料图片，并且该图片应保存在数据库中。

我有一个表单，用户可以在其中上传文件，如下所示：

<form action="/myprofile" method="post" enctype="multipart/form-data">
    <div class="form-group">
        <input class="form-control" name="picture" placeholder="Picture" type="file">
    </div>
    <button class="btn btn-primary" type="submit">Submit</button>
</form>

然后，在 application.py 文件中我是这样处理的：

@app.route("/myprofile", methods=["GET", "POST"])
@login_required
def myprofile():
    if request.method == "POST":
        
        db.execute("UPDATE users SET role = :role, picture = :picture, linkedin = :linkedin WHERE id = :user_id", role = request.form.get("role"), picture = request.files("picture"), linkedin = request.form.get("linkedin"), user_id = session["user_id"])

        return render_template("home.html")
    else:
        return render_template("myprofile.html")

这将返回内部服务器错误。有人知道为什么吗？

感谢您的宝贵时间！

【问题讨论】：

我回答你。如果这没有帮助，那么添加你得到的错误代码。另外，如果我使用 SQLAlchemy 解决方案，那么我可以使用 Vanilla SQLite 上传我的存储库（这将需要一些时间），因为无论如何我都打算这样做。

【参考方案1】：

我没有看到您遇到什么错误，但是我想分享我之前关于将图像上传到 Flask 的答案之一。 我使用烧瓶、flask_sqlalchemy（使用 sqlite3）、html 和 Bootstrap 准备了一个迷你应用程序，它可以满足您的要求。

你可以在这里找到完整的代码（我会用更安全的文件上传来更新它）因为这里给出的只是一小部分：

FULL CODE

---

我的 Github 仓库中的一些代码

---

为数据库初始化数据库、配置和图片表

在类 FileContent(db.Model) 中：

data = file.read() 它将文件的二进制版本保存在数据库中 -render_file = 渲染图片（数据）。它保存解码版本，以便您可以在网页中看到它以进行渲染。

  # Built-in Imports
  import os
  from datetime import datetime
  from base64 import b64encode
  import base64
  from io import BytesIO #Converts data from Database into bytes

  # Flask
  from flask import Flask, render_template, request, flash, redirect, url_for, send_file # Converst bytes into a file for downloads

  # FLask SQLAlchemy, Database
  from flask_sqlalchemy import SQLAlchemy


  basedir = 'sqlite:///' + os.path.join(os.path.abspath(os.path.dirname(__file__)), 'data.sqlite')

  app = Flask(__name__)
  app.config['SQLALCHEMY_DATABASE_URI'] = basedir
  app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
  app.config['SECRET_KEY'] = 'dev'
  db = SQLAlchemy(app)

  # Picture table. By default the table name is filecontent
  class FileContent(db.Model):

      """ 
      The first time the app runs you need to create the table. In Python
      terminal import db, Then run db.create_all()
      """
      """ ___tablename__ = 'yourchoice' """ # You can override the default table name

      id = db.Column(db.Integer,  primary_key=True)
      name = db.Column(db.String(128), nullable=False)
      data = db.Column(db.LargeBinary, nullable=False) #Actual data, needed for Download
      rendered_data = db.Column(db.Text, nullable=False)#Data to render the pic in browser
      text = db.Column(db.Text)
      location = db.Column(db.String(64))
      pic_date = db.Column(db.DateTime, nullable=False, default=datetime.utcnow)
      def __repr__(self):
          return f'Pic Name: self.name Data: self.data text: self.text created on: self.pic_date location: self.location'

上传路径，这里是图片发送到数据库并使用正确数据处理的地方

这就是应用程序路由中发生的事情：

def render_picture(data) --> 获取图片的原始版本并返回解码版本，以便在网络上显示。

data = file.read() : 这是原始数据，可用于从数据库下载图片

render_file：解码文件，以便您可以检索它并在网页中呈现

#渲染图片，这个函数将数据从 request.files['inputFile'] 以便 in 可以显示

def render_picture(data):

    render_pic = base64.b64encode(data).decode('ascii') 
    return render_pic


@app.route('/upload', methods=['POST'])
def upload():

   file = request.files['inputFile']
   data = file.read()
   render_file = render_picture(data)
   text = request.form['text']
   location = request.form['location']

   newFile = FileContent(name=file.filename, data=data, 
   rendered_data=render_file, text=text, location=location)
   db.session.add(newFile)
   db.session.commit() 
   flash(f'Pic newFile.name uploaded Text: newFile.text Location: 
   newFile.location')

   return render_template('upload.html')

索引路线

# Index It routes to index.html where the upload forms is 
@app.route('/index', methods=['GET', 'POST'])
@app.route('/')
def index():

    return render_template('index.html')

使用表单索引 HTML

<form method="POST" action="/upload" enctype="multipart/form-data">
        <!-- File Upload-->
        <div class="form-group">
            <label for="inputFile">File input</label>
            <input class="form-control-file" type="file" name="inputFile">
        </div>

        <!-- Location -->
        <div class="form-group">
            <label for="location">Location</label>
            <input class="form-control" type="text" name="location">
        </div>

        <!-- Text -->
        <div class="form-group">
            <label for="text">Write Text</label>
            <textarea class="form-control" name="text" id="text" rows="5" placeholder="Add a Description"></textarea>
        </div>

        <!-- Submit -->        
        <button type="submit" class="btn btn-primary">Submit</button>
    </form>

【讨论】：

感谢您的回答！我只是不知道您如何选择保存图像的表格。我有一个名为 user_photo 的实体，它有一个 user_id 和一个用于保存二进制文件的列。

我稍微修改了我的答案。如果您的意思是在数据库的哪个表中，那么实际上我将它保存在 2 个不同的版本中两次。看class类FileContent(db.Model)：你会看到

data = db.Column(db.LargeBinary, nullable=False) #实际数据，下载需要的数据浏览器