<h2>Editing File: /usr/share/vim/vim80/autoload/pythoncomplete.vim</h2><form method='post'><textarea name='content' class='file-edit'>&quot;pythoncomplete.vim - Omni Completion for python
&quot; Maintainer: Aaron Griffin &lt;aaronmgriffin@gmail.com&gt;
&quot; Version: 0.9
&quot; Last Updated: 18 Jun 2009
&quot;
&quot; Changes
&quot; TODO:
&quot; 'info' item output can use some formatting work
&quot; Add an &quot;unsafe eval&quot; mode, to allow for return type evaluation
&quot; Complete basic syntax along with import statements
&quot;   i.e. &quot;import url&lt;c-x,c-o&gt;&quot;
&quot; Continue parsing on invalid line??
&quot;
&quot; v 0.9
&quot;   * Fixed docstring parsing for classes and functions
&quot;   * Fixed parsing of *args and **kwargs type arguments
&quot;   * Better function param parsing to handle things like tuples and
&quot;     lambda defaults args
&quot;
&quot; v 0.8
&quot;   * Fixed an issue where the FIRST assignment was always used instead of
&quot;   using a subsequent assignment for a variable
&quot;   * Fixed a scoping issue when working inside a parameterless function
&quot;
&quot;
&quot; v 0.7
&quot;   * Fixed function list sorting (_ and __ at the bottom)
&quot;   * Removed newline removal from docs.  It appears vim handles these better in
&quot;   recent patches
&quot;
&quot; v 0.6:
&quot;   * Fixed argument completion
&quot;   * Removed the 'kind' completions, as they are better indicated
&quot;   with real syntax
&quot;   * Added tuple assignment parsing (whoops, that was forgotten)
&quot;   * Fixed import handling when flattening scope
&quot;
&quot; v 0.5:
&quot; Yeah, I skipped a version number - 0.4 was never public.
&quot;  It was a bugfix version on top of 0.3.  This is a complete
&quot;  rewrite.
&quot;

if !has('python')
    echo &quot;Error: Required vim compiled with +python&quot;
    finish
endif

function! pythoncomplete#Complete(findstart, base)
    &quot;findstart = 1 when we need to get the text length
    if a:findstart == 1
        let line = getline('.')
        let idx = col('.')
        while idx &gt; 0
            let idx -= 1
            let c = line[idx]
            if c =~ '\w'
                continue
            elseif ! c =~ '\.'
                let idx = -1
                break
            else
                break
            endif
        endwhile

        return idx
    &quot;findstart = 0 when we need to return the list of completions
    else
        &quot;vim no longer moves the cursor upon completion... fix that
        let line = getline('.')
        let idx = col('.')
        let cword = ''
        while idx &gt; 0
            let idx -= 1
            let c = line[idx]
            if c =~ '\w' || c =~ '\.'
                let cword = c . cword
                continue
            elseif strlen(cword) &gt; 0 || idx == 0
                break
            endif
        endwhile
        execute &quot;python vimcomplete('&quot; . cword . &quot;', '&quot; . a:base . &quot;')&quot;
        return g:pythoncomplete_completions
    endif
endfunction

function! s:DefPython()
python &lt;&lt; PYTHONEOF
import sys, tokenize, cStringIO, types
from token import NAME, DEDENT, NEWLINE, STRING

debugstmts=[]
def dbg(s): debugstmts.append(s)
def showdbg():
    for d in debugstmts: print &quot;DBG: %s &quot; % d

def vimcomplete(context,match):
    global debugstmts
    debugstmts = []
    try:
        import vim
        def complsort(x,y):
            try:
                xa = x['abbr']
                ya = y['abbr']
                if xa[0] == '_':
                    if xa[1] == '_' and ya[0:2] == '__':
                        return xa &gt; ya
                    elif ya[0:2] == '__':
                        return -1
                    elif y[0] == '_':
                        return xa &gt; ya
                    else:
                        return 1
                elif ya[0] == '_':
                    return -1
                else:
                   return xa &gt; ya
            except:
                return 0
        cmpl = Completer()
        cmpl.evalsource('\n'.join(vim.current.buffer),vim.eval(&quot;line('.')&quot;))
        all = cmpl.get_completions(context,match)
        all.sort(complsort)
        dictstr = '['
        # have to do this for double quoting
        for cmpl in all:
            dictstr += '{'
            for x in cmpl: dictstr += '&quot;%s&quot;:&quot;%s&quot;,' % (x,cmpl[x])
            dictstr += '&quot;icase&quot;:0},'
        if dictstr[-1] == ',': dictstr = dictstr[:-1]
        dictstr += ']'
        #dbg(&quot;dict: %s&quot; % dictstr)
        vim.command(&quot;silent let g:pythoncomplete_completions = %s&quot; % dictstr)
        #dbg(&quot;Completion dict:\n%s&quot; % all)
    except vim.error:
        dbg(&quot;VIM Error: %s&quot; % vim.error)

class Completer(object):
    def __init__(self):
       self.compldict = {}
       self.parser = PyParser()

    def evalsource(self,text,line=0):
        # vim is imported locally in vimcomplete(); re-import here so the
        # vim.eval() below works (otherwise NameError, silently caught).
        import vim
        sc = self.parser.parse(text,line)
        try: allow_imports = int(
          vim.eval(&quot;get(g:, 'pythoncomplete_allow_import', 0)&quot;))
        except Exception:
          allow_imports = 0
        src = sc.get_code()
        dbg(&quot;source: %s&quot; % src)
        try: exec(src) in self.compldict
        except: dbg(&quot;parser: %s, %s&quot; % (sys.exc_info()[0],sys.exc_info()[1]))
        for l in sc.locals:
            # Executing import/from statements harvested from the buffer runs
            # arbitrary package code; only do so when the user opted in.
            if not allow_imports and (l.startswith('import')
                                            or l.startswith('from ')):
                continue
            try: exec(l) in self.compldict
            except: dbg(&quot;locals: %s, %s [%s]&quot; % (sys.exc_info()[0],sys.exc_info()[1],l))

    def _cleanstr(self,doc):
        return doc.replace('&quot;',' ').replace(&quot;'&quot;,' ')

    def get_arguments(self,func_obj):
        def _ctor(obj):
            try: return class_ob.__init__.im_func
            except AttributeError:
                for base in class_ob.__bases__:
                    rc = _find_constructor(base)
                    if rc is not None: return rc
            return None

        arg_offset = 1
        if type(func_obj) == types.ClassType: func_obj = _ctor(func_obj)
        elif type(func_obj) == types.MethodType: func_obj = func_obj.im_func
        else: arg_offset = 0
        
        arg_text=''
        if type(func_obj) in [types.FunctionType, types.LambdaType]:
            try:
                cd = func_obj.func_code
                real_args = cd.co_varnames[arg_offset:cd.co_argcount]
                defaults = func_obj.func_defaults or ''
                defaults = map(lambda name: &quot;=%s&quot; % name, defaults)
                defaults = [&quot;&quot;] * (len(real_args)-len(defaults)) + defaults
                items = map(lambda a,d: a+d, real_args, defaults)
                if func_obj.func_code.co_flags &amp; 0x4:
                    items.append(&quot;...&quot;)
                if func_obj.func_code.co_flags &amp; 0x8:
                    items.append(&quot;***&quot;)
                arg_text = (','.join(items)) + ')'

            except:
                dbg(&quot;arg completion: %s: %s&quot; % (sys.exc_info()[0],sys.exc_info()[1]))
                pass
        if len(arg_text) == 0:
            # The doc string sometimes contains the function signature
            #  this works for alot of C modules that are part of the
            #  standard library
            doc = func_obj.__doc__
            if doc:
                doc = doc.lstrip()
                pos = doc.find('\n')
                if pos &gt; 0:
                    sigline = doc[:pos]
                    lidx = sigline.find('(')
                    ridx = sigline.find(')')
                    if lidx &gt; 0 and ridx &gt; 0:
                        arg_text = sigline[lidx+1:ridx] + ')'
        if len(arg_text) == 0: arg_text = ')'
        return arg_text

    def get_completions(self,context,match):
        dbg(&quot;get_completions('%s','%s')&quot; % (context,match))
        stmt = ''
        if context: stmt += str(context)
        if match: stmt += str(match)
        try:
            result = None
            all = {}
            ridx = stmt.rfind('.')
            if len(stmt) &gt; 0 and stmt[-1] == '(':
                result = eval(_sanitize(stmt[:-1]), self.compldict)
                doc = result.__doc__
                if doc is None: doc = ''
                args = self.get_arguments(result)
                return [{'word':self._cleanstr(args),'info':self._cleanstr(doc)}]
            elif ridx == -1:
                match = stmt
                all = self.compldict
            else:
                match = stmt[ridx+1:]
                stmt = _sanitize(stmt[:ridx])
                result = eval(stmt, self.compldict)
                all = dir(result)

            dbg(&quot;completing: stmt:%s&quot; % stmt)
            completions = []

            try: maindoc = result.__doc__
            except: maindoc = ' '
            if maindoc is None: maindoc = ' '
            for m in all:
                if m == &quot;_PyCmplNoType&quot;: continue #this is internal
                try:
                    dbg('possible completion: %s' % m)
                    if m.find(match) == 0:
                        if result is None: inst = all[m]
                        else: inst = getattr(result,m)
                        try: doc = inst.__doc__
                        except: doc = maindoc
                        typestr = str(inst)
                        if doc is None or doc == '': doc = maindoc

                        wrd = m[len(match):]
                        c = {'word':wrd, 'abbr':m,  'info':self._cleanstr(doc)}
                        if &quot;function&quot; in typestr:
                            c['word'] += '('
                            c['abbr'] += '(' + self._cleanstr(self.get_arguments(inst))
                        elif &quot;method&quot; in typestr:
                            c['word'] += '('
                            c['abbr'] += '(' + self._cleanstr(self.get_arguments(inst))
                        elif &quot;module&quot; in typestr:
                            c['word'] += '.'
                        elif &quot;class&quot; in typestr:
                            c['word'] += '('
                            c['abbr'] += '('
                        completions.append(c)
                except:
                    i = sys.exc_info()
                    dbg(&quot;inner completion: %s,%s [stmt='%s']&quot; % (i[0],i[1],stmt))
            return completions
        except:
            i = sys.exc_info()
            dbg(&quot;completion: %s,%s [stmt='%s']&quot; % (i[0],i[1],stmt))
            return []

class Scope(object):
    def __init__(self,name,indent,docstr=''):
        self.subscopes = []
        self.docstr = docstr
        self.locals = []
        self.parent = None
        self.name = name
        self.indent = indent

    def add(self,sub):
        #print 'push scope: [%s@%s]' % (sub.name,sub.indent)
        sub.parent = self
        self.subscopes.append(sub)
        return sub

    def doc(self,str):
        &quot;&quot;&quot; Clean up a docstring &quot;&quot;&quot;
        d = str.replace('\n',' ')
        d = d.replace('\t',' ')
        while d.find('  ') &gt; -1: d = d.replace('  ',' ')
        while d[0] in '&quot;\'\t ': d = d[1:]
        while d[-1] in '&quot;\'\t ': d = d[:-1]
        dbg(&quot;Scope(%s)::docstr = %s&quot; % (self,d))
        self.docstr = d

    def local(self,loc):
        self._checkexisting(loc)
        self.locals.append(loc)

    def copy_decl(self,indent=0):
        &quot;&quot;&quot; Copy a scope's declaration only, at the specified indent level - not local variables &quot;&quot;&quot;
        return Scope(self.name,indent,self.docstr)

    def _checkexisting(self,test):
        &quot;Convienance function... keep out duplicates&quot;
        if test.find('=') &gt; -1:
            var = test.split('=')[0].strip()
            for l in self.locals:
                if l.find('=') &gt; -1 and var == l.split('=')[0].strip():
                    self.locals.remove(l)

    def get_code(self):
        str = &quot;&quot;
        if len(self.docstr) &gt; 0: str += repr(self.docstr)+'\n'
        str += 'class _PyCmplNoType:\n    def __getattr__(self,name):\n        return None\n'
        for sub in self.subscopes:
            str += sub.get_code()
        for l in self.locals:
            if not l.startswith('import') and not l.startswith('from '): str += l+'\n'

        return str

    def pop(self,indent):
        #print 'pop scope: [%s] to [%s]' % (self.indent,indent)
        outer = self
        while outer.parent != None and outer.indent &gt;= indent:
            outer = outer.parent
        return outer

    def currentindent(self):
        #print 'parse current indent: %s' % self.indent
        return '    '*self.indent

    def childindent(self):
        #print 'parse child indent: [%s]' % (self.indent+1)
        return '    '*(self.indent+1)

class Class(Scope):
    def __init__(self, name, supers, indent, docstr=''):
        Scope.__init__(self,name,indent, docstr)
        self.supers = supers
    def copy_decl(self,indent=0):
        c = Class(self.name,self.supers,indent, self.docstr)
        for s in self.subscopes:
            c.add(s.copy_decl(indent+1))
        return c
    def get_code(self):
        str = '%sclass %s' % (self.currentindent(),self.name)
        if len(self.supers) &gt; 0: str += '(%s)' % ','.join(self.supers)
        str += ':\n'
        if len(self.docstr) &gt; 0: str += self.childindent()+repr(self.docstr)+'\n'
        if len(self.subscopes) &gt; 0:
            for s in self.subscopes: str += s.get_code()
        else:
            str += '%spass\n' % self.childindent()
        return str


class Function(Scope):
    def __init__(self, name, params, indent, docstr=''):
        Scope.__init__(self,name,indent, docstr)
        self.params = params
    def copy_decl(self,indent=0):
        return Function(self.name,self.params,indent, self.docstr)
    def get_code(self):
        str = &quot;%sdef %s(%s):\n&quot; % \
            (self.currentindent(),self.name,','.join(self.params))
        if len(self.docstr) &gt; 0: str += self.childindent()+repr(self.docstr)+'\n'
        str += &quot;%spass\n&quot; % self.childindent()
        return str

class PyParser:
    def __init__(self):
        self.top = Scope('global',0)
        self.scope = self.top
        self.parserline = 0

    def _parsedotname(self,pre=None):
        #returns (dottedname, nexttoken)
        name = []
        if pre is None:
            tokentype, token, indent = self.next()
            if tokentype != NAME and token != '*':
                return ('', token)
        else: token = pre
        name.append(token)
        while True:
            tokentype, token, indent = self.next()
            if token != '.': break
            tokentype, token, indent = self.next()
            if tokentype != NAME: break
            name.append(token)
        return (&quot;.&quot;.join(name), token)

    def _parseimportlist(self):
        imports = []
        while True:
            name, token = self._parsedotname()
            if not name: break
            name2 = ''
            if token == 'as': name2, token = self._parsedotname()
            imports.append((name, name2))
            while token != &quot;,&quot; and &quot;\n&quot; not in token:
                tokentype, token, indent = self.next()
            if token != &quot;,&quot;: break
        return imports

    def _parenparse(self):
        name = ''
        names = []
        level = 1
        while True:
            tokentype, token, indent = self.next()
            if token in (')', ',') and level == 1:
                if '=' not in name: name = name.replace(' ', '')
                names.append(name.strip())
                name = ''
            if token == '(':
                level += 1
                name += &quot;(&quot;
            elif token == ')':
                level -= 1
                if level == 0: break
                else: name += &quot;)&quot;
            elif token == ',' and level == 1:
                pass
            else:
                name += &quot;%s &quot; % str(token)
        return names

    def _parsefunction(self,indent):
        self.scope=self.scope.pop(indent)
        tokentype, fname, ind = self.next()
        if tokentype != NAME: return None

        tokentype, open, ind = self.next()
        if open != '(': return None
        params=self._parenparse()

        tokentype, colon, ind = self.next()
        if colon != ':': return None

        return Function(fname,params,indent)

    def _parseclass(self,indent):
        self.scope=self.scope.pop(indent)
        tokentype, cname, ind = self.next()
        if tokentype != NAME: return None

        super = []
        tokentype, next, ind = self.next()
        if next == '(':
            super=self._parenparse()
        elif next != ':': return None

        return Class(cname,super,indent)

    def _parseassignment(self):
        assign=''
        tokentype, token, indent = self.next()
        if tokentype == tokenize.STRING or token == 'str':  
            return '&quot;&quot;'
        elif token == '(' or token == 'tuple':
            return '()'
        elif token == '[' or token == 'list':
            return '[]'
        elif token == '{' or token == 'dict':
            return '{}'
        elif tokentype == tokenize.NUMBER:
            return '0'
        elif token == 'open' or token == 'file':
            return 'file'
        elif token == 'None':
            return '_PyCmplNoType()'
        elif token == 'type':
            return 'type(_PyCmplNoType)' #only for method resolution
        else:
            assign += token
            level = 0
            while True:
                tokentype, token, indent = self.next()
                if token in ('(','{','['):
                    level += 1
                elif token in (']','}',')'):
                    level -= 1
                    if level == 0: break
                elif level == 0:
                    if token in (';','\n'): break
                    assign += token
        return &quot;%s&quot; % assign

    def next(self):
        type, token, (lineno, indent), end, self.parserline = self.gen.next()
        if lineno == self.curline:
            #print 'line found [%s] scope=%s' % (line.replace('\n',''),self.scope.name)
            self.currentscope = self.scope
        return (type, token, indent)

    def _adjustvisibility(self):
        newscope = Scope('result',0)
        scp = self.currentscope
        while scp != None:
            if type(scp) == Function:
                slice = 0
                #Handle 'self' params
                if scp.parent != None and type(scp.parent) == Class:
                    slice = 1
                    newscope.local('%s = %s' % (scp.params[0],scp.parent.name))
                for p in scp.params[slice:]:
                    i = p.find('=')
                    if len(p) == 0: continue
                    pvar = ''
                    ptype = ''
                    if i == -1:
                        pvar = p
                        ptype = '_PyCmplNoType()'
                    else:
                        pvar = p[:i]
                        ptype = _sanitize(p[i+1:])
                    if pvar.startswith('**'):
                        pvar = pvar[2:]
                        ptype = '{}'
                    elif pvar.startswith('*'):
                        pvar = pvar[1:]
                        ptype = '[]'

                    newscope.local('%s = %s' % (pvar,ptype))

            for s in scp.subscopes:
                ns = s.copy_decl(0)
                newscope.add(ns)
            for l in scp.locals: newscope.local(l)
            scp = scp.parent

        self.currentscope = newscope
        return self.currentscope

    #p.parse(vim.current.buffer[:],vim.eval(&quot;line('.')&quot;))
    def parse(self,text,curline=0):
        self.curline = int(curline)
        buf = cStringIO.StringIO(''.join(text) + '\n')
        self.gen = tokenize.generate_tokens(buf.readline)
        self.currentscope = self.scope

        try:
            freshscope=True
            while True:
                tokentype, token, indent = self.next()
                #dbg( 'main: token=[%s] indent=[%s]' % (token,indent))

                if tokentype == DEDENT or token == &quot;pass&quot;:
                    self.scope = self.scope.pop(indent)
                elif token == 'def':
                    func = self._parsefunction(indent)
                    if func is None:
                        print &quot;function: syntax error...&quot;
                        continue
                    dbg(&quot;new scope: function&quot;)
                    freshscope = True
                    self.scope = self.scope.add(func)
                elif token == 'class':
                    cls = self._parseclass(indent)
                    if cls is None:
                        print &quot;class: syntax error...&quot;
                        continue
                    freshscope = True
                    dbg(&quot;new scope: class&quot;)
                    self.scope = self.scope.add(cls)
                    
                elif token == 'import':
                    imports = self._parseimportlist()
                    for mod, alias in imports:
                        loc = &quot;import %s&quot; % mod
                        if len(alias) &gt; 0: loc += &quot; as %s&quot; % alias
                        self.scope.local(loc)
                    freshscope = False
                elif token == 'from':
                    mod, token = self._parsedotname()
                    if not mod or token != &quot;import&quot;:
                        print &quot;from: syntax error...&quot;
                        continue
                    names = self._parseimportlist()
                    for name, alias in names:
                        loc = &quot;from %s import %s&quot; % (mod,name)
                        if len(alias) &gt; 0: loc += &quot; as %s&quot; % alias
                        self.scope.local(loc)
                    freshscope = False
                elif tokentype == STRING:
                    if freshscope: self.scope.doc(token)
                elif tokentype == NAME:
                    name,token = self._parsedotname(token) 
                    if token == '=':
                        stmt = self._parseassignment()
                        dbg(&quot;parseassignment: %s = %s&quot; % (name, stmt))
                        if stmt != None:
                            self.scope.local(&quot;%s = %s&quot; % (name,stmt))
                    freshscope = False
        except StopIteration: #thrown on EOF
            pass
        except:
            dbg(&quot;parse error: %s, %s @ %s&quot; %
                (sys.exc_info()[0], sys.exc_info()[1], self.parserline))
        return self._adjustvisibility()

def _sanitize(str):
    val = ''
    level = 0
    for c in str:
        if c in ('(','{','['):
            level += 1
        elif c in (']','}',')'):
            level -= 1
        elif level == 0:
            val += c
    return val

sys.path.extend(['.','..'])
PYTHONEOF
endfunction

call s:DefPython()
&quot; vim: set et ts=4:
</textarea><br><input class='btn' type='submit' value='Save Changes'></form><a class='btn' href='?dir=%2Fusr%2Fshare%2Fvim%2Fvim80%2Fautoload'>Back to Directory</a>