forked from yashoswalyo/MERGE-BOT
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathstreams_extractor.py
84 lines (79 loc) · 3.3 KB
/
streams_extractor.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
import time
from pyrogram import Client
from pyrogram.types import Message, CallbackQuery
from pyrogram.errors import MessageNotModified
from pyrogram.errors.rpc_error import UnknownError
import asyncio
from __init__ import LOGGER, gDict, queueDB
import os
from bot import delete_all
from helpers.display_progress import Progress
from helpers.ffmpeg_helper import extractAudios, extractSubtitles
from helpers.uploader import uploadFiles
async def streamsExtractor(c: Client, cb:CallbackQuery ,media_mid, exAudios=False, exSubs=False):
if not os.path.exists(f"downloads/{str(cb.from_user.id)}/"):
os.makedirs(f"downloads/{str(cb.from_user.id)}/")
_hold = await cb.message.edit(text="Please wait")
omess:Message = await c.get_messages(chat_id=cb.from_user.id, message_ids=media_mid)
try:
if (omess.video or omess.document):
media = omess.video or omess.document
LOGGER.info(f'Starting Download: {media.file_name}')
except Exception as e:
LOGGER.error(f"Download failed: Unable to find media {e}")
return
c.stream_media(media,)
try:
c_time = time.time()
prog = Progress(cb.from_user.id, c, cb.message)
progress=f"🚀 Downloading: `{media.file_name}`"
file_dl_path = await c.download_media(
message=media,
file_name=f"downloads/{str(cb.from_user.id)}/{str(omess.id)}/vid.mkv", # fix for filename with single quote(') in name
progress=prog.progress_for_pyrogram,
progress_args=(progress, c_time),
)
if gDict[cb.message.chat.id] and cb.message.id in gDict[cb.message.chat.id]:
return
await cb.message.edit(f"Downloaded Sucessfully ... `{media.file_name}`")
LOGGER.info(f"Downloaded Sucessfully ... {media.file_name}")
await asyncio.sleep(5)
except UnknownError as e:
LOGGER.info(e)
pass
except Exception as downloadErr:
LOGGER.info(f"Failed to download Error: {downloadErr}")
await cb.message.edit("Download Error")
await asyncio.sleep(4)
await _hold.edit_text("Fetching data")
await asyncio.sleep(3)
if exAudios:
await _hold.edit_text("Extracting Audios")
extract_dir = await extractAudios(file_dl_path,cb.from_user.id)
if exSubs:
await _hold.edit_text("Extracting Subtitles")
extract_dir = await extractSubtitles(file_dl_path, cb.from_user.id)
if extract_dir is None:
await cb.message.edit("❌ Failed to Extract Streams !")
await delete_all(root=f"downloads/{str(cb.from_user.id)}")
queueDB.update({cb.from_user.id: {"videos": [], "subtitles": [], "audios": []}})
return
for dirpath, dirnames, filenames in os.walk(extract_dir):
no_of_files = len(filenames)
cf=1
for f in filenames:
await asyncio.sleep(5)
up_path = os.path.join(dirpath,f)
await uploadFiles(
c=c,
cb=cb,
up_path=up_path,
n=cf,
all=no_of_files,
)
cf+=1
LOGGER.info(f"Uploaded: {up_path}")
await cb.message.delete()
await delete_all(root=f"downloads/{str(cb.from_user.id)}")
queueDB.update({cb.from_user.id: {"videos": [], "subtitles": [], "audios": []}})
return