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
|
#!/usr/bin/python
import ast
import json
from typing import Any, Generator, List
def create_ann_assign(name: str, type_name: str) -> ast.AnnAssign:
return ast.AnnAssign(ast.Name(name), ast.Name(type_name), simple=True)
def create_class(name: str, body: List[ast.stmt]) -> ast.ClassDef:
return ast.ClassDef(name, [ast.Name("TypedDict")], [], body=body, decorator_list=[])
def generate(names: List[str], body: Any) -> Generator[ast.stmt, None, str]:
if isinstance(body, dict):
class_body: List[ast.stmt] = []
for key, value in body.items():
if not isinstance(key, str):
raise Exception("Only string keys are supported")
type_name = yield from generate(names + [key.capitalize()], value)
item = create_ann_assign(key, type_name)
class_body.append(item)
class_name = "".join(names)
yield create_class(class_name, class_body)
return class_name
elif isinstance(body, list):
if body:
first = yield from generate(names, body[0])
return f"List[{first}]"
else:
return "List"
else:
if body is None:
return "None"
else:
return type(body).__name__
def ast_equal(first: ast.stmt, second: ast.stmt):
return ast.unparse(first) == ast.unparse(second)
def generate_from_file(path: str, name: str):
with open(path) as f:
j = json.loads(f.read())
yield from generate([name], j)
types = [
[ast.ImportFrom("typing", [ast.Name("TypedDict"), ast.Name("Union")])],
generate_from_file("mock_data/slack_info_channel_group.json", "ChannelGroup"),
generate_from_file("mock_data/slack_info_channel_private.json", "ChannelPrivate"),
generate_from_file("mock_data/slack_info_channel_public.json", "ChannelPublic"),
generate_from_file("mock_data/slack_info_im.json", "Im"),
generate_from_file("mock_data/slack_info_mpim_channel.json", "MpimChannel"),
generate_from_file("mock_data/slack_info_mpim_group.json", "MpimGroup"),
]
for x in types:
for y in x:
print(ast.unparse(y))
print()
# print('SlackConversationInfoResponse = Union[]')
|