forked from VimPlug/jedi
the auto completion is working! and may god bless you all! ;-)
This commit is contained in:
16
ftest.py
16
ftest.py
@@ -2,7 +2,7 @@
|
|||||||
|
|
||||||
import functions
|
import functions
|
||||||
|
|
||||||
functions.debug.debug_function = functions.debug.print_to_stdout
|
#functions.debug.debug_function = functions.debug.print_to_stdout
|
||||||
functions.debug.ignored_modules += ['parsing', 'builtin']
|
functions.debug.ignored_modules += ['parsing', 'builtin']
|
||||||
functions.modules.module_find_path.insert(0, '.')
|
functions.modules.module_find_path.insert(0, '.')
|
||||||
|
|
||||||
@@ -12,3 +12,17 @@ for i in range(1):
|
|||||||
completions = functions.complete(code, 50, 200)
|
completions = functions.complete(code, 50, 200)
|
||||||
|
|
||||||
print '\n', ', '.join(str(c) for c in completions)
|
print '\n', ', '.join(str(c) for c in completions)
|
||||||
|
|
||||||
|
out = []
|
||||||
|
for c in completions:
|
||||||
|
d = dict(word=str(c),
|
||||||
|
abbr=c.complete,
|
||||||
|
menu=c.description, # the stuff directly behind the completion
|
||||||
|
info=c.help, # docstr and similar stuff
|
||||||
|
kind=c.type, # completion type
|
||||||
|
icase=1, # case insensitive
|
||||||
|
dup=1, # allow duplicates (maybe later remove this)
|
||||||
|
)
|
||||||
|
out.append(d)
|
||||||
|
|
||||||
|
print str(out)
|
||||||
|
|||||||
54
functions.py
54
functions.py
@@ -6,7 +6,7 @@ import evaluate
|
|||||||
import modules
|
import modules
|
||||||
import debug
|
import debug
|
||||||
|
|
||||||
__all__ = ['complete', 'complete_test', 'set_debug_function']
|
__all__ = ['complete', 'get_completion_parts', 'complete_test', 'set_debug_function']
|
||||||
|
|
||||||
|
|
||||||
class FileWithCursor(modules.File):
|
class FileWithCursor(modules.File):
|
||||||
@@ -37,7 +37,7 @@ class FileWithCursor(modules.File):
|
|||||||
line = self.get_line(self._row_temp)
|
line = self.get_line(self._row_temp)
|
||||||
if self._is_first:
|
if self._is_first:
|
||||||
self._is_first = False
|
self._is_first = False
|
||||||
line = line[:column - 1]
|
line = line[:column]
|
||||||
else:
|
else:
|
||||||
line = line + '\n'
|
line = line + '\n'
|
||||||
# add lines with a backslash at the end
|
# add lines with a backslash at the end
|
||||||
@@ -57,6 +57,7 @@ class FileWithCursor(modules.File):
|
|||||||
close_brackets = [')', ']', '}']
|
close_brackets = [')', ']', '}']
|
||||||
|
|
||||||
gen = tokenize.generate_tokens(fetch_line)
|
gen = tokenize.generate_tokens(fetch_line)
|
||||||
|
# TODO can happen: raise TokenError, ("EOF in multi-line statement"
|
||||||
string = ''
|
string = ''
|
||||||
level = 0
|
level = 0
|
||||||
for token_type, tok, start, end, line in gen:
|
for token_type, tok, start, end, line in gen:
|
||||||
@@ -99,16 +100,44 @@ class FileWithCursor(modules.File):
|
|||||||
raise StopIteration()
|
raise StopIteration()
|
||||||
|
|
||||||
|
|
||||||
class CompletionMatch(object):
|
class Completion(object):
|
||||||
def __init__(self, name, has_dot):
|
def __init__(self, name, needs_dot, like_name_length):
|
||||||
self.name = name
|
self.name = name
|
||||||
self.has_dot = has_dot
|
self.needs_dot = needs_dot
|
||||||
|
self.like_name_length = like_name_length
|
||||||
|
|
||||||
|
@property
|
||||||
|
def complete(self):
|
||||||
|
dot = '.' if self.needs_dot else ''
|
||||||
|
return dot + self.name.names[-1][self.like_name_length:]
|
||||||
|
|
||||||
|
@property
|
||||||
|
def description(self):
|
||||||
|
return str(self.name.parent)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def help(self):
|
||||||
|
try:
|
||||||
|
return str(self.name.parent.docstr)
|
||||||
|
except:
|
||||||
|
return ''
|
||||||
|
|
||||||
|
@property
|
||||||
|
def type(self):
|
||||||
|
return '' # type(self.name)
|
||||||
|
|
||||||
def __str__(self):
|
def __str__(self):
|
||||||
dot = '' if self.has_dot else '.'
|
return self.name.names[-1]
|
||||||
return dot + self.name.names[-1]
|
|
||||||
|
|
||||||
|
|
||||||
|
def get_completion_parts(path):
|
||||||
|
"""
|
||||||
|
Returns the parts for the completion
|
||||||
|
:return: tuple - (path, dot, like)
|
||||||
|
"""
|
||||||
|
match = re.match(r'^(.*?)(\.|)(\w?[\w\d]*)$', path, flags=re.S)
|
||||||
|
return match.groups()
|
||||||
|
|
||||||
def complete(source, row, column, file_callback=None):
|
def complete(source, row, column, file_callback=None):
|
||||||
"""
|
"""
|
||||||
An auto completer for python files.
|
An auto completer for python files.
|
||||||
@@ -127,9 +156,8 @@ def complete(source, row, column, file_callback=None):
|
|||||||
path = f.get_row_path(column)
|
path = f.get_row_path(column)
|
||||||
debug.dbg('completion_start: %s in %s' % (path, scope))
|
debug.dbg('completion_start: %s in %s' % (path, scope))
|
||||||
|
|
||||||
match = re.match(r'^(.+?)(?:(\.)(\w?[\w\d]*))?$', path, flags=re.S)
|
|
||||||
path, dot, like = match.groups()
|
|
||||||
# just parse one statement, take it and evaluate it
|
# just parse one statement, take it and evaluate it
|
||||||
|
path, dot, like = get_completion_parts(path)
|
||||||
r = parsing.PyFuzzyParser(path)
|
r = parsing.PyFuzzyParser(path)
|
||||||
try:
|
try:
|
||||||
stmt = r.top.statements[0]
|
stmt = r.top.statements[0]
|
||||||
@@ -143,8 +171,12 @@ def complete(source, row, column, file_callback=None):
|
|||||||
for s in scopes:
|
for s in scopes:
|
||||||
completions += s.get_defined_names()
|
completions += s.get_defined_names()
|
||||||
|
|
||||||
completions = [CompletionMatch(c, bool(dot)) for c in completions
|
print repr(path), repr(dot), repr(like), row, column
|
||||||
if like in c.names[-1]]
|
print len(completions)
|
||||||
|
needs_dot = not dot and path
|
||||||
|
completions = [Completion(c, needs_dot, len(like)) for c in completions
|
||||||
|
if c.names[-1].lower().startswith(like.lower())]
|
||||||
|
print 'nr2', len(completions)
|
||||||
|
|
||||||
return completions
|
return completions
|
||||||
|
|
||||||
|
|||||||
66
plugin/jedi.vim
Normal file
66
plugin/jedi.vim
Normal file
@@ -0,0 +1,66 @@
|
|||||||
|
"py_fuzzycomplete.vim - Omni Completion for python in vim
|
||||||
|
" Maintainer: David Halter <davidhalter88@gmail.com>
|
||||||
|
" Version: 0.1
|
||||||
|
"
|
||||||
|
" This part of the software is just the vim interface. The main source code
|
||||||
|
" lies in the python files around it.
|
||||||
|
|
||||||
|
if !has('python')
|
||||||
|
echomsg "Error: Required vim compiled with +python"
|
||||||
|
finish
|
||||||
|
endif
|
||||||
|
|
||||||
|
|
||||||
|
function! jedi#Complete(findstart, base)
|
||||||
|
if a:findstart == 1
|
||||||
|
return col('.')
|
||||||
|
else
|
||||||
|
python << PYTHONEOF
|
||||||
|
if 1:
|
||||||
|
row, column = vim.current.window.cursor
|
||||||
|
source = '\n'.join(vim.current.buffer)
|
||||||
|
try:
|
||||||
|
completions = functions.complete(source, row, column)
|
||||||
|
out = []
|
||||||
|
for c in completions:
|
||||||
|
d = dict(word=c.complete,
|
||||||
|
abbr=str(c),
|
||||||
|
menu=c.description, # stuff directly behind the completion
|
||||||
|
info=c.help, # docstr and similar stuff
|
||||||
|
kind=c.type, # completion type
|
||||||
|
icase=1, # case insensitive
|
||||||
|
dup=1, # allow duplicates (maybe later remove this)
|
||||||
|
)
|
||||||
|
out.append(d)
|
||||||
|
|
||||||
|
strout = str(out)
|
||||||
|
except Exception as e:
|
||||||
|
print 'error:', e
|
||||||
|
strout = ''
|
||||||
|
|
||||||
|
print 'end', strout
|
||||||
|
vim.command('return ' + strout)
|
||||||
|
PYTHONEOF
|
||||||
|
endif
|
||||||
|
endfunction
|
||||||
|
|
||||||
|
|
||||||
|
" ------------------------------------------------------------------------
|
||||||
|
" Initialization of Jedi
|
||||||
|
" ------------------------------------------------------------------------
|
||||||
|
"
|
||||||
|
let s:current_file=expand("<sfile>")
|
||||||
|
|
||||||
|
python << PYTHONEOF
|
||||||
|
""" here we initialize the jedi stuff """
|
||||||
|
import vim
|
||||||
|
|
||||||
|
# update the system path, to include the python scripts
|
||||||
|
import sys
|
||||||
|
from os.path import dirname
|
||||||
|
sys.path.insert(0, dirname(dirname(vim.eval('s:current_file'))))
|
||||||
|
|
||||||
|
import functions
|
||||||
|
PYTHONEOF
|
||||||
|
|
||||||
|
" vim: set et ts=4:
|
||||||
@@ -1,63 +0,0 @@
|
|||||||
"py_fuzzycomplete.vim - Omni Completion for python in vim
|
|
||||||
" Maintainer: David Halter <davidhalter88@gmail.com>
|
|
||||||
" Version: 0.1
|
|
||||||
"
|
|
||||||
" This part of the software is just the vim interface. The main source code
|
|
||||||
" lies in the python files around it.
|
|
||||||
|
|
||||||
if !has('python')
|
|
||||||
echo "Error: Required vim compiled with +python"
|
|
||||||
finish
|
|
||||||
endif
|
|
||||||
|
|
||||||
function! pythoncomplete#Complete(findstart, base)
|
|
||||||
"findstart = 1 when we need to get the text length
|
|
||||||
" TODO check wheter this is really needed
|
|
||||||
if a:findstart == 1
|
|
||||||
let line = getline('.')
|
|
||||||
let idx = col('.')
|
|
||||||
while idx > 0
|
|
||||||
let idx -= 1
|
|
||||||
let c = line[idx]
|
|
||||||
if c =~ '\w'
|
|
||||||
continue
|
|
||||||
elseif ! c =~ '\.'
|
|
||||||
let idx = -1
|
|
||||||
break
|
|
||||||
else
|
|
||||||
break
|
|
||||||
endif
|
|
||||||
endwhile
|
|
||||||
|
|
||||||
return idx
|
|
||||||
"findstart = 0 when we need to return the list of completions
|
|
||||||
else
|
|
||||||
execute "python vimcomplete()"
|
|
||||||
return g:pythoncomplete_completions
|
|
||||||
endif
|
|
||||||
endfunction
|
|
||||||
|
|
||||||
function! s:DefPython()
|
|
||||||
python << PYTHONEOF
|
|
||||||
import functions
|
|
||||||
|
|
||||||
def vimcomplete():
|
|
||||||
(row, column) = vim.current.window.cursor
|
|
||||||
code = '\n'.join(vim.current.buffer)
|
|
||||||
all = functions.complete(code, row, column)
|
|
||||||
|
|
||||||
dictstr = '['
|
|
||||||
# have to do this for double quoting
|
|
||||||
for cmpl in all:
|
|
||||||
dictstr += '{'
|
|
||||||
for x in cmpl: dictstr += '"%s":"%s",' % (x,cmpl[x])
|
|
||||||
dictstr += '"icase":0},'
|
|
||||||
if dictstr[-1] == ',': dictstr = dictstr[:-1]
|
|
||||||
dictstr += ']'
|
|
||||||
vim.command("silent let g:pythoncomplete_completions = %s" % dictstr)
|
|
||||||
|
|
||||||
PYTHONEOF
|
|
||||||
endfunction
|
|
||||||
|
|
||||||
call s:DefPython()
|
|
||||||
" vim: set et ts=4:
|
|
||||||
Reference in New Issue
Block a user