julia – 在Jupyter笔记本中运行代码的单行或短脚本？

2019年8月6日 491次阅读

我喜欢在Jupyter(nee iJulia)笔记本中逐个运行脚本来开发脚本.但是,有时我需要在远程系统上测试内容,并且需要将代码的副本复制为.jl文件.有没有人已经编写了一个单线程或短脚本来运行.ipynb笔记本中的代码？如果没有,我会在某个时候到达并在此处发布代码. 最佳答案 这是我写的：

using JSON

get_code_cells(j::Dict) = filter(x->x["cell_type"] == "code", j["cells"])

function parse_code_cell(c::Dict)
    buf = IOBuffer()
    write(buf, "begin\n")
    map(x->write(buf, x), c["source"])
    write(buf, "\nend")

    src = bytestring(buf)
    parse(src)
end

extract_code(cells::Vector) = Expr[parse_code_cell(c) for c in cells]
extract_code(j::Dict) = extract_code(get_code_cells(j))
eval_code(j::Dict) = map(eval, extract_code(j))


# get filename, then parse to json, then run all code
const fn = ARGS[1]
eval_code(JSON.parsefile(fn))

它似乎适用于许多笔记本电脑,但不是一切.具体来说,它无法运行我所拥有的笔记本电脑

using PyCall
@pyimport seaborn as sns

当eval遇到那些抱怨@pyimport未被定义的代码块时(即使它是由PyCall导出的).

如果您感兴趣,我们肯定可以清理它,添加更多参数并将其打包到适当的命令行实用程序中.

编辑

现在完全不同的东西……

这个版本弹出到ipython nbconvert,将其写入临时文件,调用include在该临时文件上运行代码,然后删除临时文件.这应该更加健壮(它通过了另一个失败的例子).关于清洁/包装的相同评论适用.

const fn = abspath(ARGS[1])
dir = dirname(fn)

# shell out to nbconvert to get a string with code
src = readall(`ipython nbconvert --to script --stdout $fn`)

# Generate random filenamein this directory, write code string to it
script_fn = joinpath(dir, string(randstring(30), ".jl"))
open(script_fn, "w") do f
    write(f, src)
end

# now try to run the file we just write. We do this so we can make sure
# to get to the call `rm(script_fn)` below.
try
    include(script_fn)
catch
    warn("Failed executing script from file")
end

# clean up by deleting the temporary file we created
rm(script_fn)
点赞