cic-cli/cic/attachment.py

# standard imports
import logging
import os

# local imports
from .base import *

logg = logging.getLogger(__name__)


class Attachment(Data):
    """Processes, serialized and publishes all attachments found in the "attachments" subdirectory of the settings directory.

    :param path: Path to settings directory
    :type path: str
    :param writer: Writer interface receiving the output of the processor
    :type writer: cic.output.OutputWriter
    """
    def __init__(self, path='.', writer=None):
        super(Attachment, self).__init__()
        self.contents = {}
        self.path = path
        self.writer = writer
        self.attachment_path = os.path.join(self.path, 'attachments')


    def load(self):
        """Loads attachment data from settings.
        """
        for s in os.listdir(self.attachment_path):
            fp = os.path.realpath(os.path.join(self.attachment_path, s))
            f = open(fp, 'rb')
            r = f.read()
            f.close()

            z = self.hash(r).hex()
            self.contents[z] = fp

            logg.debug('loaded attachment file {} digest {}'.format(fp, z))


    def start(self):
        """Initialize attachment settings from template.
        """
        super(Attachment, self).start()
        os.makedirs(self.attachment_path)


    def get(self, k):
        """Get a single attachment by the sha256 hash of the content.

        :param k: Content hash
        :type k: str (hex)
        """
        return self.contents[k]


    def asdict(self):
        """Output attachment state to dict
        """
        return self.contents


    def process(self, token_address=None, token_symbol=None, writer=None):
        """Serialize and publish attachments.

        See cic.processor.Processor.process 
        """
        if writer == None:
            writer = self.writer

        for k in self.contents.keys():
            fp = os.path.join(self.attachment_path, self.contents[k])
            f = open(fp, 'rb')
            v = f.read()
            f.close()
            logg.debug('writing attachment {}'.format(k))
            writer.write(k, v)


    def __str__(self):
        s = ''
        for k in self.contents.keys():
            s += '{} = {}\n'.format(k, self.contents[k]) #self.digests[i].hex(), self.contents[i])

        return s