37 lines
1.3 KiB
Python
37 lines
1.3 KiB
Python
|
# 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='')
|