File size: 3,352 Bytes
618430a |
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 85 86 87 88 89 90 91 92 93 94 95 96 |
# Ultroid - UserBot
# Copyright (C) 2021-2025 TeamUltroid
#
# This file is a part of < https://github.com/TeamUltroid/Ultroid/ >
# PLease read the GNU Affero General Public License in
# <https://www.github.com/TeamUltroid/Ultroid/blob/main/LICENSE/>.
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)
# allow storing both messages and media.
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")
|