From d2447d25105a1507796e5a2301a8d31f28ddb632 Mon Sep 17 00:00:00 2001 From: Akash <64683866+akamhy@users.noreply.github.com> Date: Fri, 17 Jul 2020 20:48:10 +0530 Subject: [PATCH] Update wrapper.py --- waybackpy/wrapper.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/waybackpy/wrapper.py b/waybackpy/wrapper.py index a324b09..fa694cc 100644 --- a/waybackpy/wrapper.py +++ b/waybackpy/wrapper.py @@ -1,7 +1,8 @@ # -*- coding: utf-8 -*- +import re import sys -import json, re +import json from datetime import datetime from waybackpy.exceptions import WaybackError @@ -56,9 +57,9 @@ class Url(): def handle_HTTPError(self, e): if e.code >= 500: raise WaybackError(e) from None - elif e.code == 429: + if e.code == 429: raise WaybackError(e) from None - elif e.code == 404: + if e.code == 404: raise HTTPError(e) from None def save(self):