python如何保存数据库中图片

Python可以通过多种方式保存数据库中的图片，例如使用SQLAlchemy ORM、Django ORM等。你可以通过将图片文件存储在文件系统中，并在数据库中保存文件路径，或者将图片数据以二进制格式直接存储在数据库中。推荐使用文件系统存储图片，并在数据库中保存路径，这样可以减少数据库的负担并提高效率。

下面将详细介绍如何使用SQLAlchemy和SQLite将图片数据保存到数据库中。

一、使用文件系统存储图片，并在数据库中保存路径

1、创建目录结构和虚拟环境

在你的工作目录中创建一个新的项目目录，并在其中创建一个虚拟环境：

mkdir image_storage_project cd image_storage_project python -m venv venv source venv/bin/activate # On Windows use `venv\Scripts\activate`

2、安装所需的库

安装SQLAlchemy、Pillow和其他必要的库：

pip install sqlalchemy pillow

3、创建数据库模型和保存图片路径

在项目目录中创建一个Python脚本（例如app.py），并编写以下代码：

import os
import shutil
from sqlalchemy import create_engine, Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker
from PIL import Image
配置数据库连接
DATABASE_URL = 'sqlite:///images.db'
engine = create_engine(DATABASE_URL, echo=True)
Base = declarative_base()
定义数据库模型
class ImageModel(Base):
    __tablename__ = 'images'
    id = Column(Integer, primary_key=True)
    file_path = Column(String, unique=True, nullable=False)
创建数据库表
Base.metadata.create_all(engine)
创建会话
Session = sessionmaker(bind=engine)
session = Session()
创建保存图片的目录
IMAGE_DIR = 'images'
os.makedirs(IMAGE_DIR, exist_ok=True)
def save_image_to_db(image_path):
    # 检查文件是否存在
    if not os.path.exists(image_path):
        print(f"File {image_path} does not exist.")
        return
    # 生成新的文件名和路径
    file_name = os.path.basename(image_path)
    new_path = os.path.join(IMAGE_DIR, file_name)
    # 复制图片到新的路径
    shutil.copy(image_path, new_path)
    # 创建数据库记录
    image_record = ImageModel(file_path=new_path)
    session.add(image_record)
    session.commit()
    print(f"Image saved to database with path: {new_path}")
示例用法
save_image_to_db('path_to_your_image.jpg')

4、运行脚本

将示例用法中的path_to_your_image.jpg替换为你要保存的图片路径，然后运行脚本：

python app.py

这段代码会将图片复制到images目录中，并在数据库中保存图片的路径。

二、将图片数据以二进制格式直接存储在数据库中

虽然不推荐，但如果你确实需要将图片数据存储在数据库中，可以参考以下示例：

1、创建数据库模型并保存图片数据

修改app.py，使其包含以下代码：

import os
from sqlalchemy import create_engine, Column, Integer, String, LargeBinary
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker
配置数据库连接
DATABASE_URL = 'sqlite:///images.db'
engine = create_engine(DATABASE_URL, echo=True)
Base = declarative_base()
定义数据库模型
class ImageModel(Base):
    __tablename__ = 'images'
    id = Column(Integer, primary_key=True)
    file_name = Column(String, unique=True, nullable=False)
    data = Column(LargeBinary, nullable=False)
创建数据库表
Base.metadata.create_all(engine)
创建会话
Session = sessionmaker(bind=engine)
session = Session()
def save_image_to_db(image_path):
    # 检查文件是否存在
    if not os.path.exists(image_path):
        print(f"File {image_path} does not exist.")
        return
    # 读取图片数据
    with open(image_path, 'rb') as file:
        image_data = file.read()
    # 获取文件名
    file_name = os.path.basename(image_path)
    # 创建数据库记录
    image_record = ImageModel(file_name=file_name, data=image_data)
    session.add(image_record)
    session.commit()
    print(f"Image saved to database with name: {file_name}")
示例用法
save_image_to_db('path_to_your_image.jpg')

2、运行脚本

将示例用法中的path_to_your_image.jpg替换为你要保存的图片路径，然后运行脚本：

python app.py

这段代码会将图片数据以二进制格式存储在数据库中。

三、从数据库中读取图片并显示

1、读取图片路径并显示图片

如果你使用的是文件系统存储图片的方式，可以使用以下代码读取图片路径并显示图片：

from PIL import Image
def display_image_from_db(image_id):
    # 查询数据库记录
    image_record = session.query(ImageModel).filter_by(id=image_id).first()
    if not image_record:
        print(f"No image found with ID {image_id}")
        return
    # 打开并显示图片
    image = Image.open(image_record.file_path)
    image.show()
示例用法
display_image_from_db(1)

2、读取二进制数据并显示图片

如果你使用的是二进制格式存储图片的方式，可以使用以下代码读取图片数据并显示图片：

import io
from PIL import Image
def display_image_from_db(image_id):
    # 查询数据库记录
    image_record = session.query(ImageModel).filter_by(id=image_id).first()
    if not image_record:
        print(f"No image found with ID {image_id}")
        return
    # 将二进制数据转换为图片并显示
    image_data = image_record.data
    image = Image.open(io.BytesIO(image_data))
    image.show()
示例用法
display_image_from_db(1)