Factorize _update_args_kwargs logic in Parser

marshmallow-code · Feb 25, 2020 · ba59660 · ba59660
1 parent da9f4fc
commit ba59660
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 20 deletions.
diff --git a/src/webargs/asyncparser.py b/src/webargs/asyncparser.py
@@ -147,11 +147,9 @@ async def wrapper(*args, **kwargs):
                         error_status_code=error_status_code,
                         error_headers=error_headers,
                     )
-                    if as_kwargs:
-                        kwargs.update(parsed_args)
-                    else:
-                        # Add parsed_args after other positional arguments
-                        args += (parsed_args,)
+                    args, kwargs = self._update_args_kwargs(
+                        args, kwargs, parsed_args, as_kwargs
+                    )
                     return await func(*args, **kwargs)
 
             else:
@@ -171,11 +169,9 @@ def wrapper(*args, **kwargs):
                         error_status_code=error_status_code,
                         error_headers=error_headers,
                     )
-                    if as_kwargs:
-                        kwargs.update(parsed_args)
-                    else:
-                        # Add parsed_args after other positional arguments
-                        args += (parsed_args,)
+                    args, kwargs = self._update_args_kwargs(
+                        args, kwargs, parsed_args, as_kwargs
+                    )
                     return func(*args, **kwargs)
 
             return wrapper

diff --git a/src/webargs/core.py b/src/webargs/core.py
@@ -290,6 +290,16 @@ def get_request_from_view_args(self, view, args, kwargs):
         """
         return None
 
+    @staticmethod
+    def _update_args_kwargs(args, kwargs, parsed_args, as_kwargs):
+        """Update args or kwargs with parsed_args depending on as_kwargs"""
+        if as_kwargs:
+            kwargs.update(parsed_args)
+        else:
+            # Add parsed_args after other positional arguments
+            args += (parsed_args,)
+        return args, kwargs
+
     def use_args(
         self,
         argmap,
@@ -349,11 +359,9 @@ def wrapper(*args, **kwargs):
                     error_status_code=error_status_code,
                     error_headers=error_headers,
                 )
-                if as_kwargs:
-                    kwargs.update(parsed_args)
-                else:
-                    # Add parsed_args after other positional arguments
-                    args += (parsed_args,)
+                args, kwargs = self._update_args_kwargs(
+                    args, kwargs, parsed_args, as_kwargs
+                )
                 return func(*args, **kwargs)
 
             wrapper.__wrapped__ = func

diff --git a/src/webargs/pyramidparser.py b/src/webargs/pyramidparser.py
@@ -159,11 +159,9 @@ def wrapper(obj, *args, **kwargs):
                     error_status_code=error_status_code,
                     error_headers=error_headers,
                 )
-                if as_kwargs:
-                    kwargs.update(parsed_args)
-                else:
-                    # Add parsed_args after other positional arguments
-                    args += (parsed_args,)
+                args, kwargs = self._update_args_kwargs(
+                    args, kwargs, parsed_args, as_kwargs
+                )
                 return func(obj, *args, **kwargs)
 
             wrapper.__wrapped__ = func