386eff647c
Co-authored-by: New-dev0 <New-dev0@users.noreply.github.com> 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>
98 lines
2.7 KiB
Python
98 lines
2.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}get var <variable name>`
|
|
Get value of the given variable name.
|
|
|
|
• `{i}get type <variable name>`
|
|
Get variable type.
|
|
|
|
• `{i}get redis <key>`
|
|
Get redis value of the given key.
|
|
|
|
• `{i}get keys`
|
|
Get all redis keys.
|
|
"""
|
|
|
|
import os
|
|
|
|
from . import *
|
|
|
|
|
|
@ultroid_cmd(pattern="get")
|
|
async def get_var(event):
|
|
if not event.out and not is_fullsudo(event.sender_id):
|
|
return await eod(event, "`This Command Is Sudo Restricted.`")
|
|
if len(event.text) > 4:
|
|
if " " in event.text[4]:
|
|
opt = event.text.split(" ", maxsplit=2)[1]
|
|
else:
|
|
return
|
|
else:
|
|
return
|
|
x = await eor(event, get_string("com_1"))
|
|
if not opt == "keys":
|
|
try:
|
|
varname = event.text.split(" ", maxsplit=2)[2]
|
|
except IndexError:
|
|
return await eod(x, "Such a var doesn't exist!", time=5)
|
|
if opt == "var":
|
|
c = 0
|
|
# try redis
|
|
val = udB.get(varname)
|
|
if val is not None:
|
|
c += 1
|
|
return await x.edit(
|
|
f"**Variable** - `{varname}`\n**Value**: `{val}`\n**Type**: Redis Key."
|
|
)
|
|
# try env vars
|
|
val = os.getenv(varname)
|
|
if val is not None:
|
|
c += 1
|
|
return await x.edit(
|
|
f"**Variable** - `{varname}`\n**Value**: `{val}`\n**Type**: Env Var."
|
|
)
|
|
|
|
if c == 0:
|
|
return await eod(x, "Such a var doesn't exist!", time=5)
|
|
|
|
elif opt == "type":
|
|
c = 0
|
|
# try redis
|
|
val = udB.get(varname)
|
|
if val is not None:
|
|
c += 1
|
|
return await x.edit(f"**Variable** - `{varname}`\n**Type**: Redis Key.")
|
|
# try env vars
|
|
val = os.getenv(varname)
|
|
if val is not None:
|
|
c += 1
|
|
return await x.edit(f"**Variable** - `{varname}`\n**Type**: Env Var.")
|
|
|
|
if c == 0:
|
|
return await eod(x, "Such a var doesn't exist!", time=5)
|
|
|
|
elif opt == "redis":
|
|
val = udB.get(varname)
|
|
if val is not None:
|
|
return await x.edit(f"**Key** - `{varname}`\n**Value**: `{val}`")
|
|
else:
|
|
return await eod(x, "No such key!")
|
|
|
|
elif opt == "keys":
|
|
keys = sorted(udB.keys())
|
|
msg = ""
|
|
for i in keys:
|
|
if i.isdigit() or i.startswith("-") or i.startswith("GBAN_REASON_"):
|
|
pass
|
|
else:
|
|
msg += f"• `{i}`" + "\n"
|
|
await x.edit(f"**List of Redis Keys :**\n{msg}")
|