Spaces:
Running
Running
File size: 2,350 Bytes
156ce57 |
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 |
# Ultroid - UserBot
# Copyright (C) 2021-2023 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_database")
import re
from . import Redis, eor, get_string, udB, ultroid_cmd
@ultroid_cmd(pattern="setdb( (.*)|$)", fullsudo=True)
async def _(ult):
match = ult.pattern_match.group(1).strip()
if not match:
return await ult.eor("Provide key and value to set!")
try:
delim = " " if re.search("[|]", match) is None else " | "
data = match.split(delim, maxsplit=1)
if data[0] in ["--extend", "-e"]:
data = data[1].split(maxsplit=1)
data[1] = f"{str(udB.get_key(data[0]))} {data[1]}"
udB.set_key(data[0], data[1])
await ult.eor(
f"**DB Key Value Pair Updated\nKey :** `{data[0]}`\n**Value :** `{data[1]}`"
)
except BaseException:
await ult.eor(get_string("com_7"))
@ultroid_cmd(pattern="deldb( (.*)|$)", fullsudo=True)
async def _(ult):
key = ult.pattern_match.group(1).strip()
if not key:
return await ult.eor("Give me a key name to delete!", time=5)
_ = key.split(maxsplit=1)
try:
if _[0] == "-m":
for key in _[1].split():
k = udB.del_key(key)
key = _[1]
else:
k = udB.del_key(key)
if k == 0:
return await ult.eor("`No Such Key.`")
await ult.eor(f"`Successfully deleted key {key}`")
except BaseException:
await ult.eor(get_string("com_7"))
@ultroid_cmd(pattern="rendb( (.*)|$)", fullsudo=True)
async def _(ult):
match = ult.pattern_match.group(1).strip()
if not match:
return await ult.eor("`Provide Keys name to rename..`")
delim = " " if re.search("[|]", match) is None else " | "
data = match.split(delim)
if Redis(data[0]):
try:
udB.rename(data[0], data[1])
await eor(
ult,
f"**DB Key Rename Successful\nOld Key :** `{data[0]}`\n**New Key :** `{data[1]}`",
)
except BaseException:
await ult.eor(get_string("com_7"))
else:
await ult.eor("Key not found")
|