def save_images_to_db(image_dir, image_type): mongo = db.get_bot() image_paths = collect_images(image_dir) saves = 0 print '[ ] saving to mongodb ' for image_path in image_paths: saves += mongo.save_image(image_path, image_type) print '.', print '\nsaved %d images to database' % saves
def save_images_to_db(image_dir,image_type): mongo = db.get_bot() image_paths = collect_images(image_dir) saves = 0 print '[ ] saving to mongodb ' for image_path in image_paths: saves += mongo.save_image(image_path,image_type) print '.', print '\nsaved %d images to database' % saves
def load_images_from_db(image_type, destination_directory): mongo = db.get_bot() results = mongo.get_imagetypes(image_type) try: os.makedirs(destination_directory) except OSError: #directory exists pass os.path.dirname(destination_directory) for res in results: try: path = os.path.join(destination_directory, res['filename']) fil = open(path, 'wb') fil.write(res['binary']) fil.close() except IOError as e: print e print 'loaded %d results from db' % len(results)
def load_images_from_db(image_type,destination_directory): mongo = db.get_bot() results = mongo.get_imagetypes(image_type) try: os.makedirs(destination_directory) except OSError: #directory exists pass os.path.dirname(destination_directory) for res in results: try: path = os.path.join(destination_directory,res['filename']) fil = open(path,'wb') fil.write(res['binary']) fil.close() except IOError as e: print e print 'loaded %d results from db' % len(results)