osmarks/autobotrobot
1
0
Fork 0
mirror of https://github.com/osmarks/autobotrobot synced 2025-01-08 14:50:25 +00:00
Code Issues Releases Wiki Activity

userdata changes

Browse Source
This commit is contained in:
osmarks 2021-02-15 20:12:34 +00:00
parent 5f17885953
commit d449d5356b
1 changed files with 4 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
src/userdata.py
View File

@ -14,17 +14,10 @@ def setup(bot):
@userdata.command(help="Get a userdata key. Checks guild first, then global.") @userdata.command(help="Get a userdata key. Checks guild first, then global.")
async def get(ctx, *, key): async def get(ctx, *, key):
no_header = False
if key.startswith("noheader "):
key = key[9:]
no_header = True
row = await get_userdata(bot.database, ctx.author.id, ctx.guild.id, key) row = await get_userdata(bot.database, ctx.author.id, ctx.guild.id, key)
if not row: if not row:
raise ValueError("No such key") raise ValueError("No such key")
if no_header: await ctx.send(row["value"])
await ctx.send(row["value"])
else:
await ctx.send(f"**{key}**: {row['value']}")
@userdata.command(name="list", brief="List userdata keys in a given scope matching a query.") @userdata.command(name="list", brief="List userdata keys in a given scope matching a query.")
async def list_cmd(ctx, query="%", scope="guild", show_values: bool = False): async def list_cmd(ctx, query="%", scope="guild", show_values: bool = False):
@ -46,8 +39,8 @@ def setup(bot):
if len(key) > 128: raise ValueError("Key too long") if len(key) > 128: raise ValueError("Key too long")
def preprocess_value(value): def preprocess_value(value):
value = value.replace("\n", "").strip() value = value.strip()
if len(value) > 256: raise ValueError("Value too long") if len(value) > 1024: raise ValueError("Value too long")
return value return value
@userdata.command(name="set", help="Set a userdata key in the guild scope.") @userdata.command(name="set", help="Set a userdata key in the guild scope.")
@ -76,7 +69,7 @@ def setup(bot):
value = int(row["value"]) value = int(row["value"])
guild = row["guild_id"] guild = row["guild_id"]
new_value = value + by new_value = value + by
await set_userdata(bot.database, ctx.author.id, guild, key, str(new_value)) await set_userdata(bot.database, ctx.author.id, guild, key, preprocess_value(str(new_value)))
await ctx.send(f"**{key}** set to {new_value}") await ctx.send(f"**{key}** set to {new_value}")
@userdata.command() @userdata.command()