leigh/neogen
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(py) Always get public parameters for classes

Browse Source
This commit is contained in:
danymat
2022-01-29 13:06:53 +01:00
parent fb707c0fd3
commit fe44b20a57
1 changed files with 20 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
lua/neogen/configurations/python.lua
View File

@@ -108,25 +108,33 @@ return {
}, },
class = { class = {
["class_definition"] = { ["class_definition"] = {
["2"] = { ["0"] = {
match = "block",
extract = function(node) extract = function(node)
local results = {} local results = {}
local tree = { local tree = {
{ {
retrieve = "first", retrieve = "first",
node_type = "function_definition", node_type = "block",
subtree = { subtree = {
{ {
retrieve = "first", retrieve = "first",
node_type = "block", node_type = "function_definition",
subtree = { subtree = {
{ {
retrieve = "all", retrieve = "first",
node_type = "expression_statement", node_type = "block",
subtree = { subtree = {
{ retrieve = "first", node_type = "assignment", extract = true }, {
retrieve = "all",
node_type = "expression_statement",
subtree = {
{
retrieve = "first",
node_type = "assignment",
extract = true,
},
},
},
}, },
}, },
}, },
@@ -145,7 +153,10 @@ return {
for _, assignment in pairs(nodes["assignment"]) do for _, assignment in pairs(nodes["assignment"]) do
local left_side = assignment:field("left")[1] local left_side = assignment:field("left")[1]
local left_attribute = left_side:field("attribute")[1] local left_attribute = left_side:field("attribute")[1]
table.insert(results.attributes, ts_utils.get_node_text(left_attribute)[1]) left_attribute = ts_utils.get_node_text(left_attribute)[1]
if not vim.startswith(left_attribute, "_") then
table.insert(results.attributes, left_attribute)
end
end end
return results return results