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
77
78
79
80
81
82
83
|
#!/usr/bin/env python3
# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
import argparse
import json
import re
import sys
import textwrap
import requests
def to_underscores(item):
words = re.findall(r'[A-Z](?:[a-z0-9]+|[A-Z0-9]*(?=[A-Z]|$))', item)
return '_'.join(w.upper() for w in words)
def main():
parser = argparse.ArgumentParser()
parser.add_argument("url", help="URL of a DMTF definition")
parser.add_argument("name", help="name of the enumeration")
parser.add_argument("--compat", help="generate old-style constants",
action="store_true")
args = parser.parse_args()
if args.url.startswith("https://") or args.url.startswith("http://"):
resp = requests.get(args.url, timeout=30)
resp.raise_for_status()
content = resp.json()
else:
with open(args.url, "t") as fp:
content = json.load(fp)
try:
definition = content["definitions"][args.name]
except KeyError as exc:
sys.exit(f"Key {exc} was not found in definition at {args.url}")
try:
items = definition["enum"]
descriptions = definition.get("enumDescriptions", {})
except (TypeError, KeyError):
sys.exit(f"Definition {args.name} is malformed or not en enumeration")
items = [(to_underscores(item), item) for item in items]
print(f"class {args.name}(enum.Enum):")
for varname, item in items:
print(f" {varname} = '{item}'")
try:
description = descriptions[item]
except KeyError:
pass
else:
# 79 (expected) - 4 (indentation) - 2 * 3 (quotes) = 69
lines = textwrap.wrap(description, 69)
lines[0] = '"""' + lines[0]
lines[-1] += '"""'
for line in lines:
print(f' {line}')
print()
if args.compat:
print()
print("# Backward compatibility")
for varname, item in items:
print(f"{to_underscores(args.name)}_{varname} = "
f"{args.name}.{varname}")
if __name__ == '__main__':
sys.exit(main())
|
