mirror of
https://github.com/apache/superset.git
synced 2026-04-07 18:35:15 +00:00
289 lines
10 KiB
Python
289 lines
10 KiB
Python
# Licensed to the Apache Software Foundation (ASF) under one
|
|
# or more contributor license agreements. See the NOTICE file
|
|
# distributed with this work for additional information
|
|
# regarding copyright ownership. The ASF licenses this file
|
|
# to you under the Apache License, Version 2.0 (the
|
|
# "License"); you may not use this file except in compliance
|
|
# with the License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing,
|
|
# software distributed under the License is distributed on an
|
|
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
# KIND, either express or implied. See the License for the
|
|
# specific language governing permissions and limitations
|
|
# under the License.
|
|
import logging
|
|
|
|
from flask import request, Response
|
|
from flask_appbuilder.api import expose, protect, safe
|
|
from marshmallow import ValidationError
|
|
|
|
from superset.commands.explore.form_data.create import CreateFormDataCommand
|
|
from superset.commands.explore.form_data.delete import DeleteFormDataCommand
|
|
from superset.commands.explore.form_data.get import GetFormDataCommand
|
|
from superset.commands.explore.form_data.parameters import CommandParameters
|
|
from superset.commands.explore.form_data.update import UpdateFormDataCommand
|
|
from superset.commands.temporary_cache.exceptions import (
|
|
TemporaryCacheAccessDeniedError,
|
|
TemporaryCacheResourceNotFoundError,
|
|
)
|
|
from superset.constants import MODEL_API_RW_METHOD_PERMISSION_MAP
|
|
from superset.explore.form_data.schemas import FormDataPostSchema, FormDataPutSchema
|
|
from superset.extensions import event_logger
|
|
from superset.views.base_api import BaseSupersetApi, requires_json, statsd_metrics
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class ExploreFormDataRestApi(BaseSupersetApi):
|
|
add_model_schema = FormDataPostSchema()
|
|
edit_model_schema = FormDataPutSchema()
|
|
method_permission_name = MODEL_API_RW_METHOD_PERMISSION_MAP
|
|
allow_browser_login = True
|
|
class_permission_name = "ExploreFormDataRestApi"
|
|
resource_name = "explore"
|
|
openapi_spec_tag = "Explore Form Data"
|
|
openapi_spec_component_schemas = (FormDataPostSchema, FormDataPutSchema)
|
|
|
|
@expose("/form_data", methods=("POST",))
|
|
@protect()
|
|
@safe
|
|
@statsd_metrics
|
|
@event_logger.log_this_with_context(
|
|
action=lambda self, *args, **kwargs: f"{self.__class__.__name__}.post",
|
|
log_to_statsd=False,
|
|
)
|
|
@requires_json
|
|
def post(self) -> Response:
|
|
"""Create a new form_data.
|
|
---
|
|
post:
|
|
summary: Create a new form_data
|
|
parameters:
|
|
- in: query
|
|
schema:
|
|
type: integer
|
|
name: tab_id
|
|
requestBody:
|
|
required: true
|
|
content:
|
|
application/json:
|
|
schema:
|
|
$ref: '#/components/schemas/FormDataPostSchema'
|
|
responses:
|
|
201:
|
|
description: The form_data was stored successfully.
|
|
content:
|
|
application/json:
|
|
schema:
|
|
type: object
|
|
properties:
|
|
key:
|
|
type: string
|
|
description: The key to retrieve the form_data.
|
|
400:
|
|
$ref: '#/components/responses/400'
|
|
401:
|
|
$ref: '#/components/responses/401'
|
|
422:
|
|
$ref: '#/components/responses/422'
|
|
500:
|
|
$ref: '#/components/responses/500'
|
|
"""
|
|
try:
|
|
item = self.add_model_schema.load(request.json)
|
|
tab_id = request.args.get("tab_id")
|
|
args = CommandParameters(
|
|
datasource_id=item["datasource_id"],
|
|
datasource_type=item["datasource_type"],
|
|
chart_id=item.get("chart_id"),
|
|
tab_id=tab_id,
|
|
form_data=item["form_data"],
|
|
)
|
|
key = CreateFormDataCommand(args).run()
|
|
return self.response(201, key=key)
|
|
except ValidationError as ex:
|
|
return self.response(400, message=ex.messages)
|
|
except TemporaryCacheAccessDeniedError as ex:
|
|
return self.response(403, message=str(ex))
|
|
except TemporaryCacheResourceNotFoundError as ex:
|
|
return self.response(404, message=str(ex))
|
|
|
|
@expose("/form_data/<string:key>", methods=("PUT",))
|
|
@protect()
|
|
@safe
|
|
@statsd_metrics
|
|
@event_logger.log_this_with_context(
|
|
action=lambda self, *args, **kwargs: f"{self.__class__.__name__}.put",
|
|
log_to_statsd=True,
|
|
)
|
|
@requires_json
|
|
def put(self, key: str) -> Response:
|
|
"""Update an existing form_data.
|
|
---
|
|
put:
|
|
summary: Update an existing form_data
|
|
parameters:
|
|
- in: path
|
|
schema:
|
|
type: string
|
|
name: key
|
|
- in: query
|
|
schema:
|
|
type: integer
|
|
name: tab_id
|
|
requestBody:
|
|
required: true
|
|
content:
|
|
application/json:
|
|
schema:
|
|
$ref: '#/components/schemas/FormDataPutSchema'
|
|
responses:
|
|
200:
|
|
description: The form_data was stored successfully.
|
|
content:
|
|
application/json:
|
|
schema:
|
|
type: object
|
|
properties:
|
|
key:
|
|
type: string
|
|
description: The key to retrieve the form_data.
|
|
400:
|
|
$ref: '#/components/responses/400'
|
|
401:
|
|
$ref: '#/components/responses/401'
|
|
404:
|
|
$ref: '#/components/responses/404'
|
|
422:
|
|
$ref: '#/components/responses/422'
|
|
500:
|
|
$ref: '#/components/responses/500'
|
|
"""
|
|
try:
|
|
item = self.edit_model_schema.load(request.json)
|
|
tab_id = request.args.get("tab_id")
|
|
args = CommandParameters(
|
|
datasource_id=item["datasource_id"],
|
|
datasource_type=item["datasource_type"],
|
|
chart_id=item.get("chart_id"),
|
|
tab_id=tab_id,
|
|
key=key,
|
|
form_data=item["form_data"],
|
|
)
|
|
result = UpdateFormDataCommand(args).run()
|
|
if not result:
|
|
return self.response_404()
|
|
return self.response(200, key=result)
|
|
except ValidationError as ex:
|
|
return self.response(400, message=ex.messages)
|
|
except TemporaryCacheAccessDeniedError as ex:
|
|
return self.response(403, message=str(ex))
|
|
except TemporaryCacheResourceNotFoundError as ex:
|
|
return self.response(404, message=str(ex))
|
|
|
|
@expose("/form_data/<string:key>", methods=("GET",))
|
|
@protect()
|
|
@safe
|
|
@statsd_metrics
|
|
@event_logger.log_this_with_context(
|
|
action=lambda self, *args, **kwargs: f"{self.__class__.__name__}.get",
|
|
log_to_statsd=True,
|
|
)
|
|
def get(self, key: str) -> Response:
|
|
"""Get a form_data.
|
|
---
|
|
get:
|
|
summary: Get a form_data
|
|
parameters:
|
|
- in: path
|
|
schema:
|
|
type: string
|
|
name: key
|
|
responses:
|
|
200:
|
|
description: Returns the stored form_data.
|
|
content:
|
|
application/json:
|
|
schema:
|
|
type: object
|
|
properties:
|
|
form_data:
|
|
type: string
|
|
description: The stored form_data
|
|
400:
|
|
$ref: '#/components/responses/400'
|
|
401:
|
|
$ref: '#/components/responses/401'
|
|
404:
|
|
$ref: '#/components/responses/404'
|
|
422:
|
|
$ref: '#/components/responses/422'
|
|
500:
|
|
$ref: '#/components/responses/500'
|
|
"""
|
|
try:
|
|
args = CommandParameters(key=key)
|
|
form_data = GetFormDataCommand(args).run()
|
|
if not form_data:
|
|
return self.response_404()
|
|
return self.response(200, form_data=form_data)
|
|
except TemporaryCacheAccessDeniedError as ex:
|
|
return self.response(403, message=str(ex))
|
|
except TemporaryCacheResourceNotFoundError as ex:
|
|
return self.response(404, message=str(ex))
|
|
|
|
@expose("/form_data/<string:key>", methods=("DELETE",))
|
|
@protect()
|
|
@safe
|
|
@statsd_metrics
|
|
@event_logger.log_this_with_context(
|
|
action=lambda self, *args, **kwargs: f"{self.__class__.__name__}.delete",
|
|
log_to_statsd=True,
|
|
)
|
|
def delete(self, key: str) -> Response:
|
|
"""Delete a form_data.
|
|
---
|
|
delete:
|
|
summary: Delete a form_data
|
|
parameters:
|
|
- in: path
|
|
schema:
|
|
type: string
|
|
name: key
|
|
description: The form_data key.
|
|
responses:
|
|
200:
|
|
description: Deleted the stored form_data.
|
|
content:
|
|
application/json:
|
|
schema:
|
|
type: object
|
|
properties:
|
|
message:
|
|
type: string
|
|
description: The result of the operation
|
|
400:
|
|
$ref: '#/components/responses/400'
|
|
401:
|
|
$ref: '#/components/responses/401'
|
|
404:
|
|
$ref: '#/components/responses/404'
|
|
422:
|
|
$ref: '#/components/responses/422'
|
|
500:
|
|
$ref: '#/components/responses/500'
|
|
"""
|
|
try:
|
|
args = CommandParameters(key=key)
|
|
result = DeleteFormDataCommand(args).run()
|
|
if not result:
|
|
return self.response_404()
|
|
return self.response(200, message="Deleted successfully")
|
|
except TemporaryCacheAccessDeniedError as ex:
|
|
return self.response(403, message=str(ex))
|
|
except TemporaryCacheResourceNotFoundError as ex:
|
|
return self.response(404, message=str(ex))
|