fix extensions moar

feature-unify_handlers
Tomáš Mládek 2018-01-31 14:26:51 +01:00 committed by Tomáš Mládek
parent a949c82bce
commit 9abf01b02a
1 changed files with 5 additions and 2 deletions

View File

@ -50,9 +50,12 @@ def download_ydl(urls, date):
def download_raw(url, date):
ext = ''
if 'api.telegram.org' in url and 'animation' in url and "mp4" not in url:
if 'api.telegram.org' in url \
and ('animation' in url or 'video' in url) \
and "mp4" not in url:
ext = '.mp4'
else:
ext = ''
local_filename = OUT_DIR + '/' + datestr(date) + '__' + url.split('/')[-1] + ext
r = requests.get(url, stream=True)
with open(local_filename, 'wb') as f: