Python: добавлено импортирование с "as NAME"
Этот коммит содержится в:
родитель
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)
|
||||||
|
|
Загрузка…
Создание таблицы
Сослаться в новой задаче