Skip to content

Commit

Permalink
Merge branch 'lyftga'
Browse files Browse the repository at this point in the history
  • Loading branch information
mistercrunch committed Apr 25, 2019
2 parents f488a32 + cb7c806 commit 929fb6b
Show file tree
Hide file tree
Showing 14 changed files with 46 additions and 16 deletions.
4 changes: 2 additions & 2 deletions superset/assets/cypress/integration/dashboard/controls.js
Original file line number Diff line number Diff line change
Expand Up @@ -38,8 +38,8 @@ export default () => describe('top-level controls', () => {
.forEach((slice) => {
const sliceRequest = `getJson_${slice.slice_id}`;
sliceRequests.push(`@${sliceRequest}`);
const formData = `{"slice_id":${slice.slice_id},"viz_type":"${slice.form_data.viz_type}"}`;
cy.route('GET', `/superset/explore_json/?form_data=${formData}`).as(sliceRequest);
const formData = `{"slice_id":${slice.slice_id}}`;
cy.route('POST', `/superset/explore_json/?form_data=${formData}`).as(sliceRequest);

const forceRefresh = `postJson_${slice.slice_id}_force`;
forceRefreshRequests.push(`@${forceRefresh}`);
Expand Down
4 changes: 2 additions & 2 deletions superset/assets/cypress/integration/dashboard/edit_mode.js
Original file line number Diff line number Diff line change
Expand Up @@ -28,9 +28,9 @@ export default () => describe('edit mode', () => {
const bootstrapData = JSON.parse(data[0].dataset.bootstrap);
const dashboard = bootstrapData.dashboard_data;
const boxplotChartId = dashboard.slices.find(slice => (slice.form_data.viz_type === 'box_plot')).slice_id;
const formData = `{"slice_id":${boxplotChartId},"viz_type":"box_plot"}`;
const formData = `{"slice_id":${boxplotChartId}}`;
const boxplotRequest = `/superset/explore_json/?form_data=${formData}`;
cy.route('GET', boxplotRequest).as('boxplotRequest');
cy.route('POST', boxplotRequest).as('boxplotRequest');
});

cy.get('.dashboard-header').contains('Edit dashboard').click();
Expand Down
4 changes: 2 additions & 2 deletions superset/assets/cypress/integration/dashboard/filter.js
Original file line number Diff line number Diff line change
Expand Up @@ -39,9 +39,9 @@ export default () => describe('dashboard filter', () => {
it('should apply filter', () => {
const aliases = [];

const formData = `{"slice_id":${filterId},"viz_type":"filter_box"}`;
const formData = `{"slice_id":${filterId}}`;
const filterRoute = `/superset/explore_json/?form_data=${formData}`;
cy.route('GET', filterRoute).as('fetchFilter');
cy.route('POST', filterRoute).as('fetchFilter');
cy.wait('@fetchFilter');
sliceIds
.filter(id => (parseInt(id, 10) !== filterId))
Expand Down
4 changes: 2 additions & 2 deletions superset/assets/cypress/integration/dashboard/load.js
Original file line number Diff line number Diff line change
Expand Up @@ -34,8 +34,8 @@ export default () => describe('load', () => {
// then define routes and create alias for each requests
slices.forEach((slice) => {
const alias = `getJson_${slice.slice_id}`;
const formData = `{"slice_id":${slice.slice_id},"viz_type":"${slice.form_data.viz_type}"}`;
cy.route('GET', `/superset/explore_json/?form_data=${formData}`).as(alias);
const formData = `{"slice_id":${slice.slice_id}}`;
cy.route('POST', `/superset/explore_json/?form_data=${formData}`).as(alias);
aliases.push(`@${alias}`);
});
});
Expand Down
4 changes: 2 additions & 2 deletions superset/assets/cypress/integration/dashboard/save.js
Original file line number Diff line number Diff line change
Expand Up @@ -56,9 +56,9 @@ export default () => describe('save', () => {
cy.wait('@copyRequest');

// should have box_plot chart
const formData = `{"slice_id":${boxplotChartId},"viz_type":"box_plot"}`;
const formData = `{"slice_id":${boxplotChartId}}`;
const boxplotRequest = `/superset/explore_json/?form_data=${formData}`;
cy.route('GET', boxplotRequest).as('boxplotRequest');
cy.route('POST', boxplotRequest).as('boxplotRequest');
cy.wait('@boxplotRequest');
cy.get('.grid-container .box_plot').should('be.exist');

Expand Down
3 changes: 2 additions & 1 deletion superset/assets/src/chart/Chart.jsx
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@ import PropTypes from 'prop-types';
import React from 'react';
import { Alert } from 'react-bootstrap';

import { isFeatureEnabled, FeatureFlag } from 'src/featureFlags';
import { Logger, LOG_ACTIONS_RENDER_CHART_CONTAINER } from '../logger/LogUtils';
import Loading from '../components/Loading';
import RefreshChartOverlay from '../components/RefreshChartOverlay';
Expand Down Expand Up @@ -70,7 +71,7 @@ class Chart extends React.PureComponent {
}
componentDidMount() {
if (this.props.triggerQuery) {
if (this.props.chartId > 0) {
if (this.props.chartId > 0 && isFeatureEnabled(FeatureFlag.CLIENT_CACHE)) {
// Load saved chart with a GET request
this.props.actions.getSavedChart(
this.props.formData,
Expand Down
5 changes: 4 additions & 1 deletion superset/assets/src/chart/chartAction.js
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,7 @@
/* eslint no-param-reassign: ["error", { "props": false }] */
import { t } from '@superset-ui/translation';
import { SupersetClient } from '@superset-ui/connection';
import { isFeatureEnabled, FeatureFlag } from 'src/featureFlags';
import { getExploreUrlAndPayload, getAnnotationJsonUrl } from '../explore/exploreUtils';
import { requiresQuery, ANNOTATION_SOURCE_TYPES } from '../modules/AnnotationTypes';
import { addDangerToast } from '../messageToasts/actions';
Expand Down Expand Up @@ -194,7 +195,9 @@ export function exploreJSON(formData, force = false, timeout = 60, key, method)
};
}

const clientMethod = method === 'GET' ? SupersetClient.get : SupersetClient.post;
const clientMethod = method === 'GET' && isFeatureEnabled(FeatureFlag.CLIENT_CACHE)
? SupersetClient.get
: SupersetClient.post;
const queryPromise = clientMethod(querySettings)
.then(({ json }) => {
dispatch(logEvent(LOG_ACTIONS_LOAD_CHART, {
Expand Down
2 changes: 1 addition & 1 deletion superset/assets/src/dashboard/components/Header.jsx
Original file line number Diff line number Diff line change
Expand Up @@ -222,7 +222,7 @@ class Header extends React.PureComponent {
colorScheme,
colorNamespace,
);
const labelColors = scale.getColorMap();
const labelColors = colorScheme ? scale.getColorMap() : {};
const data = {
positions,
expanded_slices: expandedSlices,
Expand Down
2 changes: 1 addition & 1 deletion superset/assets/src/dashboard/components/SaveModal.jsx
Original file line number Diff line number Diff line change
Expand Up @@ -110,7 +110,7 @@ class SaveModal extends React.PureComponent {
colorScheme,
colorNamespace,
);
const labelColors = scale.getColorMap();
const labelColors = colorScheme ? scale.getColorMap() : {};
const data = {
positions,
css,
Expand Down
1 change: 1 addition & 0 deletions superset/assets/src/featureFlags.ts
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,7 @@
export enum FeatureFlag {
SCOPED_FILTER = 'SCOPED_FILTER',
OMNIBAR = 'OMNIBAR',
CLIENT_CACHE = 'CLIENT_CACHE',
}

export type FeatureFlagMap = {
Expand Down
5 changes: 4 additions & 1 deletion superset/config.py
Original file line number Diff line number Diff line change
Expand Up @@ -200,7 +200,10 @@
# For example, DEFAULT_FEATURE_FLAGS = { 'FOO': True, 'BAR': False } here
# and FEATURE_FLAGS = { 'BAR': True, 'BAZ': True } in superset_config.py
# will result in combined feature flags of { 'FOO': True, 'BAR': True, 'BAZ': True }
DEFAULT_FEATURE_FLAGS = {}
DEFAULT_FEATURE_FLAGS = {
# Experimental feature introducing a client (browser) cache
'CLIENT_CACHE': False,
}

# A function that receives a dict of all feature flags
# (DEFAULT_FEATURE_FLAGS merged with FEATURE_FLAGS)
Expand Down
4 changes: 4 additions & 0 deletions superset/models/core.py
Original file line number Diff line number Diff line change
Expand Up @@ -749,6 +749,10 @@ def table_cache_enabled(self):
def table_cache_timeout(self):
return self.metadata_cache_timeout.get('table_cache_timeout')

@property
def default_schemas(self):
return self.get_extra().get('default_schemas', [])

@classmethod
def get_password_masked_url_from_uri(cls, uri):
url = make_url(uri)
Expand Down
10 changes: 10 additions & 0 deletions superset/views/core.py
Original file line number Diff line number Diff line change
Expand Up @@ -1590,6 +1590,16 @@ def tables(self, db_id, schema, substr, force_refresh='false'):
table_names = [tn for tn in table_names if substr in tn]
view_names = [vn for vn in view_names if substr in vn]

if not schema and database.default_schemas:
def get_schema(tbl_or_view_name):
return tbl_or_view_name.split('.')[0] if '.' in tbl_or_view_name else None

user_schema = g.user.email.split('@')[0]
valid_schemas = set(database.default_schemas + [user_schema])

table_names = [tn for tn in table_names if get_schema(tn) in valid_schemas]
view_names = [vn for vn in view_names if get_schema(vn) in valid_schemas]

max_items = config.get('MAX_TABLE_NAMES') or len(table_names)
total_items = len(table_names) + len(view_names)
max_tables = len(table_names)
Expand Down
10 changes: 9 additions & 1 deletion superset/viz.py
Original file line number Diff line number Diff line change
Expand Up @@ -1112,16 +1112,22 @@ def to_series(self, df, classed='', title_suffix=''):
series_title = series_title + (title_suffix,)

values = []
non_nan_cnt = 0
for ds in df.index:
if ds in ys:
d = {
'x': ds,
'y': ys[ds],
}
if not np.isnan(ys[ds]):
non_nan_cnt += 1
else:
d = {}
values.append(d)

if non_nan_cnt == 0:
continue

d = {
'key': series_title,
'values': values,
Expand Down Expand Up @@ -1224,7 +1230,9 @@ def get_data(self, df):
comparison_type = fd.get('comparison_type') or 'values'
df = self.process_data(df)
if comparison_type == 'values':
chart_data = self.to_series(df)
# Filter out series with all NaN
chart_data = self.to_series(df.dropna(axis=1, how='all'))

for i, (label, df2) in enumerate(self._extra_chart_data):
chart_data.extend(
self.to_series(
Expand Down

0 comments on commit 929fb6b

Please sign in to comment.