nex/nonsensebot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix args

Browse source
This commit is contained in:
Nexus 2024-09-20 01:48:52 +01:00
parent a2274e3961
commit d5a20fdf33
1 changed files with 4 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
app/modules/msc_getter.py
View file

@ -194,13 +194,12 @@ class MSCGetter(niobot.Module):
@niobot.command("msc.cache")
@niobot.is_owner()
async def msc_cache_manager(self, ctx: niobot.Context, *args):
async def msc_cache_manager(self, ctx: niobot.Context, operation: str, arg1: str = None):
"""
MSC Cache manager
Available commands: get, download, remove, clear, list
"""
operation, *args = args
if operation == "list":
known = []
total_size = 0
@ -226,7 +225,7 @@ class MSCGetter(niobot.Module):
)
)
elif operation in ["get", "view"]:
file = self.msc_cache / ("{:0>4}.json".format(int(args[0])))
file = self.msc_cache / ("{:0>4}.json".format(int(arg1)))
if file.exists():
data = json.loads(file.read_text())
out = json.dumps(data, indent=4)
@ -235,12 +234,12 @@ class MSCGetter(niobot.Module):
)
return await ctx.respond("%s does not exist." % file.name)
elif operation in ["download", "dl", "fetch"]:
x = await self.get_msc_with_cache(int(args[0]))
x = await self.get_msc_with_cache(int(arg1))
return await ctx.respond(
"```json\n%s\n```" % json.dumps(x, indent=4)
)
elif operation in ["delete", "remove", "rm", "del"]:
file = self.msc_cache / ("{:0>4}.json".format(int(args[0])))
file = self.msc_cache / ("{:0>4}.json".format(int(arg1)))
if file.exists():
file.remove()
return await ctx.respond("Removed cached file %s." % file.name)