vimはPythonのインデントスタイルを変更します

現在の職場では、関数の複数行引数に次の形式を使用します。

function(
    arg1,
    arg2,
    arg3
)

しかし、現在のvim設定では、次のスタイルのみを受け入れます（このスタイルも気に入らない：

function(
    arg1,
    arg2,
    arg3
    )

vimでこれらのスタイルの自動インデントを許可する変更を提案できる人はいますか？

インデントに関連する私のvimrcのすべての内容は次のとおりです。

Plugin 'tpope/vim-surround'
Plugin 'edc/python-indent'
Plugin 'nvie/vim-flake8'

au BufNewFile,BufRead *.py
    \ setlocal tabstop=4 |
    \ setlocal softtabstop=4 |
    \ setlocal shiftwidth=4 |
    \ setlocal textwidth=79 |
    \ setlocal expandtab |
    \ setlocal autoindent |
    \ setlocal fileformat=unix |
    \ setlocal fileformat=unix |
    \ let b:indentLine_enabled = 0

新しいプラグインを使用したりvimrcを変更したりすることができますが、複数行の関数呼び出しをしたいときはいつでも手動で再インデントする必要があるため、これは本当に面倒です。