X-Git-Url: http://www.git.stargrave.org/?p=dotfiles.git;a=blobdiff_plain;f=vim%2F.vim%2Fpack%2Fstargrave%2Fstart%2Fpy-importcompl%2Fautoload%2Fpython%2Fimportcompl.vim;fp=vim%2F.vim%2Fpack%2Fstargrave%2Fstart%2Fpy-importcompl%2Fautoload%2Fpython%2Fimportcompl.vim;h=e516ad662bc5ce2794fa247d82e8b8b6407e382d;hp=aa9a1685df669c28aac1952af2ebbc3ecf9b64a3;hb=01fb86734d5d83bfc44978d9ebf0927f5a37944e;hpb=50c727abbb918becd61df61de534dacc2d54ddc1

diff --git a/vim/.vim/pack/stargrave/start/py-importcompl/autoload/python/importcompl.vim b/vim/.vim/pack/stargrave/start/py-importcompl/autoload/python/importcompl.vim
index aa9a168..e516ad6 100644
--- a/vim/.vim/pack/stargrave/start/py-importcompl/autoload/python/importcompl.vim
+++ b/vim/.vim/pack/stargrave/start/py-importcompl/autoload/python/importcompl.vim
@@ -56,10 +56,13 @@ function! python#importcompl#all() abort
             let imports[m[2]] = m[1]
         endfor
     endfor
+
+    let lines = getloclist(winnr())
+    if len(lines) == 0 | let lines = getqflist() | endif
     let result = []
-    for diag in lsp#ui#vim#diagnostics#get_all_buffer_diagnostics()
-        if diag["message"] !~ "^undefined name" | continue | endif
-        let m = diag["message"][strridx(diag["message"][:-2], "'")+1:-2]
+    for line in lines
+        if line.text !~ "^undefined name" | continue | endif
+        let m = line.text[strridx(line.text[:-2], "'")+1:-2]
         if len(m) == 0 || !has_key(imports, m) | continue | endif
         call insert(result, imports[m])
     endfor