steve/databricks-cli
steve
/
databricks-cli
mirror of https://github.com/databricks/cli.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

use cutSuffix; add link to pep

This commit is contained in:
Denis Bilenko 2025-03-05 11:50:09 +01:00
parent 04eed7a0d9
commit d79e0db71b
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
libs/patchwheel/parse.go
View File

@ -36,6 +36,7 @@ func calculateNewVersion(info WheelInfo, mtime time.Time) (newVersion, newFilena
// ParseWheelFilename parses a wheel filename and extracts its components. // ParseWheelFilename parses a wheel filename and extracts its components.
// Wheel filenames follow the pattern: {distribution}-{version}(-{build tag})?-{python_tag}-{abi_tag}-{platform_tag}.whl // Wheel filenames follow the pattern: {distribution}-{version}(-{build tag})?-{python_tag}-{abi_tag}-{platform_tag}.whl
// https://peps.python.org/pep-0491
func ParseWheelFilename(filename string) (WheelInfo, error) { func ParseWheelFilename(filename string) (WheelInfo, error) {
parts := strings.Split(filename, "-") parts := strings.Split(filename, "-")
if len(parts) < 5 { if len(parts) < 5 {
@ -46,11 +47,13 @@ func ParseWheelFilename(filename string) (WheelInfo, error) {
return WheelInfo{}, fmt.Errorf("invalid wheel filename format: too many parts in %s", filename) return WheelInfo{}, fmt.Errorf("invalid wheel filename format: too many parts in %s", filename)
} }
if !strings.HasSuffix(parts[len(parts)-1], ".whl") { trimmedLastTag, foundWhl := strings.CutSuffix(parts[len(parts)-1], ".whl")
if !foundWhl {
return WheelInfo{}, fmt.Errorf("invalid wheel filename format: missing .whl extension in %s", filename) return WheelInfo{}, fmt.Errorf("invalid wheel filename format: missing .whl extension in %s", filename)
} }
parts[len(parts)-1] = strings.TrimSuffix(parts[len(parts)-1], ".whl") parts[len(parts)-1] = trimmedLastTag
return WheelInfo{ return WheelInfo{
Distribution: parts[0], Distribution: parts[0],