hipstercat/wynncraft-bot
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

configure verbose

Browse Source
This commit is contained in:
Amazed 2020-08-31 20:25:45 +02:00
parent b2f03cb54b
commit 0b05407ea0
1 changed files with 29 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
cogs/wynncraft/wynncraftcog.py
View File

@ -14,7 +14,11 @@ class WynncraftCog(commands.Cog):
"tracked": [], "tracked": [],
"channel": None "channel": None
} }
default_global = {
"log": True
}
self.config.register_guild(**default_guild) self.config.register_guild(**default_guild)
self.config.register_global(**default_global)
@commands.command() @commands.command()
@checks.admin_or_permissions(manage_guild=True) @checks.admin_or_permissions(manage_guild=True)
@ -51,20 +55,37 @@ class WynncraftCog(commands.Cog):
await self.config.guild(ctx.guild).channel.set(i) await self.config.guild(ctx.guild).channel.set(i)
await ctx.send("Les messages seront désormais envoyés dans %s" % channel.mention) await ctx.send("Les messages seront désormais envoyés dans %s" % channel.mention)
@commands.command()
@checks.is_owner()
async def log(self, ctx, state):
state = state.lower()
if state == "true" or state == "on":
await self.config.log.set(True)
await ctx.send(":white_check_mark:")
elif state == "false" or state == "off":
await self.config.log.set(False)
await ctx.send(":white_check_mark:")
else:
await ctx.send(":x:")
def _log(self, s):
if await self.config.log():
print(s)
async def loop(self): async def loop(self):
await self.bot.wait_until_ready() await self.bot.wait_until_ready()
last_status = {} last_status = {}
while self is self.bot.get_cog("WynncraftCog"): while self is self.bot.get_cog("WynncraftCog"):
print("Looping guilds") self._log("Looping guilds")
async for guild in self.bot.fetch_guilds(limit=150): async for guild in self.bot.fetch_guilds(limit=150):
guild_id = guild.id guild_id = guild.id
print("Loop id %s" % guild_id) self._log("Loop id %s" % guild_id)
try: try:
channel_id = await self.config.guild(guild).channel() channel_id = await self.config.guild(guild).channel()
if channel_id: if channel_id:
print("Channel set") self._log("Channel set")
for tracked in await self.config.guild(guild).tracked(): for tracked in await self.config.guild(guild).tracked():
print("calling API for player %s" % tracked) self._log("calling API for player %s" % tracked)
r = requests.get("https://api.wynncraft.com/v2/player/%s/stats" % tracked).json() r = requests.get("https://api.wynncraft.com/v2/player/%s/stats" % tracked).json()
for cl in r["data"][0]["classes"]: for cl in r["data"][0]["classes"]:
if guild_id not in last_status: if guild_id not in last_status:
@ -80,11 +101,12 @@ class WynncraftCog(commands.Cog):
channel = self.bot.get_channel(channel_id) channel = self.bot.get_channel(channel_id)
await channel.send(":high_brightness: %s a level up au niveau %s! GG!" % (tracked, cl["professions"]["combat"]["level"])) await channel.send(":high_brightness: %s a level up au niveau %s! GG!" % (tracked, cl["professions"]["combat"]["level"]))
else: else:
print("Player %s had last level %s and current %s" % (tracked, last_status[guild_id][tracked][cl["name"]]["professions"]["combat"]["level"], cl["professions"]["combat"]["level"])) self._log("Player %s had last level %s and current %s" % (tracked, last_status[guild_id][tracked][cl["name"]]["professions"]["combat"]["level"], cl["professions"]["combat"]["level"]))
last_status[guild_id][tracked][cl["name"]] = cl last_status[guild_id][tracked][cl["name"]] = cl
else: else:
print("Guild has no channel set") self._log("Guild has no channel set")
except: except:
if await self.config.log():
traceback.print_exc() traceback.print_exc()
await asyncio.sleep(30) await asyncio.sleep(30)