Извлечение сохраненного изображения из mongodb с использованием python

Question

from pymongo import MongoClient 
from bson.objectid import ObjectId 
import numpy as np 
import gridfs 

import os,os.path 
i=0 
try: 
    for file in os.listdir("/Users/sarthakgupta/Desktop/sae/Images"): 
     if (file.endswith(".png") | file.endswith(".jpg")): 
      filename = "/Users/sarthakgupta/Desktop/sae/Images/"+file 
      datafile = open(filename,"rb") 
      thedata = datafile.read() 
      datafile.close() 


      c = MongoClient() 
      i=i+1 
      db = c.trial5 
      fs = gridfs.GridFS(db) 
      t = "class"+str(i) 


      stored = fs.put(thedata,filename=q) 

except IOError: 
    print("Image file %s not found" %datafile) 
    raise SystemExit 

Я сохранил изображение в mongo db. Теперь я хочу получить эти изображения из базы данных по имени файла и сохранить изображение (или пиксели) одного и того же имени файла в массиве или списке. Предположим, если есть 2 изображения с именем файла «class1», то они должны быть в одном массиве.

Answer 1

Создайте свой fs переменную, как и раньше, и:

data = fs.get_last_version(filename).read() 

Вы также можете запросить список файлов, как:

from bson import Regex 
for f in fs.find({'filename': Regex(r'.*\.(png|jpg)')): 
    data = f.read() 

Кроме того, комментарий о коде: это очень медленно воссоздайте экземпляры MongoClient и GridFS для каждой итерации вашего цикла. Создайте их один раз, прежде чем начинать цикл и повторно использовать их.