If exception occures - set concurrency as available

2020-12-17 18:02:14 +02:00 · 2020-12-17 18:02:14 +02:00 · 9646d112d2
commit 9646d112d2
parent a09c37a065
1 changed files with 5 additions and 1 deletions
--- a/erepublik/utils.py
+++ b/erepublik/utils.py
@ -397,7 +397,11 @@ def wait_for_lock(function):
            return None
        else:
            instance.concurrency_available.clear()
-            ret = function(instance, *args, **kwargs)
+            try:
+                ret = function(instance, *args, **kwargs)
+            except Exception as e:
+                instance.concurrency_available.set()
+                raise e
            instance.concurrency_available.set()
            return ret
    return wrapper