Use with statement to write to file

erocarrera · Jul 4, 2024 · fee84a7 · fee84a7
1 parent 7368b0c
commit fee84a7
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/pefile.py b/pefile.py
@@ -3521,10 +3521,8 @@ def write(self, filename=None):
         if not filename:
             return new_file_data
 
-        f = open(filename, "wb+")
-        f.write(new_file_data)
-        f.close()
-        return
+        with open(filename, "wb+") as f:
+            f.write(new_file_data)
 
     def parse_sections(self, offset):
         """Fetch the PE file sections.