blob: a32429fae5f90f0fd0a45ceec12c80fadf927132 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
#!/usr/bin/env python3
import argparse
import sys
import json
indent_level=4
def parse_cli():
parser = argparse.ArgumentParser(description='Ferris keymap formatter')
parser.add_argument("--input", type=argparse.FileType('r'),
default=sys.stdin, help="Input keymap (json file produced by qmk configurator)")
return parser.parse_args()
def col_index(key_index):
if key_index < 30:
col = key_index % 5
else:
col = key_index % 2
return col
def format_layers(layers):
formatted = indent_level * " " + "\"layers\": [\n"
max_key_length = {}
for layer in layers:
for (index, keycode) in enumerate(layer):
col = col_index(index)
max_length = max_key_length.get(col)
if (not max_length) or len(keycode) > max_length:
max_key_length.update({col: len(keycode)})
for (layer_index, layer) in enumerate(layers):
formatted += 2 * indent_level * " "
formatted += "["
for (index, keycode) in enumerate(layer):
if index > 30 and index % 2 == 0 or index % 5 == 0 and index != 0:
formatted += (1 + 2 * indent_level) * " "
formatted += json.dumps(keycode)
if index == 33:
formatted += "\n"
elif index > 30 and index % 2 == 1 or index % 5 == 4:
formatted += ",\n"
else:
n_spaces = max_key_length[col_index(index)] - len(keycode)
formatted += n_spaces * " "
formatted += ", "
if index % 10 == 9:
formatted += "\n"
formatted += 2 * indent_level * " "
if layer_index < len(layers) - 1:
formatted += "],\n"
else:
formatted += "]\n"
formatted += indent_level * " "
formatted += "]"
return formatted
def format_keymap(keymap_json):
formatted = "{"
for (index, k) in enumerate(keymap_json):
if k == "layers":
formatted += format_layers(keymap_json[k])
else:
formatted += f"{indent_level * ' '}{json.dumps(k)}: {json.dumps(keymap_json[k])}"
if index < len(keymap_json) - 1:
formatted += ","
formatted += "\n"
formatted += "}"
return formatted
def main():
args=parse_cli()
keymap_json = json.loads(args.input.read())
print(format_keymap(keymap_json))
main()
|
