fix: return appropriate response when payload has error (#12575)

This commit is contained in:
Yongjie Zhao 2021-01-19 14:57:25 +08:00 committed by GitHub
parent ad63823d6f
commit e7a5d0d01a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 3 deletions

View File

@ -421,9 +421,10 @@ class Superset(BaseSupersetView): # pylint: disable=too-many-public-methods
)
def get_raw_results(self, viz_obj: BaseViz) -> FlaskResponse:
return self.json_response(
{"data": viz_obj.get_df_payload()["df"].to_dict("records")}
)
payload = viz_obj.get_df_payload()
if viz_obj.has_error(payload):
return json_error_response(payload=payload, status=400)
return self.json_response({"data": payload["df"].to_dict("records")})
def get_samples(self, viz_obj: BaseViz) -> FlaskResponse:
return self.json_response({"data": viz_obj.get_samples()})