mathdatech/tootbot
1
0
Fork 0
mirror of https://github.com/cquest/tootbot.git synced 2024-11-23 20:11:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

handle local path redirects

Browse Source
This commit is contained in:
cquest 2023-01-15 09:35:04 +01:00
parent e3292c74bd
commit 1165374824
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tootbot.py
View File

@ -18,7 +18,11 @@ import requests
def unredir(redir): def unredir(redir):
r = requests.get(redir, allow_redirects=False) r = requests.get(redir, allow_redirects=False)
while r.status_code in {301, 302}: while r.status_code in {301, 302}:
redir = r.headers.get('Location') if 'http' not in r.headers.get('Location'):
redir = re.sub(r'(https?://.*)/.*', r'\1', redir) + \
r.headers.get('Location')
else:
redir = r.headers.get('Location')
print('redir', redir) print('redir', redir)
if '//ow.ly/' in redir or '//bit.ly/' in redir: if '//ow.ly/' in redir or '//bit.ly/' in redir:
redir = redir.replace('https://ow.ly/', 'http://ow.ly/') # only http redir = redir.replace('https://ow.ly/', 'http://ow.ly/') # only http