[ML-8575] Extend stack CLI to support html format (#263)

* add html support for stack cli

* clean up

Author: lu-wang-dl

databricks_cli/workspace/types.py

@@ -30,7 +30,7 @@ class WorkspaceLanguage(object):
     SQL = 'SQL'
     R = 'R'
     ALL = [SCALA, PYTHON, SQL, R]
-    EXTENSIONS = ['.scala', '.py', '.sql', '.SQL', '.r', '.R', '.ipynb']
+    EXTENSIONS = ['.scala', '.py', '.sql', '.SQL', '.r', '.R', '.ipynb', '.html']
 
     @classmethod
     def to_language_and_format(cls, path):
@@ -45,6 +45,8 @@ def to_language_and_format(cls, path):
             return (cls.R, WorkspaceFormat.SOURCE)
         elif ext == '.ipynb':
             return (cls.PYTHON, WorkspaceFormat.JUPYTER)
+        elif ext == '.html':
+            return (None, WorkspaceFormat.HTML)
 
     @classmethod
     def to_extension(cls, language):

tests/stack/test_api.py

@@ -301,6 +301,13 @@ def test_deploy_workspace(self, stack_api, tmpdir):
         assert nb_databricks_id == {api.WORKSPACE_RESOURCE_PATH:
                                     test_workspace_nb_properties[api.WORKSPACE_RESOURCE_PATH]}
 
+        # Test Input of Workspace notebook with html source
+        test_workspace_nb_properties.update(
+            {api.WORKSPACE_RESOURCE_SOURCE_PATH: 'test/notebook.html'})
+        nb_databricks_id = \
+            stack_api._deploy_workspace(test_workspace_nb_properties, None, True)
+        assert stack_api.workspace_client.import_workspace.call_args[0][0] == 'test/notebook.html'
+
         # Should raise error if resource object_type doesn't match actually is in filesystem.
         test_workspace_dir_properties.update(
             {api.WORKSPACE_RESOURCE_OBJECT_TYPE: workspace_api.NOTEBOOK})