Source code for chap_core.external.external_model

import logging

logger = logging.getLogger(__name__)


# todo: remove, remove file


class SimpleFileContextManager:
    def __init__(self, filename, mode="r"):
        self.filename = filename
        self.mode = mode
        self.file = None

    def __enter__(self):
        self.file = open(self.filename, self.mode)
        return self.file

    def __exit__(self, exc_type, exc_value, traceback):
        if self.file:
            self.file.close()



    def write(self, data):
        if self.file:
            self.file.write(data)




    def read(self):
        if self.file:
            return self.file.read()