forked from danger/python
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmodels.py
More file actions
100 lines (69 loc) · 2.24 KB
/
Copy pathmodels.py
File metadata and controls
100 lines (69 loc) · 2.24 KB
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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
from dataclasses import InitVar, dataclass, field
from typing import Any, List, Optional, Set, Tuple
@dataclass
class SchemaItem:
name: str
@dataclass
class SchemaObject(SchemaItem):
properties: List["SchemaItem"] = field(default_factory=list)
@dataclass
class SchemaValue(SchemaItem):
value_types: List[str] = field(default_factory=list)
value_type: InitVar[str] = None
def __post_init__(self, value_type: Optional[str]):
if value_type:
self.value_types = [value_type]
@dataclass
class SchemaReference(SchemaItem):
reference: str
@dataclass
class SchemaEnum(SchemaValue):
values: List[Any] = field(default_factory=list)
@dataclass
class SchemaArray(SchemaItem):
item: SchemaItem
@dataclass
class SchemaAllOf(SchemaItem):
all_of: List[SchemaItem] = field(default_factory=list)
@dataclass
class SchemaAnyOf(SchemaItem):
any_of: List[SchemaItem] = field(default_factory=list)
@dataclass
class PropertyDefinition:
name: str
key: str
value_type: str
known_type: bool
@property
def formatted_type(self) -> str:
prefixes = {"List[", "Optional["}
prefix = next(filter(lambda p: self.value_type.startswith(p), prefixes), None)
if self.known_type:
return self.value_type
elif prefix:
stripped = self.value_type[len(prefix) : -1]
return f"{prefix}'{stripped}']"
else:
return f"'{self.value_type}'"
@property
def non_reserved_name(self) -> str:
reserved_names = {"from", "self"}
return f"{self.name}_" if self.name in reserved_names else self.name
@property
def patched_type(self) -> str:
formatted_type = self.formatted_type
if formatted_type.startswith("Optional["):
return formatted_type
if formatted_type == "Any":
return formatted_type
return f"Optional[{formatted_type}]"
@dataclass
class TypeDefinition:
name: str
depends_on: Set[str] = field(default_factory=set)
@dataclass
class ClassDefinition(TypeDefinition):
properties: List[PropertyDefinition] = field(default_factory=list)
@dataclass
class EnumDefinition(TypeDefinition):
values: List[Tuple[str, str]] = field(default_factory=list)