python tempfile.py

Posted 2021-05-09

import tempfile


class TempFileManager():
    def __init__(self):
        self.files = []

    def __enter__(self):
        return self

    def __exit__(self, *args):
        self.close()

    def create(self, suffix=''):
        file = tempfile.NamedTemporaryFile(suffix=suffix)
        self.files.append(file)
        return file

    def close(self):
        for f in self.files:
            f.close()