import("lib.detect.find_path") import("lib.detect.find_library") import("lib.detect.find_program") import("lib.detect.find_programver") function _find_binary(package, opt) local result = package:find_tool("python3", opt) if not result then result = package:find_tool("python", opt) end if result then -- check if pip, setuptools and wheel are installed local ok = try { function () os.vrunv(result.program, {"-c", "import pip"}) os.vrunv(result.program, {"-c", "import setuptools"}) os.vrunv(result.program, {"-c", "import wheel"}) return true end} if not ok then return false end end end function _find_library(package, opt) -- init search options opt = opt or {} opt.paths = opt.paths or {} table.insert(opt.paths, "$(env PATH)") table.insert(opt.paths, "$(env CONDA_PREFIX)") -- find python local program = find_program("python3", opt) if not program then program = find_program("python", opt) end local version = nil if program then opt.command = function () local outs, errs = os.iorunv(program, {"--version"}) return ((outs or "") .. (errs or "")):trim() end version = find_programver(program, opt) end if not program or not version then return false end -- find library and header local exepath = path.directory(program) local link = nil local libpath = nil local includepath = nil if package:is_plat("windows") then link = format("python" .. table.concat(table.slice(version:split("%."), 1, 2), "")) libpath = find_library(link, {exepath}, {suffixes = {"libs"}}) linkdirs = {} includepath = find_path("Python.h", {exepath}, {suffixes = {"include"}}) else local pyver = table.concat(table.slice(version:split("%."), 1, 2), ".") link = format("python" .. pyver) libpath = find_library(link, {path.directory(exepath)}, {suffixes = {"lib", "lib64", "lib/x86_64-linux-gnu"}}) includepath = find_path("Python.h", {path.directory(exepath)}, {suffixes = {"include/python" .. pyver}}) end -- return if libpath and includepath then local result = { version = version, link = libpath.link, linkdirs = {libpath.linkdir}, includedirs = {includepath} } return result end end function main(package, opt) if opt.system then if package:is_binary() then return _find_binary(package, opt) else return _find_library(package, opt) end end end