diff options
Diffstat (limited to 'lib/python/qmk/cli/generate/api.py')
-rwxr-xr-x | lib/python/qmk/cli/generate/api.py | 50 |
1 files changed, 33 insertions, 17 deletions
diff --git a/lib/python/qmk/cli/generate/api.py b/lib/python/qmk/cli/generate/api.py index 6d111f244..9870f7201 100755 --- a/lib/python/qmk/cli/generate/api.py +++ b/lib/python/qmk/cli/generate/api.py | |||
@@ -9,6 +9,7 @@ from milc import cli | |||
9 | from qmk.datetime import current_datetime | 9 | from qmk.datetime import current_datetime |
10 | from qmk.info import info_json | 10 | from qmk.info import info_json |
11 | from qmk.info_json_encoder import InfoJSONEncoder | 11 | from qmk.info_json_encoder import InfoJSONEncoder |
12 | from qmk.json_schema import json_load | ||
12 | from qmk.keyboard import list_keyboards | 13 | from qmk.keyboard import list_keyboards |
13 | 14 | ||
14 | 15 | ||
@@ -18,43 +19,58 @@ def generate_api(cli): | |||
18 | """ | 19 | """ |
19 | api_data_dir = Path('api_data') | 20 | api_data_dir = Path('api_data') |
20 | v1_dir = api_data_dir / 'v1' | 21 | v1_dir = api_data_dir / 'v1' |
21 | keyboard_list = v1_dir / 'keyboard_list.json' | 22 | keyboard_all_file = v1_dir / 'keyboards.json' # A massive JSON containing everything |
22 | keyboard_all = v1_dir / 'keyboards.json' | 23 | keyboard_list_file = v1_dir / 'keyboard_list.json' # A simple list of keyboard targets |
23 | usb_file = v1_dir / 'usb.json' | 24 | keyboard_aliases_file = v1_dir / 'keyboard_aliases.json' # A list of historical keyboard names and their new name |
25 | keyboard_metadata_file = v1_dir / 'keyboard_metadata.json' # All the data configurator/via needs for initialization | ||
26 | usb_file = v1_dir / 'usb.json' # A mapping of USB VID/PID -> keyboard target | ||
24 | 27 | ||
25 | if not api_data_dir.exists(): | 28 | if not api_data_dir.exists(): |
26 | api_data_dir.mkdir() | 29 | api_data_dir.mkdir() |
27 | 30 | ||
28 | kb_all = {'last_updated': current_datetime(), 'keyboards': {}} | 31 | kb_all = {} |
29 | usb_list = {'last_updated': current_datetime(), 'devices': {}} | 32 | usb_list = {} |
30 | 33 | ||
31 | # Generate and write keyboard specific JSON files | 34 | # Generate and write keyboard specific JSON files |
32 | for keyboard_name in list_keyboards(): | 35 | for keyboard_name in list_keyboards(): |
33 | kb_all['keyboards'][keyboard_name] = info_json(keyboard_name) | 36 | kb_all[keyboard_name] = info_json(keyboard_name) |
34 | keyboard_dir = v1_dir / 'keyboards' / keyboard_name | 37 | keyboard_dir = v1_dir / 'keyboards' / keyboard_name |
35 | keyboard_info = keyboard_dir / 'info.json' | 38 | keyboard_info = keyboard_dir / 'info.json' |
36 | keyboard_readme = keyboard_dir / 'readme.md' | 39 | keyboard_readme = keyboard_dir / 'readme.md' |
37 | keyboard_readme_src = Path('keyboards') / keyboard_name / 'readme.md' | 40 | keyboard_readme_src = Path('keyboards') / keyboard_name / 'readme.md' |
38 | 41 | ||
39 | keyboard_dir.mkdir(parents=True, exist_ok=True) | 42 | keyboard_dir.mkdir(parents=True, exist_ok=True) |
40 | keyboard_info.write_text(json.dumps({'last_updated': current_datetime(), 'keyboards': {keyboard_name: kb_all['keyboards'][keyboard_name]}})) | 43 | keyboard_info.write_text(json.dumps({'last_updated': current_datetime(), 'keyboards': {keyboard_name: kb_all[keyboard_name]}})) |
41 | 44 | ||
42 | if keyboard_readme_src.exists(): | 45 | if keyboard_readme_src.exists(): |
43 | copyfile(keyboard_readme_src, keyboard_readme) | 46 | copyfile(keyboard_readme_src, keyboard_readme) |
44 | 47 | ||
45 | if 'usb' in kb_all['keyboards'][keyboard_name]: | 48 | if 'usb' in kb_all[keyboard_name]: |
46 | usb = kb_all['keyboards'][keyboard_name]['usb'] | 49 | usb = kb_all[keyboard_name]['usb'] |
47 | 50 | ||
48 | if 'vid' in usb and usb['vid'] not in usb_list['devices']: | 51 | if 'vid' in usb and usb['vid'] not in usb_list: |
49 | usb_list['devices'][usb['vid']] = {} | 52 | usb_list[usb['vid']] = {} |
50 | 53 | ||
51 | if 'pid' in usb and usb['pid'] not in usb_list['devices'][usb['vid']]: | 54 | if 'pid' in usb and usb['pid'] not in usb_list[usb['vid']]: |
52 | usb_list['devices'][usb['vid']][usb['pid']] = {} | 55 | usb_list[usb['vid']][usb['pid']] = {} |
53 | 56 | ||
54 | if 'vid' in usb and 'pid' in usb: | 57 | if 'vid' in usb and 'pid' in usb: |
55 | usb_list['devices'][usb['vid']][usb['pid']][keyboard_name] = usb | 58 | usb_list[usb['vid']][usb['pid']][keyboard_name] = usb |
56 | 59 | ||
57 | # Write the global JSON files | 60 | # Write the global JSON files |
58 | keyboard_list.write_text(json.dumps({'last_updated': current_datetime(), 'keyboards': sorted(kb_all['keyboards'])}, cls=InfoJSONEncoder)) | 61 | keyboard_all_file.write_text(json.dumps({'last_updated': current_datetime(), 'keyboards': kb_all}, cls=InfoJSONEncoder)) |
59 | keyboard_all.write_text(json.dumps(kb_all, cls=InfoJSONEncoder)) | 62 | usb_file.write_text(json.dumps({'last_updated': current_datetime(), 'usb': usb_list}, cls=InfoJSONEncoder)) |
60 | usb_file.write_text(json.dumps(usb_list, cls=InfoJSONEncoder)) | 63 | |
64 | keyboard_list = sorted(kb_all) | ||
65 | keyboard_list_file.write_text(json.dumps({'last_updated': current_datetime(), 'keyboards': keyboard_list}, cls=InfoJSONEncoder)) | ||
66 | |||
67 | keyboard_aliases = json_load(Path('data/mappings/keyboard_aliases.json')) | ||
68 | keyboard_aliases_file.write_text(json.dumps({'last_updated': current_datetime(), 'keyboard_aliases': keyboard_aliases}, cls=InfoJSONEncoder)) | ||
69 | |||
70 | keyboard_metadata = { | ||
71 | 'last_updated': current_datetime(), | ||
72 | 'keyboards': keyboard_list, | ||
73 | 'keyboard_aliases': keyboard_aliases, | ||
74 | 'usb': usb_list | ||
75 | } | ||
76 | keyboard_metadata_file.write_text(json.dumps(keyboard_metadata, cls=InfoJSONEncoder)) | ||