python - 将上传的图片保存到 Flask 上的数据库中

Question

我正在学习 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")

这将返回一个内部服务器错误。有谁知道为什么？

感谢您的宝贵时间!

Answer 1

我没有看到您收到什么错误，但是我想分享我之前关于将图像上传到 Flask 的回答之一。我使用 flask、flask_sqlalchemy(使用 sqlite3)、html 和 Bootstrap 准备了一个迷你应用程序，它正在按照您的要求进行操作。

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

FULL CODE

---

我的 Github 仓库中的一些代码

---

为数据库启动数据库、配置和图片表

In class FileContent(db.Model):

• data = file.read() 它在数据库中保存文件的二进制版本-render_file = render_picture(数据)。它保存解码版本，以便您可以在网页中看到它以呈现它。

    # 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}'
  

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

So here is what is going on in the app route:

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

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

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

  #Render the pics, 这个函数将数据从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>