fkg_extract_lines2wiki.py

import glob
import sys

def decode(infilename):
	with open(infilename, 'r', encoding='utf-8') as infile:
		# Added wiki suffix and use the specified directory.
		with open('{0}/{1}_wiki.txt'.format(directory , infilename[2:-4]), 'w', encoding='utf-8') as outfile:
			outfile.write('{| class="wikitable"\n|+{0}\n!Speaker\n!Text\n'.format(infilename[2:-4]))
			for line in infile.readlines():
				# Only extract lines that are messages.
				# The game recognizes these as lines starting with "mess".
				if line.split(',')[0] == "mess":
                                        #Define the speaker and dialogs
					speaker = line.split(',')[1]
					dialog = line.split(',')[2]
					#Convert line breaks and remove actual line breaks caused by untruncated dialogs.
					dialog = dialog.partition('\n')[0].replace('\\n','<br>')
					outfile.write('|-\n|{0}||{1}\n'.format(speaker, dialog))
			outfile.write('|}')
	
	print('Outputted the dialog to {0} formatted for the Wiki (Wiki)'.format(infilename[2:-4]))

def main():
	files = glob.glob('./*.txt')
	directory = 'processed'

	if not os.path.exists(directory):
		os.makedirs(directory)

	for filename in files:
		decode(filename)

if __name__ == '__main__':
	main()