|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
from . import get_help |
|
|
|
|
|
__doc__ = get_help("help_fileshare") |
|
|
|
|
|
import os |
|
|
|
|
|
from pyUltroid.dB.filestore_db import del_stored, get_stored_msg, list_all_stored_msgs |
|
|
from pyUltroid.fns.tools import get_file_link |
|
|
|
|
|
from . import HNDLR, asst, get_string, in_pattern, udB, ultroid_bot, ultroid_cmd |
|
|
|
|
|
|
|
|
@ultroid_cmd(pattern="store$") |
|
|
async def filestoreplg(event): |
|
|
msg = await event.get_reply_message() |
|
|
if not msg: |
|
|
return await event.eor(get_string("fsh_3"), time=10) |
|
|
|
|
|
filehash = await get_file_link(msg) |
|
|
link_to_file = f"https://t.me/{asst.me.username}?start={filehash}" |
|
|
await event.eor( |
|
|
get_string("fsh_2").format(link_to_file), |
|
|
link_preview=False, |
|
|
) |
|
|
|
|
|
|
|
|
@ultroid_cmd("delstored ?(.*)") |
|
|
async def _(event): |
|
|
match = event.pattern_match.group(1) |
|
|
if not match: |
|
|
return await event.eor("`Give stored film's link to delete.`", time=5) |
|
|
match = match.split("?start=") |
|
|
botusername = match[0].split("/")[-1] |
|
|
if botusername != asst.me.username: |
|
|
return await event.eor( |
|
|
"`Message/Media of provided link was not stored by this bot.`", time=5 |
|
|
) |
|
|
msg_id = get_stored_msg(match[1]) |
|
|
if not msg_id: |
|
|
return await event.eor( |
|
|
"`Message/Media of provided link was already deleted.`", time=5 |
|
|
) |
|
|
del_stored(match[1]) |
|
|
await ultroid_bot.delete_messages(udB.get_key("LOG_CHANNEL"), int(msg_id)) |
|
|
await event.eor("__Deleted__") |
|
|
|
|
|
|
|
|
@ultroid_cmd("liststored$") |
|
|
async def liststored(event): |
|
|
files = list_all_stored_msgs() |
|
|
if not files: |
|
|
return await event.eor(get_string("fsh_4"), time=5) |
|
|
msg = "**Stored files:**\n" |
|
|
for c, i in enumerate(files, start=1): |
|
|
msg += f"`{c}`. https://t.me/{asst.me.username}?start={i}\n" |
|
|
if len(msg) > 4095: |
|
|
with open("liststored.txt", "w") as f: |
|
|
f.write(msg.replace("**", "").replace("`", "")) |
|
|
await event.reply(get_string("fsh_1"), file="liststored.txt") |
|
|
os.remove("liststored.txt") |
|
|
return |
|
|
await event.eor(msg, link_preview=False) |
|
|
|
|
|
|
|
|
@in_pattern("filestore", owner=True) |
|
|
async def file_short(event): |
|
|
all_ = list_all_stored_msgs() |
|
|
res = [] |
|
|
if all_: |
|
|
LOG_CHA = udB.get_key("LOG_CHANNEL") |
|
|
for msg in all_[:50]: |
|
|
m_id = get_stored_msg(msg) |
|
|
if not m_id: |
|
|
continue |
|
|
message = await asst.get_messages(LOG_CHA, ids=m_id) |
|
|
if not message: |
|
|
continue |
|
|
if message.media: |
|
|
res.append(await event.builder.document(title=msg, file=message.media)) |
|
|
elif message.text: |
|
|
res.append( |
|
|
await event.builder.article(title=message.text, text=message.text) |
|
|
) |
|
|
if not res: |
|
|
title = "You have no stored file :(" |
|
|
text = f"{title}\n\nRead `{HNDLR}help fileshare` to know how to store." |
|
|
return await event.answer([await event.builder.article(title=title, text=text)]) |
|
|
await event.answer(res, switch_pm="• File Store •", switch_pm_param="start") |
|
|
|
