Python: добавлено импортирование с "as NAME"

Этот коммит содержится в:
Softonik 2022-12-10 04:42:03 +03:00 коммит произвёл Nikolay Kopitonenko
родитель ee76ec40a8
коммит 6d70364357
2 изменённых файлов: 18 добавлений и 2 удалений

Просмотреть файл

@ -458,10 +458,13 @@ func handleImportSpec(spec ast.Spec) string {
s := spec.(*ast.ImportSpec) s := spec.(*ast.ImportSpec)
code := "" code := ""
libName := strings.Trim(s.Path.Value, "\"") libName := strings.Trim(s.Path.Value, "\"")
if s.Name == nil { switch {
case s.Name == nil:
code = "import " + libName + "\n" code = "import " + libName + "\n"
} else { case s.Name.Name == ".":
code = "from " + libName + " import *\n" code = "from " + libName + " import *\n"
default:
code = "import " + libName + " as " + s.Name.Name + "\n"
} }
return code return code
} }

Просмотреть файл

@ -62,6 +62,19 @@ main()
expected := `import skidl expected := `import skidl
import FreeCAD import FreeCAD
main()
`
Compare(source, expected)
})
It("Importing with name", func() {
source := `package test
import (
cq "cadquery"
)
`
expected := `import cadquery as cq
main() main()
` `
Compare(source, expected) Compare(source, expected)