extended MastodonServerError for all 5xx error codes

1 files changed, 3 insertions, 2 deletions

diff --git a/mastodon/Mastodon.py b/mastodon/Mastodon.py
index 65ce518..bfc2a9b 100644
--- a/mastodon/Mastodon.py
+++ b/mastodon/Mastodon.py
@@ -2750,7 +2750,8 @@ class Mastodon:
                         # on any 404
                 elif response_object.status_code == 401:
                     ex_type = MastodonUnauthorizedError
-                elif response_object.status_code == 502:
+                elif response_object.status_code >= 500 and \
+                     response_object.status_code <= 511:
                     ex_type = MastodonServerError
                 else:
                     ex_type = MastodonAPIError
@@ -3068,7 +3069,7 @@ class MastodonAPIError(MastodonError):
     pass
 
 class MastodonServerError(MastodonError):
-    """Raised if the Server is malconfigured, e.g. returns a 502 error code"""
+    """Raised if the Server is malconfigured and returns a 5xx error code"""
     pass
 
 class MastodonNotFoundError(MastodonAPIError):