1 files changed, 36 insertions, 0 deletions
diff --git a/tools/insert-usb-ids.py b/tools/insert-usb-ids.py
new file mode 100644
index 0000000000..af010dad95
--- /dev/null
+++ b/tools/insert-usb-ids.py
@@ -0,0 +1,36 @@
+# Reads the USB VID and PID from the file specifed by sys.arg[1] and then
+# inserts those values into the template file specified by sys.argv[2],
+# printing the result to stdout
+
+from __future__ import print_function
+
+import sys
+import re
+import string
+
+def parse_usb_ids(filename):
+    rv = dict()
+    if filename == 'usbd_desc_cdc_msc.c':
+        for line in open(filename).readlines():
+            line = line.rstrip('\r\n')
+            match = re.match('^#define\s+(\w+)\s+0x(\d+)$', line)
+            if match:
+                if match.group(1) == 'USBD_VID':
+                    rv['USB_VID'] = match.group(2)
+                elif match.group(1) == 'USBD_PID':
+                    rv['USB_PID'] = match.group(2)
+                if 'USB_VID' in rv and 'USB_PID' in rv:
+                    break
+    else:
+        raise Exception("Don't (yet) know how to parse USB IDs from %s" % filename)
+    for k in ('USB_PID', 'USB_VID'):
+        if k not in rv:
+            raise Exception("Unable to parse %s from %s" % (k, filename))
+    return rv
+
+if __name__ == "__main__":
+    usb_ids_file = sys.argv[1]
+    template_file = sys.argv[2]
+    replacements = parse_usb_ids(usb_ids_file)
+    for line in open(template_file, 'r').readlines():
+        print(string.Template(line).safe_substitute(replacements), end='')