From c74f24d7336d757354b2c122d503fbddb5608263 Mon Sep 17 00:00:00 2001 From: Tyler Donaldson Date: Thu, 26 Mar 2020 11:45:03 -0700 Subject: [PATCH] [Closes #537] Return Action when calling add_argument --- gooey/python_bindings/gooey_parser.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/gooey/python_bindings/gooey_parser.py b/gooey/python_bindings/gooey_parser.py index 1ecb8fa..e431a2e 100644 --- a/gooey/python_bindings/gooey_parser.py +++ b/gooey/python_bindings/gooey_parser.py @@ -23,10 +23,11 @@ class GooeyArgumentGroup(_ArgumentGroup): widget = kwargs.pop('widget', None) metavar = kwargs.pop('metavar', None) options = kwargs.pop('gooey_options', None) - super(GooeyArgumentGroup, self).add_argument(*args, **kwargs) + action = super(GooeyArgumentGroup, self).add_argument(*args, **kwargs) self.parser._actions[-1].metavar = metavar self.widgets[self.parser._actions[-1].dest] = widget self.options[self.parser._actions[-1].dest] = options + return action def add_argument_group(self, *args, **kwargs): options = kwargs.pop('gooey_options', {}) @@ -85,7 +86,7 @@ class GooeyParser(object): metavar = kwargs.pop('metavar', None) options = kwargs.pop('gooey_options', None) - self.parser.add_argument(*args, **kwargs) + action = self.parser.add_argument(*args, **kwargs) self.parser._actions[-1].metavar = metavar self.widgets[self.parser._actions[-1].dest] = widget self.options[self.parser._actions[-1].dest] = options @@ -95,6 +96,7 @@ class GooeyParser(object): options or {}, **kwargs ) + return action def add_mutually_exclusive_group(self, *args, **kwargs): options = kwargs.pop('gooey_options', {})