marlin/buildroot/share/PlatformIO/scripts/mc-apply.py

#!/usr/bin/env python
#
# Create a Configuration from marlin_config.json
#
import json
import sys
import shutil
import re

opt_output = '--opt' in sys.argv
output_suffix = '.sh' if opt_output else '' if '--bare-output' in sys.argv else '.gen'

try:
	with open('marlin_config.json', 'r') as infile:
		conf = json.load(infile)
		for key in conf:
			# We don't care about the hash when restoring here
			if key == '__INITIAL_HASH':
				continue
			if key == 'VERSION':
				for k, v in sorted(conf[key].items()):
					print(k + ': ' + v)
				continue
			# The key is the file name, so let's build it now
			outfile = open('Marlin/' + key + output_suffix, 'w')
			for k, v in sorted(conf[key].items()):
				# Make define line now
				if opt_output:
					if v != '':
						if '"' in v:
							v = "'%s'" % v
						elif ' ' in v:
							v = '"%s"' % v
						define = 'opt_set ' + k + ' ' + v + '\n'
					else:
						define = 'opt_enable ' + k + '\n'
				else:
					define = '#define ' + k + ' ' + v + '\n'
				outfile.write(define)
			outfile.close()

			# Try to apply changes to the actual configuration file (in order to keep useful comments)
			if output_suffix != '':
				# Move the existing configuration so it doesn't interfere
				shutil.move('Marlin/' + key, 'Marlin/' + key + '.orig')
				infile_lines = open('Marlin/' + key + '.orig', 'r').read().split('\n')
				outfile = open('Marlin/' + key, 'w')
				for line in infile_lines:
					sline = line.strip(" \t\n\r")
					if sline[:7] == "#define":
						# Extract the key here (we don't care about the value)
						kv = sline[8:].strip().split(' ')
						if kv[0] in conf[key]:
							outfile.write('#define ' + kv[0] + ' ' + conf[key][kv[0]] + '\n')
							# Remove the key from the dict, so we can still write all missing keys at the end of the file
							del conf[key][kv[0]]
						else:
							outfile.write(line + '\n')
					else:
						outfile.write(line + '\n')
				# Process any remaining defines here
				for k, v in sorted(conf[key].items()):
					define = '#define ' + k + ' ' + v + '\n'
					outfile.write(define)
				outfile.close()

			print('Output configuration written to: ' + 'Marlin/' + key + output_suffix)
except:
	print('No marlin_config.json found.')