Improve turing-theme-extractor.py: create subfolders for extracted PNGs

Author: mathoudebine

tools/turing-theme-extractor.py

@@ -22,9 +22,12 @@
 # turing-theme-extractor.py: Extract resources from a Turing Smart Screen theme (.data files) made for Windows app
 # This program will search and extract PNGs from the theme data and extract theme in the current directory
 # The PNG can then be re-used to create a theme for System Monitor python program (see Wiki for theme creation)
+# To run for all files of a folder:
+# for file in "FOLDER"/*; do if [ -f "$file" ]; then python turing-theme-extractor.py "$file"; fi; done
 import mmap
 import os
 import sys
+from pathlib import Path
 
 PNG_SIGNATURE = b'\x89\x50\x4E\x47\x0D\x0A\x1A\x0A'
 PNG_IEND = b'\x49\x45\x4E\x44\xAE\x42\x60\x82'
@@ -46,6 +49,11 @@
 with open(sys.argv[1], "r+b") as theme_file:
     mm = mmap.mmap(theme_file.fileno(), 0)
 
+    theme_name = Path(sys.argv[1]).stem
+
+    if not os.path.isdir(theme_name):
+        os.makedirs(theme_name)
+
     # Find PNG signature in binary data
     start_pos=0
     header_found = mm.find(PNG_SIGNATURE, 0)
@@ -59,15 +67,14 @@
 
         # Extract PNG data to a file
         theme_file.seek(header_found)
-        png_file = open('theme_res_' + str(header_found) + '.png', 'wb')
+        png_file = open(theme_name + "/" + str(header_found) + '.png', 'wb')
         png_file.write(theme_file.read(iend_found - header_found + len(PNG_IEND)))
         png_file.close()
 
-        print("PNG extracted to theme_res_%s.png" % str(header_found))
+        print("PNG extracted to %s/%s.png" % (theme_name, str(header_found)))
         found_png = found_png + 1
 
         # Find next PNG signature (if any)
         header_found = mm.find(PNG_SIGNATURE, iend_found)
 
     print("\n%d PNG files extracted from theme to current directory" % found_png)
-