BlobStorage.py

import logging

import hashlib


class BlobStorage(object):
    """
    Class storing digits for training purposes
    """

    def __init__(self, storage_dir):
        self.storage_dir = storage_dir

    def blob_filename(self, blob):

        digit_list = []

        for digit in blob.getNumpy().ravel():
            digit_list.append(str(digit))

        md5 = hashlib.md5()
        md5.update("".join(digit_list))

        return str(md5.hexdigest())

    def store_blob(self, blob):
        """
        This method is called by GasMeter with a single image to be stored
        """

        store_path = self.storage_dir + "/blobstorage_" + str(self.blob_filename(blob)) + ".png"

        logging.debug("Storing blob " + store_path)

        blob.save(store_path)