Source code for PyOpenWorm.data_trans.file_ds

from contextlib import contextmanager
from rdflib.namespace import Namespace
from ..datasource import Informational, DataSource
from .common_data import DS_NS


class FileDataSource(DataSource):
    rdf_namespace = Namespace(DS_NS['FileDataSource#'])
    md5 = Informational(display_name='MD5 hash')
    sha256 = Informational(display_name='SHA-256 hash')
    sha512 = Informational(display_name='SHA-512 hash')

    @contextmanager
    def file_contents(self):
        raise NotImplementedError()

    def update_hash(self, algorithm):
        import hashlib
        hsh = hashlib.new(algorithm)
        with self.file_contents() as f:
            hsh.update(f.read())
        getattr(self, algorithm).set(hsh.hexdigest())


__yarom_mapped_classes__ = (FileDataSource,)