How can I enable showing virtual environment for python mode?

[mizhozan]

I am using lualine and currently learning it. I want the statusline show me the activated virtualenv if any. Unfortunately, the documentation seems to be in progress and I couldn't find a relevant information regarding this. Does lualine support this funcationality?

[fitrh]

If you are using venv, you can use py_lsp.nvim to create custom component

[aemonge]

If you created the custom componet, can you share it?

[lkhphuc]

My lazy.nvim components

local function get_venv(variable)
  local venv = os.getenv(variable)
  if venv ~= nil and string.find(venv, "/") then
    local orig_venv = venv
    for w in orig_venv:gmatch("([^/]+)") do
      venv = w
    end
    venv = string.format("%s", venv)
  end
  return venv
end

{
    "lualine.nvim",
    opts = function(_, opts)
      table.insert(opts.sections.lualine_y, {
        function()
          local venv = get_venv("CONDA_DEFAULT_ENV") or get_venv("VIRTUAL_ENV") or "NO ENV"
          return " " .. venv
        end,
        cond = function() return vim.bo.filetype == "python" end,
      })
    end
  }

[rodhash]

thanks