Script for extracting an SWF file from a projector executable

swfextract_kun.py

#!/usr/bin/env python3
"""Extracts an SWF file from a projector executable"""

import os
import sys


def strip_exe_extension(path):
    root, ext = os.path.splitext(path)
    if ext.casefold() == '.exe':
        return root
    else:
        return path


def main(path):
    content = open(path, 'rb').read()
    index = content.find(b'FWS')
    if index >= 0:
        output_path = strip_exe_extension(path) + '.swf'
        with open(output_path, 'wb') as output:
            output.write(content[index:])
    else:
        return 'could not find embedded Flash movie'


if __name__ == '__main__':
    if len(sys.argv) != 2:
        sys.exit('Usage: {} FILE.EXE'.format(sys.argv[0]))
    else:
        sys.exit(main(sys.argv[1]))