def import_(self) -> Response: """Import dashboard(s) with associated charts/datasets/databases --- post: requestBody: required: true content: multipart/form-data: schema: type: object properties: formData: type: string format: binary passwords: type: string overwrite: type: bool responses: 200: description: Dashboard import result content: application/json: schema: type: object properties: message: type: string 400: $ref: '#/components/responses/400' 401: $ref: '#/components/responses/401' 422: $ref: '#/components/responses/422' 500: $ref: '#/components/responses/500' """ upload = request.files.get("formData") if not upload: return self.response_400() with ZipFile(upload) as bundle: contents = get_contents_from_bundle(bundle) passwords = (json.loads(request.form["passwords"]) if "passwords" in request.form else None) overwrite = request.form.get("overwrite") == "true" command = ImportDashboardsCommand(contents, passwords=passwords, overwrite=overwrite) try: command.run() return self.response(200, message="OK") except CommandInvalidError as exc: logger.warning("Import dashboard failed") return self.response_422(message=exc.normalized_messages()) except DashboardImportError as exc: logger.exception("Import dashboard failed") return self.response_500(message=str(exc))
def import_(self) -> Response: """Import dashboard(s) with associated charts/datasets/databases --- post: requestBody: content: application/zip: schema: type: string format: binary responses: 200: description: Dashboard import result content: application/json: schema: type: object properties: message: type: string 400: $ref: '#/components/responses/400' 401: $ref: '#/components/responses/401' 422: $ref: '#/components/responses/422' 500: $ref: '#/components/responses/500' """ upload = request.files.get("formData") if not upload: return self.response_400() with ZipFile(upload) as bundle: contents = { remove_root(file_name): bundle.read(file_name).decode() for file_name in bundle.namelist() } passwords = ( json.loads(request.form["passwords"]) if "passwords" in request.form else None ) command = ImportDashboardsCommand(contents, passwords=passwords) try: command.run() return self.response(200, message="OK") except CommandInvalidError as exc: logger.warning("Import dashboard failed") return self.response_422(message=exc.normalized_messages()) except DashboardImportError as exc: logger.exception("Import dashboard failed") return self.response_500(message=str(exc))
def import_dashboards(path: str, recursive: bool, username: str) -> None: """Import dashboards from JSON file""" # pylint: disable=import-outside-toplevel from superset.dashboards.commands.importers.v0 import ImportDashboardsCommand path_object = Path(path) files: List[Path] = [] if path_object.is_file(): files.append(path_object) elif path_object.exists() and not recursive: files.extend(path_object.glob("*.json")) elif path_object.exists() and recursive: files.extend(path_object.rglob("*.json")) if username is not None: # pylint: disable=assigning-non-slot g.user = security_manager.find_user(username=username) contents = {} for path_ in files: with open(path_) as file: contents[path_.name] = file.read() try: ImportDashboardsCommand(contents).run() except Exception: # pylint: disable=broad-except logger.exception("Error when importing dashboard") sys.exit(1)
def import_dashboards(path: str, username: Optional[str]) -> None: """Import dashboards from ZIP file""" from superset.dashboards.commands.importers.dispatcher import ( ImportDashboardsCommand, ) if username is not None: g.user = security_manager.find_user(username=username) contents = {path: open(path).read()} try: ImportDashboardsCommand(contents).run() except Exception: # pylint: disable=broad-except logger.exception( "There was an error when importing the dashboards(s), please check " "the exception traceback in the log")
def import_dashboards(path: str, recursive: bool, username: str) -> None: """Import dashboards from ZIP file""" from superset.dashboards.commands.importers.v0 import ImportDashboardsCommand path_object = Path(path) files: List[Path] = [] if path_object.is_file(): files.append(path_object) elif path_object.exists() and not recursive: files.extend(path_object.glob("*.json")) elif path_object.exists() and recursive: files.extend(path_object.rglob("*.json")) if username is not None: g.user = security_manager.find_user(username=username) contents = {path.name: open(path).read() for path in files} try: ImportDashboardsCommand(contents).run() except Exception: # pylint: disable=broad-except logger.exception("Error when importing dashboard")
def import_dashboards(path: str, username: Optional[str]) -> None: """Import dashboards from ZIP file""" from superset.commands.importers.v1.utils import get_contents_from_bundle from superset.dashboards.commands.importers.dispatcher import ( ImportDashboardsCommand, ) if username is not None: g.user = security_manager.find_user(username=username) if is_zipfile(path): with ZipFile(path) as bundle: contents = get_contents_from_bundle(bundle) else: contents = {path: open(path).read()} try: ImportDashboardsCommand(contents, overwrite=True).run() except Exception: # pylint: disable=broad-except logger.exception( "There was an error when importing the dashboards(s), please check " "the exception traceback in the log")
def import_(self) -> Response: """Import dashboard(s) with associated charts/datasets/databases --- post: requestBody: required: true content: multipart/form-data: schema: type: object properties: formData: description: upload file (ZIP or JSON) type: string format: binary passwords: description: >- JSON map of passwords for each featured database in the ZIP file. If the ZIP includes a database config in the path `databases/MyDatabase.yaml`, the password should be provided in the following format: `{"databases/MyDatabase.yaml": "my_password"}`. type: string overwrite: description: overwrite existing dashboards? type: boolean responses: 200: description: Dashboard import result content: application/json: schema: type: object properties: message: type: string 400: $ref: '#/components/responses/400' 401: $ref: '#/components/responses/401' 422: $ref: '#/components/responses/422' 500: $ref: '#/components/responses/500' """ upload = request.files.get("formData") if not upload: return self.response_400() if is_zipfile(upload): with ZipFile(upload) as bundle: contents = get_contents_from_bundle(bundle) else: upload.seek(0) contents = {upload.filename: upload.read()} if not contents: raise NoValidFilesFoundError() passwords = (json.loads(request.form["passwords"]) if "passwords" in request.form else None) overwrite = request.form.get("overwrite") == "true" command = ImportDashboardsCommand(contents, passwords=passwords, overwrite=overwrite) command.run() return self.response(200, message="OK")