@@ -44,7 +44,7 @@ def from_dict(cls, data: Dict[str, Any]) -> "ExecuteStatementResponse":
44
44
error = error ,
45
45
sql_state = status_data .get ("sql_state" ),
46
46
)
47
-
47
+
48
48
# Parse manifest
49
49
manifest = None
50
50
if "manifest" in data :
@@ -59,13 +59,13 @@ def from_dict(cls, data: Dict[str, Any]) -> "ExecuteStatementResponse":
59
59
chunks = manifest_data .get ("chunks" ),
60
60
result_compression = manifest_data .get ("result_compression" ),
61
61
)
62
-
62
+
63
63
# Parse result data
64
64
result = None
65
65
if "result" in data :
66
66
result_data = data ["result" ]
67
67
external_links = None
68
-
68
+
69
69
if "external_links" in result_data :
70
70
external_links = []
71
71
for link_data in result_data ["external_links" ]:
@@ -78,11 +78,13 @@ def from_dict(cls, data: Dict[str, Any]) -> "ExecuteStatementResponse":
78
78
row_count = link_data .get ("row_count" , 0 ),
79
79
row_offset = link_data .get ("row_offset" , 0 ),
80
80
next_chunk_index = link_data .get ("next_chunk_index" ),
81
- next_chunk_internal_link = link_data .get ("next_chunk_internal_link" ),
81
+ next_chunk_internal_link = link_data .get (
82
+ "next_chunk_internal_link"
83
+ ),
82
84
http_headers = link_data .get ("http_headers" ),
83
85
)
84
86
)
85
-
87
+
86
88
result = ResultData (
87
89
data = result_data .get ("data_array" ),
88
90
external_links = external_links ,
@@ -122,7 +124,7 @@ def from_dict(cls, data: Dict[str, Any]) -> "GetStatementResponse":
122
124
error = error ,
123
125
sql_state = status_data .get ("sql_state" ),
124
126
)
125
-
127
+
126
128
# Parse manifest
127
129
manifest = None
128
130
if "manifest" in data :
@@ -137,13 +139,13 @@ def from_dict(cls, data: Dict[str, Any]) -> "GetStatementResponse":
137
139
chunks = manifest_data .get ("chunks" ),
138
140
result_compression = manifest_data .get ("result_compression" ),
139
141
)
140
-
142
+
141
143
# Parse result data
142
144
result = None
143
145
if "result" in data :
144
146
result_data = data ["result" ]
145
147
external_links = None
146
-
148
+
147
149
if "external_links" in result_data :
148
150
external_links = []
149
151
for link_data in result_data ["external_links" ]:
@@ -156,11 +158,13 @@ def from_dict(cls, data: Dict[str, Any]) -> "GetStatementResponse":
156
158
row_count = link_data .get ("row_count" , 0 ),
157
159
row_offset = link_data .get ("row_offset" , 0 ),
158
160
next_chunk_index = link_data .get ("next_chunk_index" ),
159
- next_chunk_internal_link = link_data .get ("next_chunk_internal_link" ),
161
+ next_chunk_internal_link = link_data .get (
162
+ "next_chunk_internal_link"
163
+ ),
160
164
http_headers = link_data .get ("http_headers" ),
161
165
)
162
166
)
163
-
167
+
164
168
result = ResultData (
165
169
data = result_data .get ("data_array" ),
166
170
external_links = external_links ,
@@ -208,11 +212,13 @@ def from_dict(cls, data: Dict[str, Any]) -> "GetChunksResponse":
208
212
row_count = link_data .get ("row_count" , 0 ),
209
213
row_offset = link_data .get ("row_offset" , 0 ),
210
214
next_chunk_index = link_data .get ("next_chunk_index" ),
211
- next_chunk_internal_link = link_data .get ("next_chunk_internal_link" ),
215
+ next_chunk_internal_link = link_data .get (
216
+ "next_chunk_internal_link"
217
+ ),
212
218
http_headers = link_data .get ("http_headers" ),
213
219
)
214
220
)
215
-
221
+
216
222
return cls (
217
223
statement_id = data .get ("statement_id" , "" ),
218
224
external_links = external_links ,
0 commit comments