Source code for nncore.io.handlers.base

# Copyright (c) Ye Liu. Licensed under the MIT License.



[docs]
class FileHandler(object):
    """
    Base class for file handlers. The inherited classes can optionally override
    :obj:`load_from_file`, :obj:`dump_to_file`, :obj:`load_from_str`, and
    :obj:`dump_to_str` methods to support loading or dumping data.
    """

    def load_from_file(self):
        raise NotImplementedError

    def dump_to_file(self):
        raise NotImplementedError

    def load_from_str(self):
        raise NotImplementedError

    def dump_to_str(self):
        raise NotImplementedError

    def load_from_path(self, path, mode='r', **kwargs):
        with open(path, mode) as f:
            return self.load_from_file(f, **kwargs)

    def dump_to_path(self, obj, path, mode='w', **kwargs):
        with open(path, mode) as f:
            self.dump_to_file(obj, f, **kwargs)