Files
Ultroid-fork/plugins/redis.py
xditya c4966f772a Initial Commit
Co-Authored-By: Aditya <me@xditya.me>
Co-Authored-By: Amit Sharma <48654350+buddhhu@users.noreply.github.com>
Co-Authored-By: hellboi_atul <68107352+hellboi-atul@users.noreply.github.com>
Co-Authored-By: Sρι∂у <68327188+sppidy@users.noreply.github.com>
Co-Authored-By: Anonymous <69723581+New-dev0@users.noreply.github.com>
Co-Authored-By: Danish <72792730+1Danish-00@users.noreply.github.com>
Co-Authored-By: Arnab Paryali <arnabxd@pm.me>
Co-Authored-By: Programming Error <75001577+programmingerror@users.noreply.github.com>
2021-02-22 06:19:43 +05:30

116 lines
2.9 KiB
Python

# Ultroid - UserBot
# Copyright (C) 2020 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}setredis key | value`
Redis Set Value.
e.g :
`{i}setredis hi there`
`{i}setredis hi there | ultroid here`
• `{i}getredis key`
Redis Get Value
• `{i}delredis key`
Delete Key from Redis DB
• `{i}renredis old keyname | new keyname`
Update Key Name
• `{i}getkeys`
Get the list of keys stored in Redis
"""
import re
from . import *
@ultroid_cmd(
pattern="setredis ?(.*)",
)
async def _(ult):
ok = await eor(ult, "`...`")
try:
delim = " " if re.search("[|]", ult.pattern_match.group(1)) is None else " | "
data = ult.pattern_match.group(1).split(delim)
udB.set(data[0], data[1])
redisdata = Redis(data[0])
await ok.edit(
"Redis Key Value Pair Updated\nKey : `{}`\nValue : `{}`".format(
data[0], redisdata
)
)
except BaseException:
await ok.edit("`Something Went Wrong`")
@ultroid_cmd(
pattern="getredis ?(.*)",
)
async def _(ult):
ok = await eor(ult, "`Fetching data from Redis`")
val = ult.pattern_match.group(1)
if val == "":
return await ult.edit(f"Please use `{hndlr}getkeys <keyname>`")
try:
value = Redis(val)
await ok.edit("Key: `{}`\nValue: `{}`".format(val, value))
except BaseException:
await ok.edit("`Something Went Wrong`")
@ultroid_cmd(
pattern="delredis ?(.*)",
)
async def _(ult):
ok = await eor(ult, "`Deleting data from Redis ...`")
try:
key = ult.pattern_match.group(1)
udB.delete(key)
await ok.edit(f"`Successfully deleted key {key}`")
except BaseException:
await ok.edit("`Something Went Wrong`")
@ultroid_cmd(
pattern="renredis ?(.*)",
)
async def _(ult):
ok = await eor(ult, "`...`")
delim = " " if re.search("[|]", ult.pattern_match.group(1)) is None else " | "
data = ult.pattern_match.group(1).split(delim)
if Redis(data[0]):
try:
udB.rename(data[0], data[1])
await ok.edit(
"Redis Key Rename Successful\nOld Key : `{}`\nNew Key : `{}`".format(
data[0], data[1]
)
)
except BaseException:
await ok.edit("Something went wrong ...")
else:
await ok.edit("Key not found")
@ultroid_cmd(
pattern="getkeys$",
)
async def _(ult):
ok = await eor(ult, "`Fetching Keys ...`")
keys = udB.keys()
msg = ""
for x in keys:
msg += "• `{}`".format(x) + "\n"
await ok.edit("**List of Redis Keys :**\n{}".format(msg))
HELP.update({f"{__name__.split('.')[1]}": f"{__doc__.format(i=Var.HNDLR)}"})