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

(py) Update configurations

Browse Source
This commit is contained in:
danymat
2022-01-13 09:35:41 +01:00
parent 7e7fbc4ff2
commit a79fa586a9
2 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -51,13 +51,19 @@ return {
{ retrieve = "all", node_type = "return_statement", extract = true },
},
},
{
retrieve = "all",
node_type = "type",
as = "return_statement",
extract = true,
},
}
local nodes = neogen.utilities.nodes:matching_nodes_from(node, tree)
local res = neogen.utilities.extractors:extract_from_matched(nodes)
results.type = res.type
results.parameters = res.identifier
results.return_statement = res.return_statement
results.return_statement = res.return_statement and { res.return_statement[1] } or nil
return results
end,
},

2
lua/neogen/utilities/nodes.lua
View File

@@ -87,7 +87,7 @@ neogen.utilities.nodes = {
for _, child in pairs(matched) do
if subtree.extract == true then
local name = subtree.node_type or "_"
local name = subtree.as and subtree.as or (subtree.node_type or "_")
if result[name] == nil then
result[name] = {}
end