LunarVim/ftplugin/python.lua

local python_arguments = {}

-- TODO replace with path argument
local flake8 = {
  LintCommand = "flake8 --ignore=E501 --stdin-display-name ${INPUT} -",
  lintStdin = true,
  lintFormats = { "%f:%l:%c: %m" },
}

local isort = { formatCommand = "isort --quiet -", formatStdin = true }

local yapf = { formatCommand = "yapf --quiet", formatStdin = true }
local black = { formatCommand = "black --quiet -", formatStdin = true }

if O.lang.python.linter == "flake8" then
  table.insert(python_arguments, flake8)
end

if O.lang.python.isort then
  table.insert(python_arguments, isort)
end

if O.lang.python.formatter == "yapf" then
  table.insert(python_arguments, yapf)
elseif O.lang.python.formatter == "black" then
  table.insert(python_arguments, black)
end

require("lspconfig").efm.setup {
  -- init_options = {initializationOptions},
  cmd = { DATA_PATH .. "/lspinstall/efm/efm-langserver" },
  init_options = { documentFormatting = true, codeAction = false },
  filetypes = { "python" },
  settings = {
    rootMarkers = { ".git/", "requirements.txt" },
    languages = {
      python = python_arguments,
    },
  },
}

-- npm i -g pyright
require("lspconfig").pyright.setup {
  cmd = {
    DATA_PATH .. "/lspinstall/python/node_modules/.bin/pyright-langserver",
    "--stdio",
  },
  on_attach = require("lsp").common_on_attach,
  handlers = {
    ["textDocument/publishDiagnostics"] = vim.lsp.with(vim.lsp.diagnostic.on_publish_diagnostics, {
      virtual_text = O.lang.python.diagnostics.virtual_text,
      signs = O.lang.python.diagnostics.signs,
      underline = O.lang.python.diagnostics.underline,
      update_in_insert = true,
    }),
  },
  settings = {
    python = {
      analysis = {
        typeCheckingMode = O.lang.python.analysis.type_checking,
        autoSearchPaths = O.lang.python.analysis.auto_search_paths,
        useLibraryCodeForTypes = O.lang.python.analysis.use_library_code_types,
      },
    },
  },
}
if O.lang.python.autoformat then
  require("lv-utils").define_augroups {
    _python_autoformat = {
      {
        "BufWritePre",
        "*.py",
        "lua vim.lsp.buf.formatting_sync(nil, 1000)",
      },
    },
  }
end