用于插入Python导入(如果尚未导入)的Vim函数

function! InsertPythonImport(module)
    let l:import_line = 'import ' . a:module
    let l:import_line = substitute(l:import_line, '\.', '_', 'g')
    let l:import_line = substitute(l:import_line, '_', '.', 'g')
    let l:import_line = l:import_line . "\n"
    let l:current_line = getline('.')
    let l:imported = 0

    " 检查当前行是否已经导入了该模块
    if l:current_line =~ '^import\s\+' . a:module . '\s\+$'
        let l:imported = 1
    elseif l:current_line =~ '^from\s\+' . a:module . '\s\+import\s\+'
        let l:imported = 1
    endif

    " 如果未导入，则插入导入语句
    if !l:imported
        call append(line('.'), l:import_line)
    endif
endfunction

:call InsertPythonImport('module_name')