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-12 09:41:27 +01:00
parent 30c693c1cb
commit 402cc7b52e
1 changed files with 13 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -32,7 +32,10 @@ return {
{ {
retrieve = "all", retrieve = "all",
node_type = "typed_parameter", node_type = "typed_parameter",
subtree = { { retrieve = "all", node_type = "identifier", extract = true } }, subtree = {
{ retrieve = "all", node_type = "identifier", extract = true },
{ retrieve = "all", node_type = "type", extract = true },
},
}, },
{ {
retrieve = "all", retrieve = "all",
@@ -52,6 +55,7 @@ return {
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.parameters = res.identifier results.parameters = res.identifier
results.return_statement = res.return_statement results.return_statement = res.return_statement
return results return results
@@ -130,7 +134,7 @@ return {
generator = nil, generator = nil,
template = { 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) 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) use_default_comment = false, -- If you want to prefix the template with the default comment for the language, e.g for python: # (default_generator)
position = function(node, type) position = function(node, type)
@@ -162,7 +166,7 @@ return {
{ nil, "# $1", { no_results = true, type = { "type" } } }, { nil, "# $1", { no_results = true, type = { "type" } } },
{ nil, '"""$1' }, { nil, '"""$1' },
{ "parameters", "\t%s: $1", { before_first_item = { "", "Args:" } } }, { { "parameters", "type" }, "\t%s (%s): $1", { before_first_item = { "", "Args:" }, type = { "func" } } },
{ "attributes", "\t%s: $1", { before_first_item = { "", "Attributes: " } } }, { "attributes", "\t%s: $1", { before_first_item = { "", "Attributes: " } } },
{ "return_statement", "\t$1", { before_first_item = { "", "Returns: " } } }, { "return_statement", "\t$1", { before_first_item = { "", "Returns: " } } },
{ nil, '"""' }, { nil, '"""' },
@@ -178,7 +182,12 @@ return {
{ nil, "# $1", { no_results = true, type = { "type" } } }, { nil, "# $1", { no_results = true, type = { "type" } } },
{ nil, '"""$1' }, { nil, '"""$1' },
{ "parameters", "%s: $1", { before_first_item = { "", "Parameters", "----------" } } }, {
{ "parameters", "type" },
"%s: %s",
{ before_first_item = { "", "Parameters", "----------" }, type = { "func" } },
},
{ { "parameters", "type" }, "\t$1", { required = "type" } },
{ "attributes", "%s: $1", { before_first_item = { "", "Attributes", "----------" } } }, { "attributes", "%s: $1", { before_first_item = { "", "Attributes", "----------" } } },
{ "return_statement", "$1", { before_first_item = { "", "Returns", "-------" } } }, { "return_statement", "$1", { before_first_item = { "", "Returns", "-------" } } },
{ nil, '"""' }, { nil, '"""' },