mirror of https://github.com/databricks/cli.git
24 lines
519 B
Go
24 lines
519 B
Go
|
package notebook
|
||
|
|
|
||
|
|
import "github.com/databricks/databricks-sdk-go/service/workspace"
|
||
|
|
|
||
|
|
func GetExtensionByLanguage(objectInfo *workspace.ObjectInfo) string {
|
||
|
|
if objectInfo.ObjectType != workspace.ObjectTypeNotebook {
|
||
|
|
return ""
|
||
|
|
}
|
||
|
|
|
||
|
|
switch objectInfo.Language {
|
||
|
|
case workspace.LanguagePython:
|
||
|
|
return ".py"
|
||
|
|
case workspace.LanguageR:
|
||
|
|
return ".r"
|
||
|
|
case workspace.LanguageScala:
|
||
|
|
return ".scala"
|
||
|
|
case workspace.LanguageSql:
|
||
|
|
return ".sql"
|
||
|
|
default:
|
||
|
|
// Do not add any extension to the file name
|
||
|
|
return ""
|
||
|
|
}
|
||
|
|
}
|