mathdatech/tootbot
1
0
Fork 0
mirror of https://github.com/cquest/tootbot.git synced 2025-02-23 18:58:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

Compare commits

base: mathdatech:79f922af2451b09c9cf23093ae8605dc47186f57
Branches Tags
mathdatech:master
...
compare: mathdatech:887474a1f33a46396f06ebfa0f45a873a05f81fe
Branches Tags
mathdatech:master

2 Commits
79f922af24 ... 887474a1f3

Author SHA1 Message Date
cquest
887474a1f3 do not download very large video files ! 2023-02-09 16:51:53 +01:00
cquest
fba2318929 avoid redirect inifinite loops 2023-02-09 16:51:31 +01:00
1 changed files with 5 additions and 1 deletions
Show Stats Expand all files Collapse all files

6
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')
@ -286,7 +290,7 @@ else:
c = c.replace(l, '') c = c.replace(l, '')
video = redir video = redir
print('video:', video) print('video:', video)
subprocess.run('rm -f out.mp4; yt-dlp -N 8 -o out.mp4 --recode-video mp4 --no-playlist %s' % subprocess.run('rm -f out.mp4; yt-dlp -N 8 -o out.mp4 --recode-video mp4 --no-playlist %s --max-filesize 100M' %
(video,), shell=True, capture_output=False) (video,), shell=True, capture_output=False)
print("received") print("received")
try: try: