python - 如果尚未导入，则 Vim 函数插入 Python 导入

Question

我目前有一个 Vim 函数，可以将选定的文本包装在 time.time() block 中，这样我就可以快速计时。

我希望该函数也转到文件顶部，检查 import time 是否存在，只有在不存在时才插入 import time尚不存在。

有什么方法可以检查 Vim 中是否存在文本？

Answer 1

这是我目前拥有的。它有效，但如果您有更好的解决方案，请发布您自己的解决方案!

另外，请注意具有 ^M 的行是通过按 Ctrl-V 然后在插入模式下按 Enter 按钮形成的(Stack Overflow 没有很好地复制它)。

" easily wrap the selected text in a time.time() statement for quick timing
fun! s:PythonTiming(line1, line2)

    " mark line one  &&  keep track of lines selected
    execute 'normal!' 'me'
    let l:numDiff = a:line2 - a:line1

    " start timing
    execute 'normal!' 'Ostart = time.time()'

    " end timing
    while line('.') < a:line2 + 1
        execute 'normal!' 'j'
    endwhile
    execute 'normal!' 'oend = time.time()'
    execute 'normal!' 'oprint; print("end - start: "); print(end - start)'

    " add the `import time` statement if not already imported
    let match = search('import time', 'nw')
    if match == 0
        silent! execute 'normal!' 'gg/import/^M'
        execute 'normal!' 'oimport time'
    endif

    " go back to the initial mark
    execute 'normal!' '`e'

endfun
command! -range Time :call s:PythonTiming(<line1>, <line2>)