python批量讀取圖片且存入數據庫的實現

# --* coding=utf-8 *--
from cStringIO import StringIO
from pymongo import MongoClient
import gridfs
import os
import matplotlib.pyplot as plt
import matplotlib.image as iming
import bson.binary
import numpy as np
if name == 'main':
 connect = MongoClient('127.0.0.1', 27017) # 創建連接點
 db = connect.mydb
 print db.collection_names()
 imgput = gridfs.GridFS(db)
 dirs = 'G:image'
 files = os.listdir(dirs)
 for file in files:
 filesname = dirs + '\' + file
 print filesname
 imgfile=iming.imread(filesname)
 # iming.imsave('s.jpg',imgfile)
 # print type(imgfile),imgfile
 # imgfile.shape()
 plt.imshow(imgfile)
 plt.axis('off')
 plt.show()
 f=file.split('.')
 print f
 datatmp=open(filesname,'rb')
 data=StringIO(datatmp.read())
 content=bson.binary.Binary(data.getvalue())
 # print content
 insertimg=imgput.put(data,content_type=f[1],filename=f[0])
 datatmp.close()