Skip to content

🛠️ app/forward.py -> Added video thumbnail handling #14

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
is0late wants to merge 2 commits into
base: master
Choose a base branch
from
Open

🛠️ app/forward.py -> Added video thumbnail handling #14

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
86367f3
🛠️ app/forward.py -> Added video thumbnail handling
is0late Aug 29, 2024
47ceed0
Update forward.py
is0late Aug 29, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
31 changes: 30 additions & 1 deletion app/forward.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -293,13 +293,25 @@ async def copy_message(message: Message, target: dict, edited=False,
messages = await user.get_media_group(source, message.id)
media_input = []
downloaded_media = []
downloaded_video_thumbs = []

for msg_media in messages:
text = await replace_words(forwarder, msg_media.caption, True)
entities = msg_media.caption_entities
video = None
thumb_path = None
# If the chat has protected content, download the media to send it
if msg_media.chat.has_protected_content:
path = await msg_media.download()

# Add video thumbs info
if msg_media.video:
video = msg_media.video
if video.thumbs:
thumb = video.thumbs[0] # choice the first thumb
thumb_path = await message.download(thumb.file_id)
logger.debug(f"Video thumbs downloaded to {thumb_path}")
downloaded_video_thumbs.append(thumb_path)
# If the chat has not protected content, get the file_id to send it
else:
path = await get_media_type(msg_media)
Expand All @@ -308,7 +320,8 @@ async def copy_message(message: Message, target: dict, edited=False,
if msg_media.media is MessageMediaType.PHOTO:
media_input.append(InputMediaPhoto(path, text))
elif msg_media.media is MessageMediaType.VIDEO:
media_input.append(InputMediaVideo(path, caption=text))
duration = video.duration if video else None
media_input.append(InputMediaVideo(path, caption=text, duration=duration, thumb=thumb_path))
elif msg_media.media is MessageMediaType.AUDIO:
media_input.append(InputMediaAudio(path, caption=text))
elif msg_media.media is MessageMediaType.DOCUMENT:
Expand Down Expand Up @@ -349,6 +362,10 @@ async def copy_message(message: Message, target: dict, edited=False,
if os.path.isfile(path):
logger.debug(f"Removing file {path}")
os.remove(path)
for path in downloaded_video_thumbs:
if os.path.isfile(path):
logger.debug(f'Removing thumb file {path}')
os.remove(path)

elif message.media is not None:
downloadable_media = [MessageMediaType.PHOTO, MessageMediaType.VIDEO,
Expand All @@ -372,6 +389,15 @@ async def copy_message(message: Message, target: dict, edited=False,
logger.debug(f"{from_user} has protected content, "
"downloading media")
path = await message.download()

# Add video thumbs info
if message.video:
video = message.video
thumb_path = None
if video.thumbs:
thumb = video.thumbs[0] # choice the first thumb
thumb_path = await message.download(thumb.file_id)
logger.debug(f"Video thumbs downloaded to {thumb_path}")
else:
logger.debug(f"{from_user} has no protected content, "
"using file_id")
Expand Down Expand Up @@ -557,6 +583,9 @@ async def copy_message(message: Message, target: dict, edited=False,
if os.path.isfile(path):
logger.debug(f"Removing file {path}")
os.remove(path)
if os.path.isfile(thumb_path):
logger.debug(f'Removing thumb file {thumb_path}')
os.remove(thumb_path)

# If the message is just a text message
else:
Expand Down