Call >1 function from autocommand

[savchenko]

I would like to apply one particular LSP action on save. Here is the code:

vim.api.nvim_create_autocmd("BufWritePre", {
    pattern = "*.py",
    callback = function()
        ---@diagnostic disable-next-line: undefined-global
        local clients = vim.lsp.get_clients({buffer = buf})
        for _, client in ipairs(clients) do
            if client.name == 'pylsp' then
                local params = {
                    range = nil,
                    context = {
                        diagnostics = {},
                        only = { "source.organizeImports" }
                    },
                    apply = true
                }
                vim.lsp.buf.code_action(params) -- This returns.
            end
        end
    end
})

The problem is that vim.lsp.buf.code_action(params) returns and the buffer is not written unless I save twice. Effectively I must do :w, :w.

The autocommand also prints "No code actions available" if imports are already sorted, but this is a secondary concern.