Làm thế nào để lưu tệp hình ảnh trên cơ sở dữ liệu Postgres?

Question

Vì mục đích học tập, tôi đang tạo một trang web bằng cách sử dụng Python + Flask. Tôi muốn khôi phục một hình ảnh từ cơ sở dữ liệu và hiển thị nó trên màn hình. Nhưng một bước tại một thời điểm.

Tôi không biết cách lưu hình ảnh trong cơ sở dữ liệu của mình ngay từ đầu. Tìm kiếm của tôi chỉ tiết lộ rằng tôi phải sử dụng loại bytea trong cơ sở dữ liệu của mình. Sau đó, tôi nhận được hình ảnh của tôi và bằng cách nào đó (??) chuyển đổi nó thành một mảng byte (bytea == mảng cắn?) Và bằng cách nào đó (??) sử dụng mảng này trong một lệnh chèn.

Tôi đã có thể khám phá (có thể) cách thực hiện trong Java (here) và C# (here), nhưng tôi thực sự muốn sử dụng Python, ít nhất là bây giờ.

Ai đó có thể giúp tôi không?

Có rất nhiều câu hỏi thuộc loại này trong trang web này. Nhưng hầu hết (dễ dàng hơn 85%) trong số họ được trả lời là "Bạn không nên lưu hình ảnh trong cơ sở dữ liệu của bạn, họ thuộc về fs" và không trả lời được câu hỏi. Phần còn lại không hoàn toàn giải quyết vấn đề của tôi. Vì vậy, vui lòng không đánh dấu mục này là trùng lặp nếu trùng lặp có loại câu trả lời này.

Answer 1

Tôi không thường viết các chương trình ví dụ hoàn chỉnh cho mọi người, nhưng bạn không đòi hỏi điều đó và đó là một trong khá đơn giản, vì vậy ở đây bạn đi:

#!/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) 

Được viết bằng Python 3.3. Sử dụng Python 2.7 yêu cầu bạn đọc tệp và chuyển đổi thành đối tượng buffer hoặc sử dụng các hàm đối tượng lớn. Chuyển đổi sang Python 2.6 trở lên yêu cầu cài đặt argparse, có thể là các thay đổi khác.

Bạn sẽ muốn thay đổi chuỗi kết nối cơ sở dữ liệu thành thứ gì đó phù hợp với hệ thống của bạn nếu bạn định chạy thử nó.

Nếu bạn đang làm việc với hình ảnh lớn xem xét sử dụng psycopg2's large object support thay vì bytea - nói riêng, lo_import cho cửa hàng, lo_export để viết trực tiếp vào một tập tin, và các đối tượng lớn đọc chức năng cho việc đọc khối nhỏ của hình ảnh tại một thời điểm .

Answer 2

Bạn có thể sử dụng mã số base64 của Python để mã hóa và giải mã chuỗi nhị phân tùy ý thành chuỗi văn bản.

Answer 3

Tôi hy vọng điều này sẽ phù hợp với bạn.

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 Image

Answer 4

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() 

Answer 5

đó là giải pháp của tôi, nó có thể làm việc trong trang web của tôi:

@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) 

sử dụng Flask, Flask-Alchemy để xử lý cơ sở dữ liệu.

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

đó là tệp html.you có thể nhúng nó vào html của bạn.