fix: mypy errors

jnoortheen · Apr 18, 2020 · 09111b0 · 09111b0
1 parent 1de30cd
commit 09111b0
Showing 1 changed file with 16 additions and 15 deletions.
diff --git a/arger/parser/funcs.py b/arger/parser/funcs.py
@@ -26,21 +26,22 @@ def get_param(param):
 
 def create_option(param: Param, default, option_generator):
     if isinstance(default, Option):
-        option = default
-        option.kwargs.setdefault('help', param.help)
-        option.update_flags(param.name, option_generator)
-        if 'type' not in option.kwargs:
-            option.update(param.type)
-    elif isinstance(default, Argument):
-        option = default
-        option.kwargs.setdefault('help', param.help)
-        option.update_flags(param.name)
-        if 'type' not in option.kwargs:
-            option.update(param.type)
-    else:
-        option = Option(help=param.help)
-        option.update_flags(param.name, option_generator)
-        option.update(param.type, default)
+        default.kwargs.setdefault('help', param.help)
+        default.update_flags(param.name, option_generator)
+        if 'type' not in default.kwargs:
+            default.update(param.type)
+        return default
+
+    if isinstance(default, Argument):
+        default.kwargs.setdefault('help', param.help)
+        default.update_flags(param.name)
+        if 'type' not in default.kwargs:
+            default.update(param.type)
+        return default
+
+    option = Option(help=param.help)
+    option.update_flags(param.name, option_generator)
+    option.update(param.type, default)
     return option