mirror of
https://github.com/apache/superset.git
synced 2026-06-02 22:29:26 +00:00
7b3095d6ff
* Fix examples charts/dashboards and refactor * pylinting * Fix pylint * Lint the refactor * Rebased
117 lines
3.2 KiB
Python
117 lines
3.2 KiB
Python
"""Loads datasets, dashboards and slices in a new superset instance"""
|
|
# pylint: disable=C,R,W
|
|
import gzip
|
|
import os
|
|
import textwrap
|
|
|
|
import pandas as pd
|
|
from sqlalchemy import Float, String
|
|
|
|
from superset import db
|
|
from superset.utils import core as utils
|
|
from .helpers import DATA_FOLDER, merge_slice, misc_dash_slices, Slice, TBL
|
|
|
|
|
|
def load_energy():
|
|
"""Loads an energy related dataset to use with sankey and graphs"""
|
|
tbl_name = 'energy_usage'
|
|
with gzip.open(os.path.join(DATA_FOLDER, 'energy.json.gz')) as f:
|
|
pdf = pd.read_json(f)
|
|
pdf.to_sql(
|
|
tbl_name,
|
|
db.engine,
|
|
if_exists='replace',
|
|
chunksize=500,
|
|
dtype={
|
|
'source': String(255),
|
|
'target': String(255),
|
|
'value': Float(),
|
|
},
|
|
index=False)
|
|
|
|
print('Creating table [wb_health_population] reference')
|
|
tbl = db.session.query(TBL).filter_by(table_name=tbl_name).first()
|
|
if not tbl:
|
|
tbl = TBL(table_name=tbl_name)
|
|
tbl.description = 'Energy consumption'
|
|
tbl.database = utils.get_or_create_main_db()
|
|
db.session.merge(tbl)
|
|
db.session.commit()
|
|
tbl.fetch_metadata()
|
|
|
|
slc = Slice(
|
|
slice_name='Energy Sankey',
|
|
viz_type='sankey',
|
|
datasource_type='table',
|
|
datasource_id=tbl.id,
|
|
params=textwrap.dedent("""\
|
|
{
|
|
"collapsed_fieldsets": "",
|
|
"groupby": [
|
|
"source",
|
|
"target"
|
|
],
|
|
"having": "",
|
|
"metric": "sum__value",
|
|
"row_limit": "5000",
|
|
"slice_name": "Energy Sankey",
|
|
"viz_type": "sankey",
|
|
"where": ""
|
|
}
|
|
"""),
|
|
)
|
|
misc_dash_slices.add(slc.slice_name)
|
|
merge_slice(slc)
|
|
|
|
slc = Slice(
|
|
slice_name='Energy Force Layout',
|
|
viz_type='directed_force',
|
|
datasource_type='table',
|
|
datasource_id=tbl.id,
|
|
params=textwrap.dedent("""\
|
|
{
|
|
"charge": "-500",
|
|
"collapsed_fieldsets": "",
|
|
"groupby": [
|
|
"source",
|
|
"target"
|
|
],
|
|
"having": "",
|
|
"link_length": "200",
|
|
"metric": "sum__value",
|
|
"row_limit": "5000",
|
|
"slice_name": "Force",
|
|
"viz_type": "directed_force",
|
|
"where": ""
|
|
}
|
|
"""),
|
|
)
|
|
misc_dash_slices.add(slc.slice_name)
|
|
merge_slice(slc)
|
|
|
|
slc = Slice(
|
|
slice_name='Heatmap',
|
|
viz_type='heatmap',
|
|
datasource_type='table',
|
|
datasource_id=tbl.id,
|
|
params=textwrap.dedent("""\
|
|
{
|
|
"all_columns_x": "source",
|
|
"all_columns_y": "target",
|
|
"canvas_image_rendering": "pixelated",
|
|
"collapsed_fieldsets": "",
|
|
"having": "",
|
|
"linear_color_scheme": "blue_white_yellow",
|
|
"metric": "sum__value",
|
|
"normalize_across": "heatmap",
|
|
"slice_name": "Heatmap",
|
|
"viz_type": "heatmap",
|
|
"where": "",
|
|
"xscale_interval": "1",
|
|
"yscale_interval": "1"
|
|
}
|
|
"""),
|
|
)
|
|
misc_dash_slices.add(slc.slice_name)
|
|
merge_slice(slc)
|