use context manager for opening image

alan-turing-institute · Sep 19, 2024 · b6257fe · b6257fe
1 parent 103a6b0
commit b6257fe
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/prompto/scripts/convert_images.py b/src/prompto/scripts/convert_images.py
@@ -27,13 +27,13 @@ def main():
     for item in tqdm(os.listdir(args.folder)):
         file = os.path.join(args.folder, item)
         if file.lower().endswith(".jpg") or file.lower().endswith(".jpeg"):
-            image = Image.open(file)
-            converted_image = image.convert("RGB")
-            converted_image.save(file, "JPEG")
+            with Image.open(file) as image:
+                converted_image = image.convert("RGB")
+                converted_image.save(file, "JPEG")
         elif file.lower().endswith(".png"):
-            image = Image.open(file)
-            converted_image = image.convert("RGB")
-            converted_image.save(file, "PNG")
+            with Image.open(file) as image:
+                converted_image = image.convert("RGB")
+                converted_image.save(file, "PNG")
         else:
             logging.info(f"Skipping {file} as it is not a supported file type")