Akumatic/Akuma-Matata
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Made updater commands only callable by owner

Browse Source
This commit is contained in:
Akumatic 2019-10-11 03:15:00 +02:00
parent 8c55959633
commit 78b4b03aa2
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
extensions/updater.py
View File

@ -12,11 +12,15 @@ class Updater(commands.Cog):
self.cfg = self.bot.loadJSON("updater.json", {"token":""}) self.cfg = self.bot.loadJSON("updater.json", {"token":""})
def updaterCfgCheck(self, key : str, default): def updaterCfgCheck(self, key : str, default):
change = False
if key not in self.cfg[key]: if key not in self.cfg[key]:
self.bot.serverCfg[key] = default self.cfg[key] = default
self.bot.writeJSON("updater.json", self.serverCfg) change = True
if change:
self.bot.writeJSON("updater.json", self.cfg)
@commands.command() @commands.command()
@commands.is_owner()
async def setPrivateToken(self, ctx, token: str = None): async def setPrivateToken(self, ctx, token: str = None):
self.updaterCfgCheck("token", "") self.updaterCfgCheck("token", "")
e = discord.Embed(title="<< Set Private Token >>") e = discord.Embed(title="<< Set Private Token >>")
@ -24,20 +28,21 @@ class Updater(commands.Cog):
e.color = discord.Color.red() e.color = discord.Color.red()
e.add_field(name="No Token given", value="Please spcify the token.") e.add_field(name="No Token given", value="Please spcify the token.")
return await ctx.send(embed=e) return await ctx.send(embed=e)
await ctx.message.delete()
e.color = discord.Color.green() e.color = discord.Color.green()
self.cfg["token"] = token self.cfg["token"] = token
self.bot.writeJSON("updater.json", self.serverCfg) self.bot.writeJSON("updater.json", self.cfg)
e.add_field(name="Token set", value="The given token was stored successfully.") e.add_field(name="Token set", value="The given token was stored successfully.")
await ctx.send(embed=e) await ctx.send(embed=e)
def getRequest(self, update: dict): def getRequest(self, update: dict):
self.updaterCfgCheck("token", "")
r = urllib.request.Request(update["url"]) r = urllib.request.Request(update["url"])
if update["private"]: if update["private"] and self.cfg["token"] != "":
r.add_header("Authorization", f"token {self.cfg['token']}") r.add_header("Authorization", f"token {self.cfg['token']}")
return r return r
@commands.command() @commands.command()
@commands.is_owner()
async def update(self, ctx): async def update(self, ctx):
e = discord.Embed(title="<< Updating Modules >>") e = discord.Embed(title="<< Updating Modules >>")
if sys.platform == "win32": if sys.platform == "win32":