25
25
26
26
prop_type_file_template = """// AUTOGENERATED FILE - DO NOT EDIT
27
27
28
- var PropTypes = window.PropTypes;
29
-
28
+ var pt = window.PropTypes;
29
+ var pk = window['{package_name}'];
30
30
31
31
{components_prop_types}
32
32
"""
33
33
34
- component_prop_types_template = (
35
- "window['{package_name}'].{component_name}.propTypes = {prop_types}"
36
- )
34
+ component_prop_types_template = "pk.{component_name}.propTypes = {prop_types};"
37
35
38
36
39
37
def generate_type (type_name ):
40
38
def wrap (* _ ):
41
- return f"PropTypes .{ type_name } "
39
+ return f"pt .{ type_name } "
42
40
43
41
return wrap
44
42
45
43
46
44
def generate_union (prop_info ):
47
45
types = [generate_prop_type (t ) for t in prop_info ["value" ]]
48
- return f"PropTypes .oneOfType([{ ',' .join (types )} ])"
46
+ return f"pt .oneOfType([{ ',' .join (types )} ])"
49
47
50
48
51
49
def generate_shape (prop_info ):
52
50
props = []
53
51
for key , value in prop_info ["value" ].items ():
54
52
props .append (f"{ key } :{ generate_prop_type (value )} " )
55
53
inner = "{" + "," .join (props ) + "}"
56
- return f"PropTypes .shape({ inner } )"
54
+ return f"pt .shape({ inner } )"
57
55
58
56
59
57
def generate_array_of (prop_info ):
60
58
inner_type = generate_prop_type (prop_info ["value" ])
61
- return f"PropTypes .arrayOf({ inner_type } )"
59
+ return f"pt .arrayOf({ inner_type } )"
62
60
63
61
64
62
def generate_any (* _ ):
65
- return "PropTypes .any"
63
+ return "pt .any"
66
64
67
65
68
66
def generate_enum (prop_info ):
69
67
values = str ([v ["value" ] for v in prop_info ["value" ]])
70
- return f"PropTypes .oneOf({ values } )"
68
+ return f"pt .oneOf({ values } )"
71
69
72
70
73
71
def generate_object_of (prop_info ):
74
- return f"PropTypes .objectOf({ generate_prop_type (prop_info ['value' ])} )"
72
+ return f"pt .objectOf({ generate_prop_type (prop_info ['value' ])} )"
75
73
76
74
77
75
def generate_tuple (* _ ):
78
76
# PropTypes don't have a tuple... just generate an array.
79
- return "PropTypes .array"
77
+ return "pt .array"
80
78
81
79
82
80
prop_types = {
@@ -135,13 +133,13 @@ def generate_prop_types(
135
133
136
134
props = []
137
135
for prop_name , prop_data in data .get ("props" , {}).items ():
138
- props .append (f" { prop_name } :{ generate_prop_type (prop_data ['type' ])} " )
136
+ props .append (f"{ prop_name } :{ generate_prop_type (prop_data ['type' ])} " )
139
137
140
138
patched .append (
141
139
component_prop_types_template .format (
142
140
package_name = package_name ,
143
141
component_name = component_name ,
144
- prop_types = "{" + ",\n " .join (props ) + "}" ,
142
+ prop_types = "{" + ",\n " .join (props ) + "}" ,
145
143
)
146
144
)
147
145
@@ -151,7 +149,7 @@ def generate_prop_types(
151
149
) as f :
152
150
f .write (
153
151
prop_type_file_template .format (
154
- components_prop_types = "\n \n " .join (patched )
152
+ package_name = package_name , components_prop_types = "\n " .join (patched )
155
153
)
156
154
)
157
155
0 commit comments