add circuitpython code

1 files changed, 41 insertions, 0 deletions

diff --git a/circuitpython/tools/insert-usb-ids.py b/circuitpython/tools/insert-usb-ids.py
new file mode 100644
index 0000000..b0843ab
--- /dev/null
+++ b/circuitpython/tools/insert-usb-ids.py
@@ -0,0 +1,41 @@
+# SPDX-FileCopyrightText: 2014 MicroPython & CircuitPython contributors (https://github.com/adafruit/circuitpython/graphs/contributors)
+#
+# SPDX-License-Identifier: MIT
+
+# Reads the USB VID and PID from the file specified by sys.argv[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
+
+config_prefix = "MICROPY_HW_USB_"
+needed_keys = ("USB_PID_CDC_MSC", "USB_PID_CDC_HID", "USB_PID_CDC", "USB_VID")
+
+
+def parse_usb_ids(filename):
+    rv = dict()
+    for line in open(filename).readlines():
+        line = line.rstrip("\r\n")
+        match = re.match("^#define\s+(\w+)\s+\(0x([0-9A-Fa-f]+)\)$", line)
+        if match and match.group(1).startswith(config_prefix):
+            key = match.group(1).replace(config_prefix, "USB_")
+            val = match.group(2)
+            # print("key =", key, "val =", val)
+            if key in needed_keys:
+                rv[key] = val
+    for k in needed_keys:
+        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="")