gen_rel_device.py

import xml.etree.ElementTree as ET
import os
import argparse
import shutil

def update_xml_file (xml_filepath, rel_devices):
    safe_dirs = []
    # parse the XML file
    tree = ET.parse(xml_filepath)
    root = tree.getroot()
    # iterate through all the devices and remove those that are not rel_devices 
    for device in root.findall('device'):
        if device.get('name') in rel_devices:
            for internal in device.findall('internal'):
                if internal.attrib.get('type') == 'vpr_arch':
                    value = internal.get('file')
                    if value is not None:
                        safe_dirs.append(value.split('/')[1])
        else:
            root.remove(device)

    # write the updated XML file
    tree.write(xml_filepath)
    return safe_dirs

def remove_dirs (devices_dir,not_to_remove):
    # Get the list of devices directories
    dirs = [name for name in os.listdir(devices_dir) if os.path.isdir(os.path.join(devices_dir, name))]
    # Remove all directories except the one to keep
    for dir in dirs:
        if os.path.normpath(dir) not in not_to_remove:
            try:
                shutil.rmtree(os.path.join(devices_dir, dir))
                print(f"Removed directory {dir}")
            except OSError as e:
                print(f"Error removing directory {dir}: {e}")

def update_examples (examples_top_dir,device_name):
    ospr_files = []
    for root, _, files in os.walk(examples_top_dir):
        for file_ in files:
            if file_.endswith(".ospr"):
                ospr_files.append(os.path.join(root, file_))
    print("examples to be updated are:\n\t",ospr_files)
    print("Device name that will be updated in examples is: ",device_name)
    for example in ospr_files:
        tree = ET.parse(example)
        root = tree.getroot()

        # Find the specific 'Option' element with Name="Device" and update its 'Val' attribute
        for option in root.findall('.//Option[@Name="Device"]'):
            option.set('Val', device_name)

        # Write the updated XML data back to a file
        with open(example, 'wb') as f:
            tree.write(f, encoding='utf-8', xml_declaration=True)

def main ():
    parser = argparse.ArgumentParser()
    parser.add_argument('--production_devices',
                        type=str,default='',metavar='',nargs='+',required=True,help='One or list of devices for production build', dest="rel_devices_list")
    parser.add_argument('--xml_filepath',
                        type=str,default='',metavar='',required=True,help='Path for the device.xml which will be updated', dest="XML_filepath")
    parser.add_argument('--devices_dirs_path',
                        type=str,default='1',metavar='',help='Path of devices directories', dest="devices_dirs")
    parser.add_argument('--examples_path',
                        type=str,default='1',metavar='',help='Path of devices directories', dest="examples")
    args = parser.parse_args()
    args.rel_devices_list = [device.strip() for device in args.rel_devices_list[0].split(',')]
    print("Given Production Devices are:\n\t\t",args.rel_devices_list)
    print("XML file that will be update:\n\t\t",args.XML_filepath)
    print("directory from which not required devices directories will be removed:\n\t\t",args.devices_dirs)
    save_dirs = update_xml_file(args.XML_filepath,args.rel_devices_list)
    remove_dirs(args.devices_dirs,save_dirs)
    update_examples(args.examples,args.rel_devices_list[0])

if __name__ == "__main__":
    main()