feat(async): Make on_http_error decorator work with coroutines

vishes-shell · vishes-shell · commit 0591787dc46c · 2020-02-27T20:25:25.000+03:00
on_http_error should work with sync functions that return coroutines and
handle errors when coroutines will be awaited
diff --git a/gitlab/exceptions.py b/gitlab/exceptions.py
@@ -15,6 +15,7 @@
 # You should have received a copy of the GNU Lesser General Public License
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
+import asyncio
 import functools
 
 
@@ -262,11 +263,22 @@ def on_http_error(error):
 
     def wrap(f):
         @functools.wraps(f)
-        async def wrapped_f(*args, **kwargs):
+        def wrapped_f(*args, **kwargs):
             try:
-                return await f(*args, **kwargs)
+                result = f(*args, **kwargs)
             except GitlabHttpError as e:
                 raise error(e.error_message, e.response_code, e.response_body)
+            else:
+                if not asyncio.iscoroutine(result):
+                    return result
+
+                async def awaiter():
+                    try:
+                        await result
+                    except GitlabHttpError as e:
+                        raise error(e.error_message, e.response_code, e.response_body)
+
+                return awaiter
 
         return wrapped_f
 
