Init commit

BatImgCon.py

@@ -0,0 +1,111 @@
+# Batch Image Converter (BatImgCon) v1.0.0
+
+# TODO
+# - Linux support for priorities
+
+import os, glob, sys, argparse, signal, time
+import pillow_avif
+import typer
+import win32api, win32process
+from concurrent.futures import ThreadPoolExecutor, as_completed
+from PIL import Image
+from rich import print
+
+# Flag to indicate if the script should be interrupted
+interrupted = False
+def signal_handler(sig, frame):
+    global interrupted
+    print("\n[bold red]Interrupt received, stopping ASAP. Please wait while existing threads finish...[/bold red]\n")
+    interrupted = True
+
+# Register the signal handler
+signal.signal(signal.SIGINT, signal_handler)
+
+def reduce_prio():
+    if sys.platform == "win32":
+        try:
+            win32process.SetPriorityClass(
+                win32api.GetCurrentProcess(),
+                win32process.BELOW_NORMAL_PRIORITY_CLASS
+            )
+            print("Process priority reduced successfully to 'Below Normal'")
+        except Exception as error:
+            print(f"Failed to reduce process priority: {error}")
+    else:
+        print("Unsupported operating system, process priority not reduced")
+
+def convert_file(file: str, output_dir: str, input_format: str, output_format: str) -> bool:
+    if interrupted:
+        return False  # Exit early if interrupted
+    try:
+        image = Image.open(file)
+        output_path = os.path.join(output_dir, os.path.basename(file).replace(f".{input_format}", f".{output_format}"))
+        image.save(output_path, format=output_format)
+        image.close()
+        print(f"Converted {file} to {output_path}")
+        return True
+    except Exception as e:
+        image.close()
+        print(f"Failed to convert {file}: {e}")
+        return False
+
+def main(input_dir: str, input_format: str, output_dir: str, output_format: str, workers_count: int = os.cpu_count(), reduced_prio: bool = True):
+    """
+    Converts all images of INPUT_FORMAT inside INPUT_DIR to OUTPUT_FORMAT inside OUTPUT_DIR
+
+    Example: D:\\Screenshots png      D:\\Screenshots_AVIF avif
+
+    All PNG images here ^  to AVIF images in there ^
+
+    --workers-count will default to your CPU core count.
+
+    --reduced-prio reduces the process priority of the task, to make background processing more pleasant.
+    """
+
+    # Verify that the paths don't end in slash (we add it later)
+    if input_dir[-1] in ['/', '\\']:
+        input_dir = input_dir[0:-1]
+    if output_dir[-1] in ['/', '\\']:
+        output_dir = output_dir[0:-1]
+
+    if not os.path.isdir(input_dir):
+        print("Failed to find input directory! Verify that your arguments are correct")
+        exit()
+
+    # Create the output directory if it doesn't exist
+    if not os.path.isdir(output_dir):
+        try:
+            print(f"Creating output directory: {output_dir}")
+            os.mkdir(output_dir)
+        except Exception as e:
+            print(f"Failed to create output directory: {e}")
+            exit()
+
+    # Reduce prio
+    if reduced_prio:
+        reduce_prio()
+
+    print(f"Starting conversion from [bold]{input_dir} to {output_dir}[/bold]\n")
+
+    # Get list of PNG files
+    files = glob.glob(input_dir + f"/*.{input_format}")
+
+    start_time = time.time()
+    successful_conversions = 0
+    try:
+        with ThreadPoolExecutor(max_workers=workers_count) as executor:
+            futures = {executor.submit(convert_file, file, output_dir, input_format, output_format): file for file in files}
+            for future in as_completed(futures):
+                if future.result():
+                    successful_conversions += 1
+                if interrupted:
+                    executor.shutdown()
+    except Exception as e:
+        print(f"An error occurred: {e}")
+
+    print(f"[green]Successfully converted [bold]{successful_conversions} out of {len(files)}[/bold] files.[green]")
+    print(f"Execution time: {round(time.time() - start_time)}s")
+
+if __name__ == "__main__":
+    print("Welcome to Batch Image Converter (BatImgCon) version 1.0.0")
+    typer.run(main)

requirements.txt

@@ -0,0 +1,12 @@
+click==8.1.7
+colorama==0.4.6
+markdown-it-py==3.0.0
+mdurl==0.1.2
+pillow==10.4.0
+pillow-avif-plugin==1.4.6
+Pygments==2.18.0
+pywin32==306
+rich==13.7.1
+shellingham==1.5.4
+typer==0.12.3
+typing_extensions==4.12.2