channel_download.py

"""
Telegram Channel Media Downloader Plugin for ULTRA.
usage: .geta channel_username [will  get all media from channel, tho there is limit of 3000 there to prevent API limits.]
       .getc number_of_messsages channel_username  
By: @Zero_cool7870
"""
from telethon import events
import asyncio
import os
import subprocess
import sys
from ULTRA.utils import admin_cmd, humanbytes, progress, time_formatter
from ULTRA import CMD_HELP

@borg.on(admin_cmd(pattern=r"getc"))
async def get_media(event):
    if event.fwd_from:
        return
    dir= "./temp/"
    try:
        os.makedirs("./temp/")
    except:
    	pass
    channel_username= event.text
    command = ['ls','temp','|','wc','-l' ]
    limit = channel_username[6:9]
    print(limit)
    channel_username = channel_username[11: ]
    print(channel_username)
    await event.edit("Downloading Media From this Channel.")
    msgs = await borg.get_messages(channel_username, limit=int(limit))
    with open('log.txt','w') as f:
    	f.write(str(msgs))
    for msg in msgs:
       if msg.media is not None:
	        await borg.download_media(
                msg,dir)
    ps = subprocess.Popen(('ls', 'temp'), stdout=subprocess.PIPE)
    output = subprocess.check_output(('wc', '-l'), stdin=ps.stdout)
    ps.wait()
    output = str(output)
    output = output.replace("b'","")
    output = output.replace("\n'","")
    await event.edit("Downloaded "+output+" files.")






@borg.on(admin_cmd(pattern=r"geta"))
async def get_media(event):
    if event.fwd_from:
        return
    dir= "./temp/"
    try:
        os.makedirs("./temp/")
    except:
    	pass
    channel_username= event.text
    command = ['ls','temp','|','wc','-l' ]
    channel_username = channel_username[7:]


    print(channel_username)
    await event.edit("Downloading All Media From this Channel.")
    msgs = await borg.get_messages(channel_username,limit=3000)
    with open('log.txt','w') as f:
    	f.write(str(msgs))
    for msg in msgs:
       if msg.media is not None:
	        await borg.download_media(
                msg,dir)          
    ps = subprocess.Popen(('ls', 'temp'), stdout=subprocess.PIPE)
    output = subprocess.check_output(('wc', '-l'), stdin=ps.stdout)
    ps.wait()
    output = str(output)
    output = output.replace("b'","")
    output = output.replace("\n'","")
    await event.edit("Downloaded "+output+" files.")



CMD_HELP.update(
    {
        "channel_download": f"""**Plugin : **`channel_download`
**Telegram Channel Media Downloader Plugin for ULTRA.**
  • **Syntax : **`.geta channel_username` 
  • **Function : **__will  download all media from channel into your bot server but there is limit of 3000 to prevent API limits.__
  
  • **Syntax : **`.getc number channel_username` 
  • **Function : **__will  download latest given number of media from channel into your bot server .__"""
    }
)