From 6dc6124dc4c70415f849717390637dbee82e9b7a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jon=C3=A1=C5=A1=20Jan=C4=8Da=C5=99=C3=ADk?=
 <jonas.jancarik@gmail.com>
Date: Fri, 3 Sep 2021 04:34:36 +0200
Subject: [PATCH] Raise error on a 509 response (too many sessions) (#99)

* Raise error on a 509 response (too many sessions)

When the response code is 509, raise an error with an explanation (based on the actual error message contained in the response HTML).

* Raise error on a 509 response (too many sessions) - linting
---
 waybackpy/wrapper.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/waybackpy/wrapper.py b/waybackpy/wrapper.py
index 6b68d1c..173b0df 100644
--- a/waybackpy/wrapper.py
+++ b/waybackpy/wrapper.py
@@ -228,6 +228,14 @@ class Url:
             response=response,
         )
 
+        if response.status_code == 509:
+            raise WaybackError(
+                "Can not save '{url}'. You have probably reached the limit of active "
+                "sessions. Try later.".format(
+                    url=_cleaned_url(self.url), text=response.text
+                )
+            )
+
         m = re.search(
             r"https?://web.archive.org/web/([0-9]{14})/http", self._archive_url
         )