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

Removed return if body is not found (python)

Browse Source
This commit is contained in:
Daniel Mathiot
2021-08-23 20:03:14 +02:00
parent db3a136f7f
commit 790c98edf1
1 changed files with 46 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -10,21 +10,28 @@ return {
["0"] = {
extract = function (node)
local results = {
parameters = {}
parameters = {},
return_statement = {}
}
local params = neogen.utilities.nodes:matching_child_nodes(node, "parameters")[1]
if #params == 0 then
results.parameters = nil
end
local found_nodes
if #params ~= 0 then
-- Find regular parameters
local regular_params = neogen.utilities.extractors:extract_children_text("identifier")(params)
if #regular_params == 0 then
regular_params = nil
end
for _, _params in pairs(regular_params) do
table.insert(results.parameters, _params)
end
results.parameters = regular_params
-- Find regular optional parameters
@@ -48,18 +55,18 @@ return {
table.insert(results.parameters, _params)
end
end
local body = neogen.utilities.nodes:matching_child_nodes(node, "block")[1]
if body == nil then
return
end
if body ~= nil then
local return_statement = neogen.utilities.nodes:matching_child_nodes(body, "return_statement")
if #return_statement == 0 then
return_statement = nil
end
results.return_statement = return_statement
end
return results
end
@@ -108,7 +115,7 @@ return {
generator = nil,
template = {
annotation_convention = "google_docstrings", -- required: Which annotation convention to use (default_generator)
annotation_convention = "numpydoc", -- required: Which annotation convention to use (default_generator)
append = { position = "after", child_name = "block" }, -- optional: where to append the text (default_generator)
use_default_comment = false, -- If you want to prefix the template with the default comment for the language, e.g for python: # (default_generator)
google_docstrings = {