2023-06-19 15:59:04 +08:00
|
|
|
local util = require 'lspconfig.util'
|
|
|
|
|
|
|
|
local root_files = {
|
|
|
|
'pyproject.toml',
|
|
|
|
'setup.py',
|
|
|
|
'setup.cfg',
|
|
|
|
'requirements.txt',
|
|
|
|
'Pipfile',
|
|
|
|
'pyrightconfig.json',
|
2024-05-21 12:31:14 +08:00
|
|
|
'.git',
|
2023-06-19 15:59:04 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
local function organize_imports()
|
|
|
|
local params = {
|
|
|
|
command = 'pyright.organizeimports',
|
|
|
|
arguments = { vim.uri_from_bufnr(0) },
|
|
|
|
}
|
2024-05-21 12:31:14 +08:00
|
|
|
|
|
|
|
local clients = util.get_lsp_clients {
|
|
|
|
bufnr = vim.api.nvim_get_current_buf(),
|
|
|
|
name = 'pyright',
|
|
|
|
}
|
|
|
|
for _, client in ipairs(clients) do
|
|
|
|
client.request('workspace/executeCommand', params, nil, 0)
|
|
|
|
end
|
2023-06-19 15:59:04 +08:00
|
|
|
end
|
|
|
|
|
|
|
|
local function set_python_path(path)
|
2024-05-21 12:31:14 +08:00
|
|
|
local clients = util.get_lsp_clients {
|
2023-06-19 15:59:04 +08:00
|
|
|
bufnr = vim.api.nvim_get_current_buf(),
|
|
|
|
name = 'pyright',
|
|
|
|
}
|
|
|
|
for _, client in ipairs(clients) do
|
2024-05-21 12:31:14 +08:00
|
|
|
if client.settings then
|
|
|
|
client.settings.python = vim.tbl_deep_extend('force', client.settings.python, { pythonPath = path })
|
|
|
|
else
|
|
|
|
client.config.settings = vim.tbl_deep_extend('force', client.config.settings, { python = { pythonPath = path } })
|
|
|
|
end
|
2023-06-19 15:59:04 +08:00
|
|
|
client.notify('workspace/didChangeConfiguration', { settings = nil })
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
return {
|
|
|
|
default_config = {
|
|
|
|
cmd = { 'pyright-langserver', '--stdio' },
|
|
|
|
filetypes = { 'python' },
|
2024-05-21 12:31:14 +08:00
|
|
|
root_dir = function(fname)
|
|
|
|
return util.root_pattern(unpack(root_files))(fname)
|
|
|
|
end,
|
2023-06-19 15:59:04 +08:00
|
|
|
single_file_support = true,
|
|
|
|
settings = {
|
|
|
|
python = {
|
|
|
|
analysis = {
|
|
|
|
autoSearchPaths = true,
|
|
|
|
useLibraryCodeForTypes = true,
|
2024-05-21 12:31:14 +08:00
|
|
|
diagnosticMode = 'openFilesOnly',
|
2023-06-19 15:59:04 +08:00
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
commands = {
|
|
|
|
PyrightOrganizeImports = {
|
|
|
|
organize_imports,
|
|
|
|
description = 'Organize Imports',
|
|
|
|
},
|
|
|
|
PyrightSetPythonPath = {
|
|
|
|
set_python_path,
|
|
|
|
description = 'Reconfigure pyright with the provided python path',
|
|
|
|
nargs = 1,
|
|
|
|
complete = 'file',
|
|
|
|
},
|
|
|
|
},
|
|
|
|
docs = {
|
|
|
|
description = [[
|
|
|
|
https://github.com/microsoft/pyright
|
|
|
|
|
|
|
|
`pyright`, a static type checker and language server for python
|
|
|
|
]],
|
|
|
|
},
|
|
|
|
}
|