import os import toml from glob import glob abspath = os.path.abspath(__file__) dname = os.path.dirname(abspath) os.chdir(dname) supported_families = [ 'STM32F4', 'STM32L0', 'STM32L4', 'STM32H7', 'STM32WB55', ] # ======= load chip list features = {} for f in sorted(glob('../stm32-data/data/chips/*.yaml')): # Use the filename to get the chip name. Ultra fast, we don't have to read YAML! name = os.path.splitext(os.path.basename(f))[0] if any((family in name for family in supported_families)): name = name.lower() features[name] = ['stm32-metapac/' + name] # ========= Update Cargo features SEPARATOR_START = '# BEGIN GENERATED FEATURES\n' SEPARATOR_END = '# END GENERATED FEATURES\n' HELP = '# Generated by gen_features.py. DO NOT EDIT.\n' with open('Cargo.toml', 'r') as f: cargo = f.read() before, cargo = cargo.split(SEPARATOR_START, maxsplit=1) _, after = cargo.split(SEPARATOR_END, maxsplit=1) cargo = before + SEPARATOR_START + HELP + toml.dumps(features) + SEPARATOR_END + after with open('Cargo.toml', 'w') as f: f.write(cargo)