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 = "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 nodes = neogen.utilities.nodes:matching_nodes_from(node, tree)
local res = neogen.utilities.extractors:extract_from_matched(nodes) local res = neogen.utilities.extractors:extract_from_matched(nodes)
results.type = res.type results.type = res.type
results.parameters = res.identifier 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 return results
end, end,
}, },

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

@@ -87,7 +87,7 @@ neogen.utilities.nodes = {
for _, child in pairs(matched) do for _, child in pairs(matched) do
if subtree.extract == true then 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 if result[name] == nil then
result[name] = {} result[name] = {}
end end