How to use Python program to store images to MySQL

Environment

Python 3.7.4
pymysql
8.0.11 MySQL Community Server

Read pictures

Read pictures in binary format

with open("./test.jpg", "rb") as file:
	image = file.read()

Create a table to store pictures

The attributes of the picture fields are longblog, that is, long binary large object

def create_image_table(self):
	sql = 'create table if not exists picture ( \
        image longblob);'

    try:
        self.cursor.execute(sql)

        self.connection.commit()

    except pymysql.Error:
        print(pymysql.Error)

is stored in MySQL

Stores image data in binary format into MySQL

def insert_image(self, image):
    sql = "insert into picture(image) values(%s)"
    self.cursor.execute(sql, image)
    self.connection.commit()

Save the picture data obtained by MySQL query as picture

Write the picture in binary format

def get_image(self, path):
    sql = 'select * from picture'
    try:
        self.cursor.execute(sql)
        image = self.cursor.fetchone()[0]
        with open(path, "wb") as file:
            file.write(image)
    except pymysql.Error:
        print(pymysql.Error)
    except IOError:
        print(IOError)

Implementation code

import pymysql


class Database():
	
	'''
		Description:
			database demo to store image in MySQL RDBMS
		Attributes:
			None
	'''
    
    def __init__(self):
        self.connection = pymysql.connect(host=&#39;<host name>&#39;,user=&#39;<user name>&#39;,passwd=&#39;<password>&#39;,db=&#39;<database name>&#39;,charset=&#39;utf8&#39;)
        self.cursor = self.connection.cursor()

	&#39;&#39;&#39;
		Description:
			create table to store images
		Args:
			None
		Return:
			None
	&#39;&#39;&#39;
    
    def create_image_table(self):
        sql = &#39;create table if not exists picture ( \
            image longblob);&#39;

        try:
            self.cursor.execute(sql)

            self.connection.commit()

        except pymysql.Error:
            print(pymysql.Error)
	
	&#39;&#39;&#39;
		Description:
			insert image into table
		Args:
			image:
				image to store
		Returns:
			None
	&#39;&#39;&#39;

    def insert_image(self, image):
        sql = "insert into picture(image) values(%s)"
        self.cursor.execute(sql, image)
        self.connection.commit()
	
	&#39;&#39;&#39;
		Description:
			get image from database
		Args:
			path:
				path to save image
		Returns:
			None
	&#39;&#39;&#39;	

    def get_image(self, path):
        sql = &#39;select * from picture&#39;
        try:
            self.cursor.execute(sql)
            image = self.cursor.fetchone()[0]
            with open(path, "wb") as file:
                file.write(image)
        except pymysql.Error:
            print(pymysql.Error)
        except IOError:
            print(IOError)
            
	&#39;&#39;&#39;
		Description:
			destruction method
		Args:
			None
		Returns:
			None
	&#39;&#39;&#39;
	
    def __del__(self):
        self.connection.close()
        self.cursor.close()

if __name__ == "__main__":
    database = Database()
    # read image from current directory
    with open("./test.jpg", "rb") as file:
        image = file.read()

    database.create_image_table()
    database.insert_image(image)

    database.get_image(&#39;./result.jpg&#39;)

Test result

The above is the detailed content of How to use Python program to store images to MySQL. For more information, please follow other related articles on the PHP Chinese website!