extract.py

import struct
import magic
from PIL import Image
import sys

img = Image.open(open(sys.argv[1], "rb"))
img_data = b""
for x in range(img.width):
    if len(img_data) >= 2 and not img_data.startswith(b"MZ"):
         print("Not a valid PE file")
         break
    for y in range(img.height):
        r, g, b, *_ = img.getpixel((x, y))
        img_data += struct.pack("<BBB", r, g, b)

if (file_type := magic.from_buffer(img_data)).startswith("PE32"):
   print("Extracted stage2:", file_type)