mirror of https://github.com/databricks/cli.git
63 lines
1.5 KiB
Go
63 lines
1.5 KiB
Go
package whl
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/databricks/cli/bundle"
|
|
"github.com/databricks/cli/bundle/config"
|
|
"github.com/databricks/cli/bundle/libraries"
|
|
"github.com/databricks/cli/libs/diag"
|
|
"github.com/databricks/cli/libs/log"
|
|
)
|
|
|
|
type detectPkg struct{}
|
|
|
|
func DetectPackage() bundle.Mutator {
|
|
return &detectPkg{}
|
|
}
|
|
|
|
func (m *detectPkg) Name() string {
|
|
return "artifacts.whl.AutoDetect"
|
|
}
|
|
|
|
func (m *detectPkg) Apply(ctx context.Context, b *bundle.Bundle) diag.Diagnostics {
|
|
if b.Config.Artifacts != nil {
|
|
log.Debugf(ctx, "artifacts block is defined, skipping auto-detecting")
|
|
return nil
|
|
}
|
|
|
|
tasks := libraries.FindTasksWithLocalLibraries(b)
|
|
if len(tasks) == 0 {
|
|
log.Infof(ctx, "No local tasks in databricks.yml config, skipping auto detect")
|
|
return nil
|
|
}
|
|
|
|
log.Infof(ctx, "Detecting Python wheel project...")
|
|
|
|
// checking if there is setup.py in the bundle root
|
|
setupPy := filepath.Join(b.BundleRootPath, "setup.py")
|
|
_, err := os.Stat(setupPy)
|
|
if err != nil {
|
|
log.Infof(ctx, "No Python wheel project found at bundle root folder")
|
|
return nil
|
|
}
|
|
|
|
log.Infof(ctx, "Found Python wheel project at %s", b.BundleRootPath)
|
|
|
|
pkgPath, err := filepath.Abs(b.BundleRootPath)
|
|
if err != nil {
|
|
return diag.FromErr(err)
|
|
}
|
|
|
|
b.Config.Artifacts = make(map[string]*config.Artifact)
|
|
b.Config.Artifacts["python_artifact"] = &config.Artifact{
|
|
Path: pkgPath,
|
|
Type: config.ArtifactPythonWheel,
|
|
// BuildCommand will be set by bundle/artifacts/whl/infer.go to "python3 setup.py bdist_wheel"
|
|
}
|
|
|
|
return nil
|
|
}
|