async def sample_fn(c, m):
chat_id = m.from_user.id
if CURRENT_PROCESSES.get(chat_id, 0) == Config.MAX_PROCESSES_PER_USER:
await m.answer(
'You have reached the maximum parallel processes! Try again after one of them finishes.',
show_alert=True)
return
if not CURRENT_PROCESSES.get(chat_id):
CURRENT_PROCESSES[chat_id] = 0
CURRENT_PROCESSES[chat_id] += 1
media_msg = m.message.reply_to_message
if media_msg.empty:
await edit_message_text(
m, text='Why did you delete the file 😠, Now i cannot help you 😒.')
CURRENT_PROCESSES[chat_id] -= 1
return
uid = str(uuid.uuid4())
output_folder = Config.SMPL_OP_FLDR.joinpath(uid)
if not output_folder.exists():
os.makedirs(output_folder)
if media_msg.media:
typ = 1
else:
typ = 2
try:
start_time = time.time()
await edit_message_text(m,
text='Processing your request, Please wait! 😴')
if typ == 2:
file_link = media_msg.text
else:
file_link = await generate_stream_link(media_msg)
if file_link is None:
await edit_message_text(
m,
text=
"😟 Sorry! I cannot help you right now, I'm having hard time processing the file."
)
l = await media_msg.forward(Config.LOG_CHANNEL)
await l.reply_text(
f'@{Config.LINK_GEN_BOT} did not respond with stream url',
True)
CURRENT_PROCESSES[chat_id] -= 1
return
await edit_message_text(
m, text='😀 Generating Sample Video! This might take some time.')
duration = await get_duration(file_link)
if isinstance(duration, str):
await edit_message_text(m, text="😟 Sorry! I cannot open the file.")
l = await media_msg.forward(Config.LOG_CHANNEL)
await l.reply_text(
f'stream link : {file_link}\n\nSample video requested\n\n{duration}',
True)
CURRENT_PROCESSES[chat_id] -= 1
return
reduced_sec = duration - int(duration * 10 / 100)
print(f"Total seconds: {duration}, Reduced seconds: {reduced_sec}")
sample_duration = await db.get_sample_duration(chat_id)
start_at = get_random_start_at(reduced_sec, sample_duration)
sample_file = output_folder.joinpath(f'sample_video.mkv')
ffmpeg_cmd = f"ffmpeg -hide_banner -ss {start_at} -i {shlex.quote(file_link)} -t {sample_duration} -map 0 -c copy {sample_file}"
output = await run_subprocess(ffmpeg_cmd)
#print(output[1].decode())
if not sample_file.exists():
await edit_message_text(
m,
text=
'😟 Sorry! Sample video generation failed possibly due to some infrastructure failure 😥.'
)
l = await media_msg.forward(Config.LOG_CHANNEL)
await l.reply_text(
f'stream link : {file_link}\n\n duration {sample_duration} sample video generation failed\n\n{output[1].decode()}',
True)
CURRENT_PROCESSES[chat_id] -= 1
return
thumb = await generate_thumbnail_file(sample_file, uid)
await edit_message_text(
m,
text=
f'🤓 Sample video was generated successfully!, Now starting to upload!'
)
await media_msg.reply_chat_action("upload_video")
await media_msg.reply_video(
video=sample_file,
quote=True,
caption=
f"Sample video. {sample_duration}s from {datetime.timedelta(seconds=start_at)}",
duration=sample_duration,
thumb=thumb,
supports_streaming=True)
await edit_message_text(
m,
text=
f'Successfully completed process in {datetime.timedelta(seconds=int(time.time()-start_time))}\n\nIf You find me helpful, please rate me [here](tg://resolve?domain=botsarchive&post=1206)'
)
CURRENT_PROCESSES[chat_id] -= 1
except:
traceback.print_exc()
await edit_message_text(
m,
text=
'😟 Sorry! Sample video generation failed possibly due to some infrastructure failure 😥.'
)
l = await media_msg.forward(Config.LOG_CHANNEL)
await l.reply_text(
f'sample video requested and some error occoured\n\n{traceback.format_exc()}',
True)
CURRENT_PROCESSES[chat_id] -= 1
async def trim_fn(c, m):
chat_id = m.chat.id
if CURRENT_PROCESSES.get(chat_id, 0) == Config.MAX_PROCESSES_PER_USER:
await m.reply_text(
'You have reached the maximum parallel processes! Try again after one of them finishes.',
True)
return
if not CURRENT_PROCESSES.get(chat_id):
CURRENT_PROCESSES[chat_id] = 0
CURRENT_PROCESSES[chat_id] += 1
message = await c.get_messages(chat_id, m.reply_to_message.message_id)
await m.reply_to_message.delete()
media_msg = message.reply_to_message
if media_msg.empty:
await m.reply_text(
'Why did you delete the file 😠, Now i cannot help you 😒.', True)
CURRENT_PROCESSES[chat_id] -= 1
return
try:
start, end = [int(i) for i in m.text.split(':')]
except:
await m.reply_text('Please follow the specified format', True)
CURRENT_PROCESSES[chat_id] -= 1
return
if (start >= end) or (start < 0):
await m.reply_text('Invalid range!', True)
CURRENT_PROCESSES[chat_id] -= 1
return
request_duration = end - start
if request_duration > Config.MAX_TRIM_DURATION:
await m.reply_text(
f'Please provide any range that\'s upto {Config.MAX_TRIM_DURATION}s. Your requested range **{start}:{end}** is `{request_duration}s` long!',
True)
CURRENT_PROCESSES[chat_id] -= 1
return
uid = str(uuid.uuid4())
output_folder = Config.SMPL_OP_FLDR.joinpath(uid)
if not output_folder.exists():
os.makedirs(output_folder)
if media_msg.media:
typ = 1
else:
typ = 2
snt = await m.reply_text('Processing your request, Please wait! 😴', True)
try:
start_time = time.time()
if typ == 2:
file_link = media_msg.text
else:
file_link = await generate_stream_link(media_msg)
if file_link is None:
await snt.edit_text(
"😟 Sorry! I cannot help you right now, I'm having hard time processing the file."
)
l = await media_msg.forward(Config.LOG_CHANNEL)
await l.reply_text(
f'@{Config.LINK_GEN_BOT} did not respond with stream url',
True)
CURRENT_PROCESSES[chat_id] -= 1
return
await snt.edit_text('😀 Trimming Your Video! This might take some time.'
)
duration = await get_duration(file_link)
if isinstance(duration, str):
await snt.edit_text("😟 Sorry! I cannot open the file.")
l = await media_msg.forward(Config.LOG_CHANNEL)
await l.reply_text(
f'stream link : {file_link}\n\ntrim video requested\n\n{start}:{end}',
True)
CURRENT_PROCESSES[chat_id] -= 1
return
if (start >= duration) or (end >= duration):
await snt.edit_text(
"😟 Sorry! The requested range is out of the video's duration!."
)
CURRENT_PROCESSES[chat_id] -= 1
return
sample_file = output_folder.joinpath(f'trim_video.mkv')
ffmpeg_cmd = f"ffmpeg -hide_banner -ss {start} -i {shlex.quote(file_link)} -t {request_duration} -map 0 -c copy {sample_file}"
output = await run_subprocess(ffmpeg_cmd)
#print(output[1].decode())
if not sample_file.exists():
await snt.edit_text(
'😟 Sorry! video trimming failed possibly due to some infrastructure failure 😥.'
)
l = await media_msg.forward(Config.LOG_CHANNEL)
await l.reply_text(
f'stream link : {file_link}\n\nVideo trimm failed. **{start}:{end}**\n\n{output[1].decode()}',
True)
CURRENT_PROCESSES[chat_id] -= 1
return
thumb = await generate_thumbnail_file(sample_file, uid)
await snt.edit_text(
'🤓 Video trimmed successfully!, Now starting to upload!')
await m.reply_chat_action("upload_video")
await m.reply_video(
video=sample_file,
quote=True,
caption=
f"Trimmed video from {datetime.timedelta(seconds=start)} to {datetime.timedelta(seconds=end)}",
duration=request_duration,
thumb=thumb,
supports_streaming=True)
await snt.edit_text(
f'Successfully completed process in {datetime.timedelta(seconds=int(time.time()-start_time))}\n\nIf You find me helpful, please rate me [here](tg://resolve?domain=botsarchive&post=1206)'
)
CURRENT_PROCESSES[chat_id] -= 1
except:
traceback.print_exc()
await snt.edit_text(
'😟 Sorry! Sample video generation failed possibly due to some infrastructure failure 😥.'
)
l = await media_msg.forward(Config.LOG_CHANNEL)
await l.reply_text(
f'sample video requested and some error occoured\n\n{traceback.format_exc()}',
True)
CURRENT_PROCESSES[chat_id] -= 1
async def screenshot_fn(c, m):
chat_id = m.from_user.id
if CURRENT_PROCESSES.get(chat_id, 0) == Config.MAX_PROCESSES_PER_USER:
await m.answer(
'You have reached the maximum parallel processes! Try again after one of them finishes.',
show_alert=True)
return
if not CURRENT_PROCESSES.get(chat_id):
CURRENT_PROCESSES[chat_id] = 0
CURRENT_PROCESSES[chat_id] += 1
_, num_screenshots = m.data.split('+')
num_screenshots = int(num_screenshots)
media_msg = m.message.reply_to_message
#print(media_msg)
if media_msg.empty:
await edit_message_text(
m, text='Why did you delete the file 😠, Now i cannot help you 😒.')
CURRENT_PROCESSES[chat_id] -= 1
return
uid = str(uuid.uuid4())
output_folder = Config.SCRST_OP_FLDR.joinpath(uid)
if not output_folder.exists():
os.makedirs(output_folder)
if media_msg.media:
typ = 1
else:
typ = 2
try:
start_time = time.time()
await edit_message_text(m,
text='Processing your request, Please wait! 😴')
if typ == 2:
file_link = media_msg.text
else:
file_link = await generate_stream_link(media_msg)
if file_link is None:
await edit_message_text(
m,
text=
"😟 Sorry! I cannot help you right now, I'm having hard time processing the file."
)
l = await media_msg.forward(Config.LOG_CHANNEL)
await l.reply_text(
f'@{Config.LINK_GEN_BOT} did not respond with stream url',
True)
CURRENT_PROCESSES[chat_id] -= 1
return
await edit_message_text(m, text='😀 Generating screenshots!')
duration = await get_duration(file_link)
if isinstance(duration, str):
await edit_message_text(m, text="😟 Sorry! I cannot open the file.")
l = await media_msg.forward(Config.LOG_CHANNEL)
await l.reply_text(
f'stream link : {file_link}\n\nRequested screenshots: {num_screenshots} \n\n{duration}',
True)
CURRENT_PROCESSES[chat_id] -= 1
return
reduced_sec = duration - int(duration * 2 / 100)
print(f"Total seconds: {duration}, Reduced seconds: {reduced_sec}")
screenshots = []
watermark = await db.get_watermark_text(chat_id)
watermark_color_code = await db.get_watermark_color(chat_id)
watermark_color = Config.COLORS[watermark_color_code]
as_file = await db.is_as_file(chat_id)
screenshot_mode = await db.get_screenshot_mode(chat_id)
ffmpeg_errors = ''
if screenshot_mode == 0:
screenshot_secs = [
int(reduced_sec / 10) * i
for i in range(1, 1 + num_screenshots)
]
else:
screenshot_secs = [
get_random_start_at(reduced_sec)
for i in range(1, 1 + num_screenshots)
]
width, height = await get_dimentions(file_link)
fontsize = int((math.sqrt(width**2 + height**2) / 1388.0) *
Config.DEFAULT_FONT_SIZE)
for i, sec in enumerate(screenshot_secs):
thumbnail_template = output_folder.joinpath(f'{i+1}.png')
print(sec)
ffmpeg_cmd = f"ffmpeg -hide_banner -ss {sec} -i {shlex.quote(file_link)} -vf \"drawtext=fontcolor={watermark_color}:fontsize={fontsize}:y=H-th-10:text='{shlex.quote(watermark)}'\" -vframes 1 '{thumbnail_template}'"
output = await run_subprocess(ffmpeg_cmd)
await edit_message_text(
m, text=f'😀 `{i+1}` of `{num_screenshots}` generated!')
if thumbnail_template.exists():
if as_file:
screenshots.append({
'document':
str(thumbnail_template),
'caption':
f"ScreenShot at {datetime.timedelta(seconds=sec)}"
})
else:
screenshots.append(
InputMediaPhoto(
str(thumbnail_template),
caption=
f"ScreenShot at {datetime.timedelta(seconds=sec)}")
)
continue
ffmpeg_errors += output[1].decode() + '\n\n'
#print(screenshots)
if not screenshots:
await edit_message_text(
m,
text=
'😟 Sorry! Screenshot generation failed possibly due to some infrastructure failure 😥.'
)
l = await media_msg.forward(Config.LOG_CHANNEL)
if ffmpeg_errors:
error_file = f"{uid}-errors.txt"
with open(error_file, 'w') as f:
f.write(ffmpeg_errors)
await l.reply_document(
error_file,
caption=
f"stream link : {file_link}\n\n{num_screenshots} screenshots where requested and Screen shots where not generated."
)
os.remove(error_file)
else:
await l.reply_text(
f'stream link : {file_link}\n\n{num_screenshots} screenshots where requested and Screen shots where not generated.',
True)
CURRENT_PROCESSES[chat_id] -= 1
return
await edit_message_text(
m,
text=
f'🤓 You requested {num_screenshots} screenshots and {len(screenshots)} screenshots generated, Now starting to upload!'
)
await media_msg.reply_chat_action("upload_photo")
if as_file:
aws = [
media_msg.reply_document(quote=True, **photo)
for photo in screenshots
]
await asyncio.gather(*aws)
else:
await media_msg.reply_media_group(screenshots, True)
await edit_message_text(
m,
text=
f'Successfully completed process in {datetime.timedelta(seconds=int(time.time()-start_time))}\n\nIf You find me helpful, please rate me [here](tg://resolve?domain=botsarchive&post=1206)'
)
CURRENT_PROCESSES[chat_id] -= 1
except:
traceback.print_exc()
await edit_message_text(
m,
text=
'😟 Sorry! Screenshot generation failed possibly due to some infrastructure failure 😥.'
)
l = await media_msg.forward(Config.LOG_CHANNEL)
await l.reply_text(
f'{num_screenshots} screenshots where requested and some error occoured\n\n{traceback.format_exc()}',
True)
CURRENT_PROCESSES[chat_id] -= 1