blankie
/
redditbot
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add thumbnails

This commit is contained in:
blank X 2021-10-23 11:34:55 +07:00
parent 25d9251208
commit 2b399d4bd5
Signed by: blankie
GPG Key ID: CC15FC822C7F61F5
1 changed files with 24 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
redditbot.py
View File

@ -212,6 +212,7 @@ async def main():
except TypeError: except TypeError:
# (for now) telethon doesn't easily support attributes for grouped media # (for now) telethon doesn't easily support attributes for grouped media
mimetype = '' mimetype = ''
thumb = None
if mimetype.startswith('video/'): if mimetype.startswith('video/'):
try: try:
data = await _get_video_data(k) data = await _get_video_data(k)
@ -227,7 +228,14 @@ async def main():
logging.exception('Exception when getting video data') logging.exception('Exception when getting video data')
else: else:
attributes.append(DocumentAttributeVideo(duration, w, h, supports_streaming=mimetype == 'video/mp4' or None)) attributes.append(DocumentAttributeVideo(duration, w, h, supports_streaming=mimetype == 'video/mp4' or None))
await client.send_message(chat, j, file=k, link_preview=False, attributes=attributes) dn, _ = os.path.split(k)
try:
nthumb = os.path.join(dn, f'{time.time()}.jpg')
if await _make_thumbnail(nthumb, k):
thumb = nthumb
except Exception:
logging.exception('Exception while making thumbnail')
await client.send_message(chat, j, file=k, link_preview=False, attributes=attributes, thumb=thumb)
async def _get_video_data(filename): async def _get_video_data(filename):
proc = await asyncio.create_subprocess_exec('ffprobe', '-show_format', '-show_streams', '-print_format', 'json', filename, stdout=asyncio.subprocess.PIPE) proc = await asyncio.create_subprocess_exec('ffprobe', '-show_format', '-show_streams', '-print_format', 'json', filename, stdout=asyncio.subprocess.PIPE)
@ -242,6 +250,21 @@ async def main():
data['format']['duration'] = ndata['format']['duration'] data['format']['duration'] = ndata['format']['duration']
return data return data
async def _make_thumbnail(filename, video):
data = await _get_video_data(video)
if not data.get('format'):
return False
if data['format'].get('duration') is None:
return False
for i in (0, 5, 10, 15):
if i and data['format']['duration'] > i:
continue
proc = await asyncio.create_subprocess_exec('ffmpeg', '-an', '-sn', '-ss', str(i), '-i', video, '-frames:v', '1', filename)
await proc.communicate()
if not proc.returncode:
return True
return False
async def _download_file(filename, url): async def _download_file(filename, url):
print(url) print(url)
async with session.get(url) as resp: async with session.get(url) as resp: