fix: Add Hub.flush as well

untitaker · untitaker · commit 8418fd18bda6 · 2019-02-15T11:16:21.000+01:00
diff --git a/sentry_sdk/api.py b/sentry_sdk/api.py
@@ -85,8 +85,8 @@ def inner():
 @hubmethod
 def flush(timeout=None, callback=None):
     hub = Hub.current
-    if hub is not None and hub.client is not None:
-        return hub.client.flush(timeout=timeout, callback=callback)
+    if hub is not None:
+        return hub.flush(timeout=timeout, callback=callback)
 
 
 @hubmethod
diff --git a/sentry_sdk/hub.py b/sentry_sdk/hub.py
@@ -338,6 +338,12 @@ def inner():
 
         return inner()
 
+    def flush(self, timeout=None, callback=None):
+        """Alias for self.client.flush"""
+        client, scope = self._stack[-1]
+        if client is not None:
+            return client.flush(timeout=timeout, callback=callback)
+
 
 GLOBAL_HUB = Hub()
 _local.set(GLOBAL_HUB)
