mirror of
https://github.com/MarlinFirmware/Marlin.git
synced 2025-02-02 07:00:42 +00:00
100 lines
2.9 KiB
Python
Executable file
100 lines
2.9 KiB
Python
Executable file
#!/usr/bin/env python
|
|
#
|
|
# mc-apply.py
|
|
#
|
|
# Apply firmware configuration from a JSON file (marlin_config.json).
|
|
#
|
|
# usage: mc-apply.py [-h] [--opt] [config_file]
|
|
#
|
|
# Process Marlin firmware configuration.
|
|
#
|
|
# positional arguments:
|
|
# config_file Path to the configuration file.
|
|
#
|
|
# optional arguments:
|
|
# -h, --help show this help message and exit
|
|
# --opt Output as an option setting script.
|
|
#
|
|
import json, sys, os
|
|
import config
|
|
import argparse
|
|
|
|
def report_version(conf):
|
|
if 'VERSION' in conf:
|
|
for k, v in sorted(conf['VERSION'].items()):
|
|
print(k + ': ' + v)
|
|
|
|
def write_opt_file(conf, outpath='Marlin/apply_config.sh'):
|
|
with open(outpath, 'w') as outfile:
|
|
for key, val in conf.items():
|
|
if key in ('__INITIAL_HASH', 'VERSION'): continue
|
|
|
|
# Other keys are assumed to be configs
|
|
if not type(val) is dict:
|
|
continue
|
|
|
|
# Write config commands to the script file
|
|
lines = []
|
|
for k, v in sorted(val.items()):
|
|
if v != '':
|
|
v.replace('"', '\\"').replace("'", "\\'").replace(' ', '\\ ')
|
|
lines += [f'opt_set {k} {v}']
|
|
else:
|
|
lines += [f'opt_enable {k}']
|
|
|
|
outfile.write('\n'.join(lines))
|
|
|
|
print('Config script written to: ' + outpath)
|
|
|
|
def back_up_config(name):
|
|
# Back up the existing file before modifying it
|
|
conf_path = 'Marlin/' + name
|
|
with open(conf_path, 'r') as f:
|
|
# Write a filename.bak#.ext retaining the original extension
|
|
parts = conf_path.split('.')
|
|
nr = ''
|
|
while True:
|
|
bak_path = '.'.join(parts[:-1]) + f'.bak{nr}.' + parts[-1]
|
|
if os.path.exists(bak_path):
|
|
nr = 1 if nr == '' else nr + 1
|
|
continue
|
|
|
|
with open(bak_path, 'w') as b:
|
|
b.writelines(f.readlines())
|
|
break
|
|
|
|
def apply_config(conf):
|
|
for key in conf:
|
|
if key in ('__INITIAL_HASH', 'VERSION'): continue
|
|
|
|
back_up_config(key)
|
|
|
|
for k, v in conf[key].items():
|
|
if v:
|
|
config.set('Marlin/' + key, k, v)
|
|
else:
|
|
config.enable('Marlin/' + key, k)
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description='Process Marlin firmware configuration.')
|
|
parser.add_argument('--opt', action='store_true', help='Output as an option setting script.')
|
|
parser.add_argument('config_file', nargs='?', default='marlin_config.json', help='Path to the configuration file.')
|
|
|
|
args = parser.parse_args()
|
|
|
|
try:
|
|
infile = open(args.config_file, 'r')
|
|
except:
|
|
print(f'No {args.config_file} found.')
|
|
sys.exit(1)
|
|
|
|
conf = json.load(infile)
|
|
report_version(conf)
|
|
|
|
if args.opt:
|
|
write_opt_file(conf)
|
|
else:
|
|
apply_config(conf)
|
|
|
|
if __name__ == '__main__':
|
|
main()
|