genScripts.py

#!/usr/bin/env python3
import shlex, PKSMScript, sys, glob, shutil, os

games = ["LGPE", "USUM", "SM", "ORAS", "XY", "B2W2", "BW", "HGSS", "PT", "DP"]

def main(args):
	print("Generating scripts...")
	shutil.rmtree("scripts", True)
	shutil.rmtree("build", True)
	os.mkdir("scripts")
	if os.path.exists("src/universal"):
		# shutil.copytree("src/universal", "scripts/universal")
		cleanCScripts("src/universal")
	for game in games:
		generate(game)

		if os.path.exists("src/" + game.lower()):
			cleanCScripts("src/" + game.lower())
		else:
			os.mkdir("scripts/" + game.lower())

		if os.path.exists("build"):
			for f in os.listdir("build"):
				shutil.move("build/" + f, "scripts/" + game.lower())
		scriptFiles = glob.glob("*.pksm")
		for pksmFile in scriptFiles:
			shutil.move(pksmFile, "scripts/" + game.lower())

def generate(game):
	with open(os.path.join("src", "scripts%s.txt" % game)) as pksmArgFile:
		for line in pksmArgFile:
			if not line.startswith('#'):
				line.replace('\\', '/')
				pksmArgs = PKSMScript.parser.parse_args(shlex.split(line))
				PKSMScript.main(pksmArgs)

def cleanCScripts(folder):
	os.mkdir(folder.replace("src/", "scripts/"))
	for path, _, files in os.walk(folder):
		for fullname in files:
			data = ""
			with open(os.path.join(path, fullname), 'r', encoding="UTF-8") as f:
				data = f.read()

			data = data.replace("\r\n", "\n")

			# Get rid of all comments
			commentIndex = data.find("/*")
			while commentIndex != -1:
				data = data[:commentIndex] + data[data.find("*/")+2:]
				commentIndex = data.find("/*")

			commentIndex = data.find("//")
			while commentIndex != -1:
				data = data[:commentIndex] + \
					data[data[commentIndex+2:].find("\n")+1 + commentIndex+2:]
				commentIndex = data.find("//")

			with open(os.path.join(path.replace("src/", "scripts/"), fullname), 'w', encoding="UTF-8") as f:
				f.write(data)

if __name__ == '__main__':
	main(sys.argv)