QT/superset2
1
0
Fork 0
mirror of https://github.com/apache/superset.git synced 2026-04-18 07:35:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(explore): export csv data pivoted for Pivot Table [ID-9] (#17512)

Browse Source 
* feat(explore): export csv data pivoted for Pivot Table

* Implement dropdown with download csv options

* Change label to "Original"

* Add tests

* Add form data to query context

* Add form data to query context generator

* Explicitly make form_data optional
This commit is contained in:
Kamil Gabryjelski
2021-12-03 12:42:28 +01:00
committed by GitHub
parent b2ffa268cd
commit 07e8837093
12 changed files with 291 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
tests/common/query_context_generator.py
View File

@@ -16,7 +16,7 @@
# under the License.
import copy
import dataclasses
from typing import Any, Dict, List
from typing import Any, Dict, List, Optional
from superset.common.chart_data import ChartDataResultType
from superset.utils.core import AnnotationType, DTTM_ALIAS, TimeRangeEndpoint
@@ -242,7 +242,9 @@ class QueryContextGenerator:
add_time_offsets: bool = False,
table_id=1,
table_type="table",
form_data: Optional[Dict[str, Any]] = None,
) -> Dict[str, Any]:
form_data = form_data or {}
table_name = query_name.split(":")[0]
table = self.get_table(table_name, table_id, table_type)
return {
@@ -253,7 +255,8 @@ class QueryContextGenerator:
)
],
"result_type": ChartDataResultType.FULL,
"form_data": form_data,
}
def get_table(self, name, id, type):
return Table(id, type, name)
def get_table(self, name, id_, type_):
return Table(id_, type_, name)