(py) Update configurations

2022-01-29 18:50:59 +01:00
parent 10f5d4e5e5
commit 6c086dcb2f
1 changed files with 4 additions and 1 deletions
--- a/lua/neogen/configurations/python.lua
+++ b/lua/neogen/configurations/python.lua
@@ -157,10 +157,13 @@ return {
                            local left_side = assignment:field("left")[1]
                            local left_attribute = left_side:field("attribute")[1]
                            left_attribute = ts_utils.get_node_text(left_attribute)[1]
-                            if not vim.startswith(left_attribute, "_") then
+                            if left_attribute and not vim.startswith(left_attribute, "_") then
                                table.insert(results.attributes, left_attribute)
                            end
                        end
+                        if vim.tbl_isempty(results.attributes) then
+                            results.attributes = nil
+                        end

                        return results
                    end,