QMK/lib/chibios-contrib/tools/update_configs.py

110 lines
3.6 KiB
Python
Raw Normal View History

2020-03-23 10:48:11 +01:00
#!/usr/bin/python3
# -*- coding: utf-8 -*-
__author__ = 'Fabien Poussin'
__version__ = '0.1'
import os
import re
from argparse import ArgumentParser
from traceback import print_exc
from shutil import copy
parser = ArgumentParser(description='Generate ChibiOS-Contrib config and Makefiles from ChibiOS')
parser.add_argument('-s', '--src', default='../../ChibiOS', type=str, help="ChibiOS folder")
parser.add_argument('-d', '--dst', default='..', type=str, help='ChibiOS-Contrib folder')
FOLDERS = ['testhal']
def makefile(lines):
for l in range(len(lines)):
if 'CHIBIOS =' in lines[l]:
lines[l] = lines[l][:-1] + '/../ChibiOS\n'
lines.insert(l + 1, 'CHIBIOS_CONTRIB = $(CHIBIOS)/../ChibiOS-Contrib\n')
if '$(CHIBIOS)/os/hal/hal.mk' in lines[l] \
or '$(CHIBIOS)/os/hal/ports/' in lines[l]:
lines[l] = lines[l].replace('CHIBIOS', 'CHIBIOS_CONTRIB')
return "".join(lines)
def halconf(lines):
idx_end = lines.index('#endif /* HALCONF_H */\n')
lines.insert(idx_end - 1, '\n')
lines.insert(idx_end - 1, '#include "halconf_community.h"')
lines.insert(idx_end - 1, '\n')
return "".join(lines)
def mcuconf(lines):
idx_end = lines.index('#endif /* MCUCONF_H */\n')
lines.insert(idx_end - 1, '\n')
lines.insert(idx_end - 1, '#include "mcuconf_community.h"')
lines.insert(idx_end - 1, '\n')
return "".join(lines)
if __name__ == '__main__':
args = parser.parse_args()
sources = {}
for folder in FOLDERS:
for family in os.scandir(args.src + '/{}/STM32/'.format(folder)):
if not family.name[0].isupper() or not family.is_dir():
continue
for test in os.scandir(family.path):
try:
sources[family.name] = {'makefile': None, 'halconf': None, 'mcuconf': None, 'chconf': None}
with open(test.path + '/Makefile', 'r') as file:
sources[family.name]['makefile'] = makefile(file.readlines())
with open(test.path + '/chconf.h', 'r') as file:
sources[family.name]['chconf'] = file.read()
with open(test.path + '/halconf.h', 'r') as file:
sources[family.name]['halconf'] = halconf(file.readlines())
with open(test.path + '/mcuconf.h', 'r') as file:
sources[family.name]['mcuconf'] = mcuconf(file.readlines())
except Exception as e:
print(test.path, e)
del sources[family.name]
continue
break
for family in os.scandir(args.dst + '/{}/STM32/'.format(folder)):
if not family.name[0].isupper() or not family.is_dir():
continue
for test in os.scandir(family.path):
copy('templates/halconf_community.h', test.path)
copy('templates/mcuconf_community.h', test.path)
try:
with open(test.path + '/Makefile', 'w') as file:
file.write(sources[family.name]['makefile'])
with open(test.path + '/chconf.h', 'w') as file:
file.write(sources[family.name]['chconf'])
with open(test.path + '/halconf.h', 'w') as file:
file.write(sources[family.name]['halconf'])
with open(test.path + '/mcuconf.h', 'w') as file:
file.write(sources[family.name]['mcuconf'])
print('updated', test.path)
except KeyError as e:
print('Missing family data', e)