Script to extract hard-coded VVVVVV scripts into separate .vsc files

extract_vvvvvv_scripts.py

#!/usr/bin/python3
import os
import re
import sys

SCRIPT_REGEX = re.compile('if \(.+?"(.+?)".+?\).+?lines\[\] = {\n(.+?)}', re.MULTILINE | re.DOTALL)
LINE_CONTENT_REGEX = re.compile('^\s{8}(\S)', re.MULTILINE)

if len(sys.argv) < 3:
    print(f'Usage: {sys.argv[0]} <input c++ file> <output directory>')
script_cpp_file, out_dir = sys.argv[1:3]

matches = []
with open(script_cpp_file, 'r') as in_file:
    contents = in_file.read()
    matches = SCRIPT_REGEX.findall(contents)

for match in matches:
    script_name, script_body = match

    # Hack to skip false positive match
    if script_name == 'custom_':
        continue

    # Fix formatting and strip C++ characters
    script_body = script_body.replace('\t', '')
    script_body = LINE_CONTENT_REGEX.sub('\\1', script_body)
    script_body = script_body.replace('",', '')
    script_body = script_body.replace('" ,', '')
    script_body = script_body.replace('"', '')
    script_body = script_body.replace('// ', '# ')
    script_body = script_body.replace('//', '# ')
    script_body = script_body.strip()

    with open(os.path.join(out_dir, f'{script_name}.vsc'), 'w') as out_file:
        print(f'Writing {out_file.name}')
        out_file.write(script_body)