mirror of
https://gitlab.com/octospacc/TelegramIndex-Fork.git
synced 2025-02-22 22:37:47 +01:00
72 lines
2.4 KiB
Python
72 lines
2.4 KiB
Python
|
import logging
|
||
|
|
from PIL import Image
|
||
|
|
import random
|
||
|
|
import io
|
||
|
|
|
||
|
|
from aiohttp import web
|
||
|
|
from telethon.tl import types
|
||
|
|
|
||
|
|
|
||
|
|
log = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
class ThumbnailView:
|
||
|
|
|
||
|
|
async def thumbnail_get(self, req):
|
||
|
|
file_id = int(req.match_info["id"])
|
||
|
|
alias_id = req.match_info['chat']
|
||
|
|
chat = [i for i in self.chat_ids if i['alias_id'] == alias_id][0]
|
||
|
|
chat_id = chat['chat_id']
|
||
|
|
try:
|
||
|
|
message = await self.client.get_messages(entity=chat_id, ids=file_id)
|
||
|
|
except:
|
||
|
|
log.debug(f"Error in getting message {file_id} in {chat_id}", exc_info=True)
|
||
|
|
message = None
|
||
|
|
|
||
|
|
if not message or not message.file:
|
||
|
|
log.debug(f"no result for {file_id} in {chat_id}")
|
||
|
|
return web.Response(status=410, text="410: Gone. Access to the target resource is no longer available!")
|
||
|
|
|
||
|
|
if message.document:
|
||
|
|
media = message.document
|
||
|
|
thumbnails = media.thumbs
|
||
|
|
location = types.InputDocumentFileLocation
|
||
|
|
else:
|
||
|
|
media = message.photo
|
||
|
|
thumbnails = media.sizes
|
||
|
|
location = types.InputPhotoFileLocation
|
||
|
|
|
||
|
|
if not thumbnails:
|
||
|
|
color = tuple([random.randint(0, 255) for i in range(3)])
|
||
|
|
im = Image.new("RGB", (100, 100), color)
|
||
|
|
temp = io.BytesIO()
|
||
|
|
im.save(temp, "PNG")
|
||
|
|
body = temp.getvalue()
|
||
|
|
else:
|
||
|
|
thumb_pos = int(len(thumbnails)/2)
|
||
|
|
thumbnail = self.client._get_thumb(thumbnails, thumb_pos)
|
||
|
|
if not thumbnail or isinstance(thumbnail, types.PhotoSizeEmpty):
|
||
|
|
return web.Response(status=410, text="410: Gone. Access to the target resource is no longer available!")
|
||
|
|
|
||
|
|
if isinstance(thumbnail, (types.PhotoCachedSize, types.PhotoStrippedSize)):
|
||
|
|
body = self.client._download_cached_photo_size(thumbnail, bytes)
|
||
|
|
else:
|
||
|
|
actual_file = location(
|
||
|
|
id=media.id,
|
||
|
|
access_hash=media.access_hash,
|
||
|
|
file_reference=media.file_reference,
|
||
|
|
thumb_size=thumbnail.type
|
||
|
|
)
|
||
|
|
|
||
|
|
body = self.client.iter_download(actual_file)
|
||
|
|
|
||
|
|
r = web.Response(
|
||
|
|
status=200,
|
||
|
|
body=body,
|
||
|
|
headers={
|
||
|
|
"Content-Type": "image/jpeg",
|
||
|
|
"Content-Disposition": 'inline; filename="thumbnail.jpg"'
|
||
|
|
}
|
||
|
|
)
|
||
|
|
return r
|