neolooong/whichpy.nvim

github

programming-languages-support

32

1

1

3

CREATED

2024-05-09

UPDATED

2 months ago

---

whichpy.nvim

Yet another python interpreter selector plugin for neovim. Make LSPs (pyright, pylsp, basedpyright) and DAP work with specific python.

https://github.com/user-attachments/assets/bddd568a-947a-49d2-a403-efae2787f60a

Features

• Support multiple Lsp servers. (Pylsp, Pyright, BasedPyright)
• Support nvim-dap-python.
• Switch between python interpreters without restart LSPs. (Except WhichPy reset on Pyright)
• Support multiple pickers. (builtin, fzf-lua, telescope)
• Automatically select the previously chosen interpreter based on the directory.
• Search on common directories, currently support:
  • workspace (lsp.WorkspaceFolder or vim.fs.root or vim.fn.getcwd)
  • global (vim.env.Path and common posix paths)
  • global virtual environment
  • pyenv
  • poetry
  • pdm
  • conda
  • uv

Requirements

• Neovim >= 0.10.0

Installation

• Using lazy.nvim:

{
  "neolooong/whichpy.nvim",
  dependencies = {
    -- optional for dap
    -- "mfussenegger/nvim-dap-python",
    -- optional for picker support
    -- "ibhagwan/fzf-lua",
    -- "nvim-telescope/telescope.nvim",
  }
  opts = {},
}

Configuration

{
  cache_dir = vim.fn.stdpath("cache") .. "/whichpy.nvim",
  update_path_env = false,  -- Whether to modify $PATH when switching interpreters.
  after_handle_select = nil,  -- Equivalent to venv-selector.nvim's on_venv_activate_callback()
  -- after_handle_select = function(selected) vim.print(selected) end,
  picker = {
    name = "builtin",  -- must be one of ("builtin", "fzf-lua", "telescope")
    -- You can customize the picker as follows. For available options, refer to the respective documentation.
    -- ["fzf-lua"] = {
    --   prompt="fzf-lua",
    -- },
    -- telescope = {
    --   prompt_title="telescope",
    -- },
    -- builtin = {
    --   prompt="vim.ui.select",
    -- },
  },
  locator = {
    -- You can disable locator like this
    -- locator_name = { enable = false },
    workspace = {
      display_name = "Workspace",
      search_pattern = ".*env.*", -- `:help lua-patterns`
      depth = 2,
      ignore_dirs = {
        ".git",
        ".mypy_cache",
        ".pytest_cache",
        ".ruff_cache",
        "__pycache__",
        "__pypackages__",
      },
    },
    global = {
      display_name = "Global",
    },
    global_virtual_environment = {
      display_name = "Global Virtual Environment",
      dirs = {
        -- accept following structure
        -- path
        -- { path, vim.uv.os_uname().sysname }
        "~/envs",
        "~/.direnv",
        "~/.venvs",
        "~/.virtualenvs",
        "~/.local/share/virtualenvs",
        { "~/Envs", "Windows_NT" },  -- only search on Windows
        vim.env.WORKON_HOME,
      }
    },
    pyenv = {
      display_name = "Pyenv",
      venv_only = true,
    },
    poetry = {
      display_name = "Poetry",
    },
    pdm = {
      display_name = "PDM",
    },
    conda = {
      display_name = "Conda",
    },
    -- uv = {  -- disabled by default
    --   display_name = "uv",
    -- },
  },
  lsp = {
    pylsp = require("whichpy.lsp.handlers.pylsp").new(),
    pyright = require("whichpy.lsp.handlers.pyright").new(),
    basedpyright = require("whichpy.lsp.handlers.pyright").new(),
  },
}

Commands

This plugin provide these commands:

:WhichPy select [path?]

If path provided, LSPs would be configured. Otherwise, picker would show up.

:WhichPy reset

Reset LSPs configuration, and clear the cache.

:WhichPy rescan

Search python interpreter again.

FAQ

{
  "nvim-neotest/neotest",
  dependencies = {
    "nvim-neotest/nvim-nio",
    "nvim-neotest/neotest-python",
  },
  config = function()
    local python_adapter = require("neotest-python")({
      python = function()
        local whichpy_python = require("whichpy.envs").current_selected()
        if whichpy_python then
          return whichpy_python
        end
        return require("neotest-python.base").get_python_command
      end,
    })
    require("neotest").setup({
      adapters = { python_adapter },
    })
  end,
}

1. I'm not familiar with fd. (main reason)
2. I only want to search a specific directory. vim.uv.fs_stat is fast enough for me.

Once I'm become more familiar with fd and have free time. I'll try.

lualine component

A lualine component is included which shows the current env name. To use it, simple include "whichpy" in your lualine config.

eg.

require("lualine").setup({
  sections = {
    lualine_x = {
      "whichpy"
    },
  },
})

Acknowledgments

• venv-selector.nvim for inspiring the Python path setup.
• vscode-python for numerous locator implementation.
• and many publicly shared dotfiles on Github.