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

(py) Update configurations (#51)

Browse Source
This commit is contained in:
danymat
2022-01-29 19:13:41 +01:00
parent 6c086dcb2f
commit 0d00e05b20
1 changed files with 18 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -55,16 +55,14 @@ return {
subtree = { subtree = {
{ {
retrieve = "all", retrieve = "all",
node_type = "true|false|integer|binary_operator|expression_list", node_type = "true|false|integer|binary_operator|expression_list|call",
as = "anonymous_return", as = "anonymous_return",
recursive = true,
extract = true, extract = true,
}, },
{ {
retrieve = "all", retrieve = "all",
node_type = "identifier", node_type = "identifier",
as = "return_statement", as = "return_statement",
recursive = true,
extract = true, extract = true,
}, },
}, },
@@ -96,14 +94,24 @@ return {
if nodes["anonymous_return"] then if nodes["anonymous_return"] then
local _res = extractors:extract_from_matched(nodes, { type = true }) local _res = extractors:extract_from_matched(nodes, { type = true })
results.anonymous_return = _res.anonymous_return res.anonymous_return = _res.anonymous_return
end
-- Return type hints takes precedence over all other types for generating template
if res["return_type"] then
res["return_statement"] = nil
res["anonymous_return"] = nil
end end
results.has_identifier = (res.typed_parameter or res.identifier) and { true } or nil results.has_identifier = (res.typed_parameter or res.identifier) and { true } or nil
results.type = res.type results.type = res.type
results.parameters = res.identifier results.parameters = res.identifier
results.anonymous_return = res.anonymous_return
results.return_statement = res.return_statement results.return_statement = res.return_statement
results.has_return = (res.return_statement or res.anonymous_return) and { true } or nil results.return_type = res.return_type
results.has_return = (res.return_statement or res.anonymous_return or res.return_type)
and { true }
or nil
return results return results
end, end,
}, },
@@ -264,6 +272,11 @@ return {
{ "has_return", "", { type = { "func" } } }, { "has_return", "", { type = { "func" } } },
{ "has_return", "Returns", { type = { "func" } } }, { "has_return", "Returns", { type = { "func" } } },
{ "has_return", "-------", { type = { "func" } } }, { "has_return", "-------", { type = { "func" } } },
{
"return_type",
"%s",
{ after_each = "\t$1" },
},
{ {
"return_statement", "return_statement",
"%s : $1", "%s : $1",