Files
Ultroid-fork/plugins/sudo.py
Devesh Pal 6c7af44f09 Ultroid v0.1 Updates 13.09.2021
Co-authored-by: Danish <danish@ultroid.tech>
Co-authored-by: Aditya <me@xditya.me>
Co-authored-by: Amit Sharma <48654350+buddhhu@users.noreply.github.com>
Co-authored-by: sppidy <sppidy@users.noreply.github.com>
Co-authored-by: Arnab Paryali <Arnabxd@users.noreply.github.com>
Co-authored-by: divkix <divkix@users.noreply.github.com>
Co-authored-by: hellboi_atul <hellboi-atul@users.noreply.github.com>
Co-authored-by: Programming Error <error@notavailable.live>
2021-09-13 01:03:39 +05:30

118 lines
3.7 KiB
Python

# Ultroid - UserBot
# Copyright (C) 2021 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/>.
"""
✘ Commands Available -
• `{i}addsudo`
Add Sudo Users by replying to user or using <space> separated userid(s)
• `{i}delsudo`
Remove Sudo Users by replying to user or using <space> separated userid(s)
• `{i}listsudo`
List all sudo users.
"""
from . import *
@ultroid_cmd(pattern="addsudo ?(.*)", fullsudo=True)
async def _(ult):
inputs = ult.pattern_match.group(1)
if ult.reply_to_msg_id:
replied_to = await ult.get_reply_message()
sender = await replied_to.get_sender()
id = replied_to.sender_id
name = get_display_name(sender)
elif inputs:
id = await get_user_id(inputs)
try:
name = (await ult.client.get_entity(int(id))).first_name
except BaseException:
name = ""
elif ult.is_private:
id = ult.chat_id
name = get_display_name(ult.chat)
else:
return await eor(ult, "`Reply to a msg or add it's id/username.`", time=5)
if id == ultroid_bot.me.id:
mmm = "You cant add yourself as Sudo User..."
elif is_sudo(id):
if name != "":
mmm = f"[{name}](tg://user?id={id}) `is already a SUDO User ...`"
else:
mmm = f"`{id} is already a SUDO User...`"
elif add_sudo(id):
udB.set("SUDO", "True")
if name != "":
mmm = f"**Added [{name}](tg://user?id={id}) as SUDO User**"
else:
mmm = f"**Added **`{id}`** as SUDO User**"
else:
mmm = "`SEEMS LIKE THIS FUNCTION CHOOSE TO BREAK ITSELF`"
await eor(ult, mmm, time=5)
@ultroid_cmd(pattern="delsudo ?(.*)", fullsudo=True)
async def _(ult):
inputs = ult.pattern_match.group(1)
if ult.reply_to_msg_id:
replied_to = await ult.get_reply_message()
id = replied_to.sender_id
name = get_display_name(replied_to.sender)
elif inputs:
id = await get_user_id(inputs)
try:
name = (await ult.client.get_entity(int(id))).first_name
except BaseException:
name = ""
elif ult.is_private:
id = ult.chat_id
name = get_display_name(ult.chat)
else:
return await eor(ult, "`Reply to a msg or add it's id/username.`", time=5)
if not is_sudo(id):
if name != "":
mmm = f"[{name}](tg://user?id={id}) `wasn't a SUDO User ...`"
else:
mmm = f"`{id} wasn't a SUDO User...`"
elif del_sudo(id):
if name != "":
mmm = f"**Removed [{name}](tg://user?id={id}) from SUDO User(s)**"
else:
mmm = f"**Removed **`{id}`** from SUDO User(s)**"
else:
mmm = "`SEEMS LIKE THIS FUNCTION CHOOSE TO BREAK ITSELF`"
await eor(ult, mmm, time=5)
@ultroid_cmd(
pattern="listsudo$",
)
async def _(ult):
sudos = Redis("SUDOS")
if sudos == "" or sudos is None:
return await eor(ult, "`No SUDO User was assigned ...`", time=5)
sumos = sudos.split(" ")
msg = ""
for i in sumos:
try:
name = (await ult.client.get_entity(int(i))).first_name
except BaseException:
name = ""
if name != "":
msg += f"• [{name}](tg://user?id={i}) ( `{i}` )\n"
else:
msg += f"• `{i}` -> Invalid User\n"
m = udB.get("SUDO") or "False"
if m == "False":
m = "[False](https://telegra.ph/Ultroid-04-06)"
return await eor(
ult, f"**SUDO MODE : {m}\n\nList of SUDO Users :**\n{msg}", link_preview=False
)