Just use vim-python.