mirror of
https://github.com/MuRuLOSE/limoka.git
synced 2026-06-16 14:34:17 +02:00
56 lines
2.5 KiB
Python
56 lines
2.5 KiB
Python
import os
|
||
from .. import loader, utils
|
||
from asyncio import sleep
|
||
|
||
def register(cb):
|
||
cb(ReplyDownloaderMod())
|
||
|
||
class ReplyDownloaderMod(loader.Module):
|
||
"""Скачать файлом реплай."""
|
||
strings = {'name': 'Reply Downloader'}
|
||
|
||
async def dlrcmd(self, message):
|
||
"""Команда .dlr <реплай на файл> <название (по желанию)> скачивает файл, либо сохраняет текст в файл на который был сделан реплай."""
|
||
name = utils.get_args_raw(message)
|
||
reply = await message.get_reply_message()
|
||
if reply:
|
||
await message.edit('Скачиваем...')
|
||
if reply.text:
|
||
text = reply.text
|
||
fname = f'{name or message.id+reply.id}.txt'
|
||
file = open(fname, 'w')
|
||
file.write(text)
|
||
file.close()
|
||
await message.edit(f'Файл сохранён как: <code>{fname}</code>.\n\nВы можете отправить его в этот чат с помощью команды <code>.ulf {fname}</code>.')
|
||
else:
|
||
ext = reply.file.ext
|
||
fname = f'{name or message.id+reply.id}{ext}'
|
||
await message.client.download_media(reply, fname)
|
||
await message.edit(f'Этот файл сохранён как: <code>{fname}</code>.\n\nВы можете отправить его в этот чат с помощью команды <code>.ulf {fname}</code>.')
|
||
else:
|
||
return await message.edit('Нет реплая.')
|
||
|
||
|
||
async def ulfcmd(self, message):
|
||
"""Команда .ulf <d>* <название файла> отправляет файл в чат.\n* - удалить файл после отправки."""
|
||
name = utils.get_args_raw(message)
|
||
d = False
|
||
if('d ' in name):
|
||
d = True
|
||
if name:
|
||
try:
|
||
name = name.replace('d ', '')
|
||
await message.edit(f'Отправляем <code>{name}</code>...')
|
||
if d == True:
|
||
await message.client.send_file(message.to_id, f'{name}')
|
||
await message.edit(f'Отправляем <code>{name}</code>... Успешно!\nУдаляем <code>{name}</code>...')
|
||
os.remove(name)
|
||
await message.edit(f'Отправляем <code>{name}</code>... Успешно!\nУдаляем <code>{name}</code>... Успешно!')
|
||
await sleep(0.5)
|
||
else:
|
||
await message.client.send_file(message.to_id, name)
|
||
except:
|
||
return await message.edit('Такой файл не существует.')
|
||
await message.delete()
|
||
else:
|
||
return await message.edit('Нет аргументов.') |