Module `osbot_utils.testing.Temp_File`

Expand source code

from osbot_utils.utils.Files import Files, file_delete, folder_delete_all, files_list, file_create, \
    parent_folder, file_exists, file_contents
from osbot_utils.utils.Misc import random_filename


class Temp_File:
    def __init__(self, contents='...', extension='tmp',file_name=None, ):
        self.tmp_file            = file_name or random_filename(extension)
        self.tmp_folder          = None
        self.file_path           = None
        self.original_contents   = contents

    def __enter__(self):
        self.tmp_folder = Files.temp_folder(prefix='temp_folder_')
        self.file_path = Files.path_combine(self.tmp_folder, self.tmp_file)
        file_create(self.file_path, self.original_contents)
        return self

    def __exit__(self, type, value, traceback):
        file_delete      (self.file_path)
        folder_delete_all(self.tmp_folder)

    def contents(self):
        return file_contents(self.file_path)

    def delete(self):
        return file_delete(self.file_path)

    def exists(self):
        return file_exists(self.file_path)

    def file_name(self):
        return Files.file_name(self.path())

    def files_in_folder(self):
        return files_list(self.tmp_folder)

    def folder(self):
        return parent_folder(self.path())

    def path(self):
        return self.file_path

    def write(self, contents):
        file_create(self.file_path, contents)
        return self

Classes

class Temp_File (contents='...', extension='tmp', file_name=None)

Expand source code

class Temp_File:
    def __init__(self, contents='...', extension='tmp',file_name=None, ):
        self.tmp_file            = file_name or random_filename(extension)
        self.tmp_folder          = None
        self.file_path           = None
        self.original_contents   = contents

    def __enter__(self):
        self.tmp_folder = Files.temp_folder(prefix='temp_folder_')
        self.file_path = Files.path_combine(self.tmp_folder, self.tmp_file)
        file_create(self.file_path, self.original_contents)
        return self

    def __exit__(self, type, value, traceback):
        file_delete      (self.file_path)
        folder_delete_all(self.tmp_folder)

    def contents(self):
        return file_contents(self.file_path)

    def delete(self):
        return file_delete(self.file_path)

    def exists(self):
        return file_exists(self.file_path)

    def file_name(self):
        return Files.file_name(self.path())

    def files_in_folder(self):
        return files_list(self.tmp_folder)

    def folder(self):
        return parent_folder(self.path())

    def path(self):
        return self.file_path

    def write(self, contents):
        file_create(self.file_path, contents)
        return self

Methods

def contents(self)

Expand source code

def contents(self):
    return file_contents(self.file_path)

def delete(self)

Expand source code

def delete(self):
    return file_delete(self.file_path)

def exists(self)

Expand source code

def exists(self):
    return file_exists(self.file_path)

def file_name(self)

Expand source code

def file_name(self):
    return Files.file_name(self.path())

def files_in_folder(self)

Expand source code

def files_in_folder(self):
    return files_list(self.tmp_folder)

def folder(self)

Expand source code

def folder(self):
    return parent_folder(self.path())

def path(self)

Expand source code

def path(self):
    return self.file_path

def write(self, contents)

Expand source code

def write(self, contents):
    file_create(self.file_path, contents)
    return self