aswi2021jmsd-gitlab/src/utils/temporary_file.py @ 2510f01a

import os


# allows the caller, when using "with" keyword, access a temporary file with the desired content that is deleted
# when "with" context is destroyed
class TemporaryFile(object):
    def __init__(self, file_name, content):
        self.file_name = file_name
        self.content = content

    def __enter__(self):
        # if content was passed, write it to a new file specified by the given file name
        if len(self.content) > 0:
            with open(self.file_name, "w") as file:
                file.write(self.content)
        return self.file_name

    def __exit__(self, type_, value, traceback):
        if len(self.content) > 0:
            # if content was passed then the file has been created => delete it.
            os.remove(self.file_name)