Benjamin_Loison/waybackpy
1
0
Fork 0
Code Issues 1 Pull Requests Packages Projects Releases Wiki Activity

Update wrapper.py

Browse Source
This commit is contained in:
Akash
2020-07-19 19:50:23 +05:30
committed by GitHub
parent 6f82041ec9
commit 0ac4689944
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
waybackpy/wrapper.py
View File

@@ -79,10 +79,16 @@ class Url():
except Exception as e: except Exception as e:
raise WaybackError(e) raise WaybackError(e)
header = response.headers header = response.headers
try: try:
arch = re.search(r"rel=\"memento.*?web\.archive\.org(/web/[0-9]{14}/.*?)>", str(header)).group(1) arch = re.search(r"rel=\"memento.*?web\.archive\.org(/web/[0-9]{14}/.*?)>", str(header))
if arch:
arch = arch.group(1)
else:
raise WaybackError("Can't create a new archive. No archive url found in reply headers.")
except KeyError as e: except KeyError as e:
raise WaybackError(e) raise WaybackError(e)
return "https://web.archive.org" + arch return "https://web.archive.org" + arch
def get(self, url=None, user_agent=None, encoding=None): def get(self, url=None, user_agent=None, encoding=None):