diff --git a/tools/verification/rvgen/rvgen/automata.py b/tools/verification/rvgen/rvgen/automata.py index 4f5681265ee2..10146b6061ed 100644 --- a/tools/verification/rvgen/rvgen/automata.py +++ b/tools/verification/rvgen/rvgen/automata.py @@ -91,13 +91,11 @@ class Automata: cursor = 0 dot_lines = [] try: - dot_file = open(self.__dot_path) + with open(self.__dot_path) as dot_file: + dot_lines = dot_file.read().splitlines() except OSError as exc: raise AutomataError(exc.strerror) from exc - dot_lines = dot_file.read().splitlines() - dot_file.close() - # checking the first line: line = dot_lines[cursor].split() diff --git a/tools/verification/rvgen/rvgen/generator.py b/tools/verification/rvgen/rvgen/generator.py index 61174b139123..d932e96dd66d 100644 --- a/tools/verification/rvgen/rvgen/generator.py +++ b/tools/verification/rvgen/rvgen/generator.py @@ -51,11 +51,8 @@ class RVGenerator: raise FileNotFoundError("Could not find the rv directory, do you have the kernel source installed?") def _read_file(self, path): - fd = open(path, 'r') - - content = fd.read() - - fd.close() + with open(path, 'r') as fd: + content = fd.read() return content def _read_template_file(self, file): @@ -199,9 +196,8 @@ obj-$(CONFIG_RV_MON_{name_up}) += monitors/{name}/{name}.o return def __write_file(self, file_name, content): - file = open(file_name, 'w') - file.write(content) - file.close() + with open(file_name, 'w') as file: + file.write(content) def _create_file(self, file_name, content): path = f"{self.name}/{file_name}"