Support both Python 2 and Python 3

- And specify MIT license

Author: yssl

README.md

@@ -1,7 +1,5 @@
 # AutoCWD.vim
 
-*Currently, this plugin only supports the vim compiled with python 2 support (+python).*
-
 AutoCWD.vim automatically changes the current working directory (CWD) of vim when you change the current buffer (or window).
 You can define patterns that may be included in a file path or buffer name, and corresponding working directories in your .vimrc. 
 <!--The predefined working directories can be the directory of the current buffer's file, absolute paths of specific directories, and so on.-->
@@ -20,8 +18,10 @@ Screenshot:
 - Manual install (not recommended)
     - Download this plugin and extract it in `~/.vim/`
 
-This plugin requires a version of vim with python support. You can check your vim with `:echo has('python')`.
-- Linux: If your vim doesn't support python, one of the easiest solutions would be installing a more featured version of vim by `sudo apt-get install vim-gtk-py2`.
+## Requirements
+
+This plugin requires a version of vim with Python 2 (+python) or Python 3 (+python3) support. You can check your vim with `:echo has('python')` or `:echo has('python3')`.
+- Ubuntu: If your vim doesn't support python, one of the easiest solutions would be installing a more featured version of vim by `sudo apt-get install vim-gtk`.
 - Windows: gvim for Windows is already equipped with python support.
 
 ## Usage

autoload/autocwd.vim

@@ -1,7 +1,7 @@
 " File:         autoload/autocwd.vim
-" Description:  Auto current working directory update system
+" Description:  Automatic CWD(current working directory) updating system
 " Author:       yssl <http://github.com/yssl>
-" License:      
+" License:      MIT License
 
 " wrappers
 function! autocwd#PrintWorkDirs()
@@ -10,8 +10,19 @@ endfunction
 
 " script variables
 
+" Support for Python3 and Python2
+" from https://github.com/Valloric/YouCompleteMe
+function! s:UsingPython3()
+	if has('python3')
+		return 1
+	endif
+	return 0
+endfunction
+let s:using_python3 = s:UsingPython3()
+let s:pythonX_until_EOF = s:using_python3 ? "python3 << EOF" : "python << EOF"
+
 " initialize python 
-python << EOF
+exec s:pythonX_until_EOF
 import vim
 import fnmatch
 
@@ -31,17 +42,17 @@ def toWidthColMat(rowMat):
 	return colMat
 
 def ltrunc(s, width, prefix=''):
-    if width >= len(s): prefix = ''
-    return prefix+s[-width+len(prefix):]
-    
+	if width >= len(s): prefix = ''
+	return prefix+s[-width+len(prefix):]
+	
 def rtrunc(s, width, postfix=''):
-    if width >= len(s): postfix = ''
-    return s[:width-len(postfix)]+postfix
+	if width >= len(s): postfix = ''
+	return s[:width-len(postfix)]+postfix
 EOF
 
 " functions
 function! s:GetWorkDir(filepath)
-python << EOF
+exec s:pythonX_until_EOF
 filepath = vim.eval('a:filepath')
 patternwd_pairs = vim.eval('g:autocwd_patternwd_pairs')
 inpatternwd = False
@@ -50,7 +61,7 @@ for pattern, wd in patternwd_pairs:
 		if '*REPO*' in wd:
 			wd = wd.replace('*REPO*', findRepoDirFrom(filepath))
 		wd = vim.eval('expand(\'%s\')'%wd)
-	   	if os.path.isdir(wd):
+		if os.path.isdir(wd):
 			inpatternwd = True
 			vim.command('return \'%s\''%os.path.abspath(wd))
 			break
@@ -60,7 +71,7 @@ EOF
 endfunction
 
 function! s:GetWorkDirPattern(filepath)
-python << EOF
+exec s:pythonX_until_EOF
 import fnmatch
 filepath = vim.eval('a:filepath')
 patternwd_pairs = vim.eval('g:autocwd_patternwd_pairs')
@@ -70,7 +81,7 @@ for pattern, wd in patternwd_pairs:
 		if '*REPO*' in wd:
 			wd = wd.replace('*REPO*', findRepoDirFrom(filepath))
 		wd = vim.eval('expand(\'%s\')'%wd)
-	   	if os.path.isdir(wd):
+		if os.path.isdir(wd):
 			inpatternwd = True
 			vim.command('return \'%s\''%pattern)
 			break
@@ -80,7 +91,7 @@ EOF
 endfunction
 
 function! s:BuildAllWinPropMat(propTypes)
-python << EOF
+exec s:pythonX_until_EOF
 vim.command('let mat = []')
 curwin = vim.current.window
 curwin_r = 0
@@ -112,7 +123,7 @@ for r in range(len(vim.windows)):
 
 		elif type=='workdir':
 			dir = vim.eval('s:GetWorkDir(\'%s\')'%getWinName(bufname, buftype))
-			#print '|%s|%s|%s|'%(bufname,buftype,dir)
+			#print('|%s|%s|%s|'%(bufname,buftype,dir))
 			vim.command('call add(mat[-1], \'%s\')'%dir)
 
 		elif type=='workdir_pattern':
@@ -126,7 +137,7 @@ endfunction
 
 function! s:PrintWorkDirs()
 	let a:type = 'workdir'
-python << EOF
+exec s:pythonX_until_EOF
 vim.command('let propTypes = ["iscurwin", "winnr", "winname", "workdir_pattern", "workdir"]')
 wpMat = vim.eval('s:BuildAllWinPropMat(propTypes)')
 propTypes = vim.eval('propTypes')
@@ -178,6 +189,5 @@ if user_input.isdigit():
 	winnum = int(vim.eval('winnr("$")'))
 	if winidx<=winnum:
 		vim.command('%dwincmd w'%winidx)
-
 EOF
 endfunction

doc/autocwd.txt

@@ -21,8 +21,8 @@ when you change the current buffer (or window).
 You can define patterns that may be included in a file path or buffer name,
  and corresponding working directories in your .vimrc. 
 
-This plugin requires a version of vim with python support.
-You can check your vim with `:echo has('python')`.
+This plugin requires a version of vim with Python 2 (+python) or Python 3 (+python3) 
+support. You can check your vim with `:echo has('python')` or `:echo has('python3')`.
 
 ==============================================================================
 2. Usage                                                    *autocwd-usage*

plugin/autocwd.vim

@@ -1,7 +1,7 @@
 " File:         plugin/autocwd.vim
-" Description:  Auto current working directory update system
+" Description:  Automatic CWD(current working directory) updating system
 " Author:       yssl <http://github.com/yssl>
-" License:      
+" License:      MIT License
 
 if exists("g:loaded_autocwd") || &cp
 	finish
@@ -11,6 +11,14 @@ let s:keepcpo           = &cpo
 set cpo&vim
 """""""""""""""""""""""""""""""""""""""""""""
 
+" vim version checking
+if !has('python3') && !has('python')
+	echohl WarningMsg
+	echomsg 'AutoCWD.vim unavailable: requires vim with Python support'
+	echohl None
+	finish
+endif
+
 " global variables
 if !exists('g:autocwd_patternwd_pairs')
 	let g:autocwd_patternwd_pairs = []
@@ -35,8 +43,19 @@ augroup AutoCWDAutoCmds
 	autocmd BufLeave * call s:OnLeaveBuf() 
 augroup END
 
+" Support for Python3 and Python2
+" from https://github.com/Valloric/YouCompleteMe
+function! s:UsingPython3()
+	if has('python3')
+		return 1
+	endif
+	return 0
+endfunction
+let s:using_python3 = s:UsingPython3()
+let s:pythonX_until_EOF = s:using_python3 ? "python3 << EOF" : "python << EOF"
+
 " initialize python 
-python << EOF
+exec s:pythonX_until_EOF
 import vim
 import os, fnmatch
 
@@ -105,7 +124,7 @@ function! s:OnLeaveBuf()
 endfunction
 
 function! s:ExistPattern(bufname, buftype)
-python << EOF
+exec s:pythonX_until_EOF
 bufname = vim.eval('a:bufname')
 buftype = vim.eval('a:buftype')
 filepath = getWinName(bufname, buftype)
@@ -116,7 +135,7 @@ for pattern, wd in patternwd_pairs:
 		if '*REPO*' in wd:
 			wd = wd.replace('*REPO*', findRepoDirFrom(filepath))
 		wd = vim.eval('expand(\'%s\')'%wd)
-	   	if os.path.isdir(wd):
+		if os.path.isdir(wd):
 			inpatternwd = True
 			vim.command('return [1, \'%s\']'%wd)
 			break