Как сохранить файл изображения в базе данных Postgres?

для учебных целей я создаю сайт, используя Python+Flask. Я хочу восстановить изображение из базы данных и показать его на экране. Но шаг за шагом.

Я понятия не имею, как сохранить изображение в моей базе данных в первую очередь. Мои поиски только показали, что я должен использовать bytea введите мою базу данных. Затем я получаю свой образ и как-то (??) преобразовать его в массив байтов (bytea == массив укусов?) и как-то (??) используйте этот массив в команде insert.

I смог обнаружить (возможно), как это сделать на Java (здесь) и C# (здесь), но я бы очень хотел использовать Python, по крайней мере сейчас.

может кто-нибудь помочь мне?

есть тонны вопросов такого рода на этом сайте. Но большинство (легко более 85%) из них отвечают как "вы не должны сохранять изображения в своей базе данных, они принадлежат к fs" и не отвечают на вопрос. Остальное не решает мою проблему. Поэтому, пожалуйста, не отмечайте это как дубликат, если у дубликата есть такой ответ.

5 ответов


Я обычно не пишу полные примеры программ для людей, но вы этого не требовали, и это довольно просто, поэтому вот вы идете:

#!/usr/bin/env python3

import os
import sys
import psycopg2
import argparse

db_conn_str = "dbname=regress user=craig"

create_table_stm = """
CREATE TABLE files (
    id serial primary key,
    orig_filename text not null,
    file_data bytea not null
)
"""

def main(argv):
    parser = argparse.ArgumentParser()
    parser_action = parser.add_mutually_exclusive_group(required=True)
    parser_action.add_argument("--store", action='store_const', const=True, help="Load an image from the named file and save it in the DB")
    parser_action.add_argument("--fetch", type=int, help="Fetch an image from the DB and store it in the named file, overwriting it if it exists. Takes the database file identifier as an argument.", metavar='42')
    parser.add_argument("filename", help="Name of file to write to / fetch from")

    args = parser.parse_args(argv[1:])

    conn = psycopg2.connect(db_conn_str)
    curs = conn.cursor()

    # Ensure DB structure is present
    curs.execute("SELECT 1 FROM information_schema.tables WHERE table_schema = %s AND table_name = %s", ('public','files'))
    result = curs.fetchall()
    if len(result) == 0:
        curs.execute(create_table_stm)

    # and run the command
    if args.store:
        # Reads the whole file into memory. If you want to avoid that,
        # use large object storage instead of bytea; see the psycopg2
        # and postgresql documentation.
        f = open(args.filename,'rb')

        # The following code works as-is in Python 3.
        #
        # In Python 2, you can't just pass a 'str' directly, as psycopg2
        # will think it's an encoded text string, not raw bytes. You must
        # either use psycopg2.Binary to wrap it, or load the data into a
        # "bytearray" object.
        #
        # so either:
        #
        #   filedata = psycopg2.Binary( f.read() )
        #
        # or
        #
        #   filedata = buffer( f.read() )
        #
        filedata = f.read()
        curs.execute("INSERT INTO files(id, orig_filename, file_data) VALUES (DEFAULT,%s,%s) RETURNING id", (args.filename, filedata))
        returned_id = curs.fetchone()[0]
        f.close()
        conn.commit()
        print("Stored {0} into DB record {1}".format(args.filename, returned_id))

    elif args.fetch is not None:
        # Fetches the file from the DB into memory then writes it out.
        # Same as for store, to avoid that use a large object.
        f = open(args.filename,'wb')
        curs.execute("SELECT file_data, orig_filename FROM files WHERE id = %s", (int(args.fetch),))
        (file_data, orig_filename) = curs.fetchone()

            # In Python 3 this code works as-is.
            # In Python 2, you must get the str from the returned buffer object.
        f.write(file_data)
        f.close()
        print("Fetched {0} into file {1}; original filename was {2}".format(args.fetch, args.filename, orig_filename))

    conn.close()

if __name__ == '__main__':
    main(sys.argv)

написано на Python 3.3. Использование Python 2.7 требует, чтобы вы прочитали файл и конвертировали его в buffer объект или большой объект функции. Преобразование в Python 2.6 и старше требует установки argparse, возможно, других изменений.

вы захотите изменить строку подключения к базе данных на что-то подходящее для вашей системы, если вы собираетесь протестировать ее.

если вы работаете с большими изображениями рекомендуется использовать psycopg2 это вместо


Я надеюсь, что это сработает для вас.

import Image
import StringIO
im = Image.open("file_name.jpg") # Getting the Image
fp = StringIO.StringIO()
im.save(fp,"JPEG")
output = fp.getvalue() # The output is 8-bit String.

StringIO изображения


import psycopg2
import sys

def readImage():
    try:
        fin = open("woman.jpg", "rb")
        img = fin.read()
        return img

    except IOError, e:
        print "Error %d: %s" % (e.args[0],e.args[1])
        sys.exit(1)

    finally:
        if fin:
            fin.close()
try:
    con = psycopg2.connect(database="testdb", user="abc")
    cur = con.cursor()
    data = readImage()
    binary = psycopg2.Binary(data)
    cur.execute("INSERT INTO images(id, data) VALUES (1, %s)", (binary,) )
    con.commit()
except psycopg2.DatabaseError, e:
    if con:
        con.rollback()
    print 'Error %s' % e    
    sys.exit(1)
finally: 
    if con:
        con.close()

вы можете использовать Python в base64 для кодирования и декодирования произвольных двоичных строк в текстовых строк.


Это мое решение, оно может работать на моем сайте:

@main.route('/upload', methods=['GET', 'POST'])
def upload_avatar():
    if request.method == 'POST':
        file = request.files['file']
        if file and allowed_file(file.filename):
            current_user.avatar_local = file.read()
            db.session.add(current_user)
            db.session.commit()
            return redirect(url_for('main.user_page', username=current_user.username))
    return render_template('upload_avatar.html', user=current_user)

используя колбу, колбу-алхимию для обработки базы данных.

{% block edit_avatar  %}
    <form action="" method=post enctype=multipart/form-data>
      <p><input type=file name=file>
         <input type=submit value=Upload>
    </form>
{% endblock %}

это html-файл.вы можете вставить его в вас html.