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

avoid redirect inifinite loops

Browse Source
This commit is contained in:
cquest 2023-02-09 16:51:31 +01:00
parent 79f922af24
commit fba2318929
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tootbot.py
View File

@ -17,7 +17,11 @@ import requests
def unredir(redir): def unredir(redir):
r = requests.get(redir, allow_redirects=False) r = requests.get(redir, allow_redirects=False)
redir_count = 0
while r.status_code in {301, 302}: while r.status_code in {301, 302}:
redir_count = redir_count + 1
if redir_count > 10:
break
if 'http' not in r.headers.get('Location'): if 'http' not in r.headers.get('Location'):
redir = re.sub(r'(https?://.*)/.*', r'\1', redir) + \ redir = re.sub(r'(https?://.*)/.*', r'\1', redir) + \
r.headers.get('Location') r.headers.get('Location')