eschutho commented on a change in pull request #15107: URL: https://github.com/apache/superset/pull/15107#discussion_r650189466
########## File path: superset-frontend/src/utils/export.ts ########## @@ -0,0 +1,48 @@ +/** + * 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 parseCookie from 'src/utils/parseCookie'; +import rison from 'rison'; +import shortid from 'shortid'; + +export default function handleResourceExport( + resource: string, + ids: number[], + done: () => void, + interval = 200, +): void { + const token = shortid.generate(); + const url = `/api/v1/${resource}/export/?q=${rison.encode( + ids, + )}&token=${token}`; + + // create new iframe for export + const iframe = document.createElement('iframe'); + iframe.style.display = 'none'; + iframe.src = url; + document.body.appendChild(iframe); + + const timer = window.setInterval(() => { + const cookie = parseCookie(); Review comment: do you want to add a type to the cookie so that we know that it will never be undefined or null, so we don't have to worry about cookie[token] erroring? ########## File path: superset-frontend/src/utils/export.ts ########## @@ -0,0 +1,48 @@ +/** + * 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 parseCookie from 'src/utils/parseCookie'; +import rison from 'rison'; +import shortid from 'shortid'; + +export default function handleResourceExport( + resource: string, + ids: number[], + done: () => void, + interval = 200, +): void { + const token = shortid.generate(); + const url = `/api/v1/${resource}/export/?q=${rison.encode( + ids, + )}&token=${token}`; + + // create new iframe for export + const iframe = document.createElement('iframe'); + iframe.style.display = 'none'; Review comment: where do you change this so that it's visible? ########## File path: superset/charts/api.py ########## @@ -918,12 +919,15 @@ def export(self, **kwargs: Any) -> Response: return self.response_404() buf.seek(0) - return send_file( + response = send_file( buf, mimetype="application/zip", as_attachment=True, attachment_filename=filename, ) + if token: + response.set_cookie(token, "done", max_age=600) Review comment: what do you think about just removing/expiring the cookie instead of changing the value? ########## File path: superset/charts/api.py ########## @@ -918,12 +919,15 @@ def export(self, **kwargs: Any) -> Response: return self.response_404() buf.seek(0) - return send_file( + response = send_file( buf, mimetype="application/zip", as_attachment=True, attachment_filename=filename, ) + if token: + response.set_cookie(token, "done", max_age=600) Review comment: oh, nm. I see.. this is the only cookie that you set, and it doesn't change. ########## File path: superset-frontend/src/views/CRUD/dashboard/DashboardList.tsx ########## @@ -170,6 +172,14 @@ function DashboardList(props: DashboardListProps) { ); } + const handleBulkDashboardExport = (dashboardsToExport: Dashboard[]) => { + const ids = dashboardsToExport.map(({ id }) => id); + handleResourceExport('chart', ids, () => { Review comment: is the 'chart' value a typo here for bulk dashboard? ########## File path: superset-frontend/src/views/CRUD/data/savedquery/SavedQueryList.tsx ########## @@ -238,6 +240,16 @@ function SavedQueryList({ ); }; + const handleBulkSavedQueryExport = ( + savedQueriesToExport: SavedQueryObject[], + ) => { + const ids = savedQueriesToExport.map(({ id }) => id); + handleResourceExport('chart', ids, () => { Review comment: same... query? ########## File path: superset-frontend/src/utils/export.ts ########## @@ -0,0 +1,48 @@ +/** + * 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 parseCookie from 'src/utils/parseCookie'; +import rison from 'rison'; +import shortid from 'shortid'; + +export default function handleResourceExport( + resource: string, + ids: number[], + done: () => void, + interval = 200, +): void { + const token = shortid.generate(); + const url = `/api/v1/${resource}/export/?q=${rison.encode( + ids, + )}&token=${token}`; + + // create new iframe for export + const iframe = document.createElement('iframe'); + iframe.style.display = 'none'; + iframe.src = url; + document.body.appendChild(iframe); + + const timer = window.setInterval(() => { + const cookie = parseCookie(); Review comment: yeah, I was going to make a comment below that we should make sure that `cookie` exists before trying to access a property on it, and then looked up the return value for cookie and realized that it will never be null or undefined. I was thinking we can define that in typescript here for readability and also for any future changes. -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: [email protected] --------------------------------------------------------------------- To unsubscribe, e-mail: [email protected] For additional commands, e-mail: [email protected]
