From 8efe6ed009a8ea0617ce5ee02ea041f98798e379 Mon Sep 17 00:00:00 2001 From: Hipstercat Date: Tue, 1 Sep 2020 01:10:39 +0200 Subject: [PATCH] update class role --- cogs/wynncraft/wynncraftcog.py | 49 +++++++++++++++++++++++++++++++++- 1 file changed, 48 insertions(+), 1 deletion(-) diff --git a/cogs/wynncraft/wynncraftcog.py b/cogs/wynncraft/wynncraftcog.py index c3fe2d9..6fc009e 100644 --- a/cogs/wynncraft/wynncraftcog.py +++ b/cogs/wynncraft/wynncraftcog.py @@ -5,6 +5,7 @@ from discord import TextChannel import discord import traceback import requests +import datetime class WynncraftCog(commands.Cog): @@ -35,7 +36,7 @@ class WynncraftCog(commands.Cog): "archer": 750097050867204246, "warrior": 750097044433272833 }, - "update_msg": 123456 + "update_msg": 750124413529358467 } default_global = { "log": True @@ -95,7 +96,9 @@ class WynncraftCog(commands.Cog): await self._log("Looping guilds") for guild in self.bot.guilds: roles_combat_level = await self.config.guild(guild).role_levels() + roles_classes = await self.config.guild(guild).role_class() guild_id = guild.id + online_players = [] await self._log("Loop id %s" % guild_id) try: guild_name = await self.config.guild(guild).guild_name() @@ -137,11 +140,16 @@ class WynncraftCog(commands.Cog): # await self._log("cl: %s, max_combat_lvl_class: %s" % (cl, max_combat_lvl_class)) max_combat_lvl_class = cl + if r["data"][0]["meta"]["location"]["online"] == "true": + online_players.append(r) + await self._log("Setting max_combat_lvl_class role") # set max_combat_lvl_class role discord_member = discord.utils.find(lambda m: m.display_name == member_name, guild.members) if max_combat_lvl_class and discord_member: await self._log("max_combat_lvl_class and discord_member True") + + # update level max_combat_lvl = max_combat_lvl_class["professions"]["combat"]["level"] combat_role_index = max_combat_lvl // 10 if combat_role_index > 10: combat_role_index = 10 @@ -155,8 +163,47 @@ class WynncraftCog(commands.Cog): if roles_to_remove: await self._log("remove roles: %s" % roles_to_remove) await discord_member.remove_roles(*roles_to_remove) + + # update class + class_name = max_combat_lvl_class["name"] + + class_role_id = None + if class_name.startswith("archer") or class_name.startswith("hunter"): + class_role_id = roles_classes["archer"] + if class_name.startswith("mage") or class_name.startswith("darkwizard"): + class_role_id = roles_classes["mage"] + if class_name.startswith("assassin") or class_name.startswith("ninja"): + class_role_id = roles_classes["assassin"] + if class_name.startswith("warrior") or class_name.startswith("knight"): + class_role_id = roles_classes["warrior"] + if class_name.startswith("shaman") or class_name.startswith("skyseer"): + class_role_id = roles_classes["shaman"] + + class_role = guild.get_role(class_role_id) + if class_role not in discord_member.roles: + await self._log("Add role: %s" % class_role) + await discord_member.add_roles(class_role) + + roles_to_remove = list(filter(lambda role: role.id in roles_classes and role.id != class_role_id, discord_member.roles)) + if roles_to_remove: + await self._log("remove roles: %s" % roles_to_remove) + await discord_member.remove_roles(*roles_to_remove) + else: await self._log("max_combat_lvl_class: %s, discord_member: %s (name: %s)" % (max_combat_lvl_class, discord_member, member_name)) + + # update online players + full_text = "" + for online_player in online_players: + s = ":green_circle: **%s** connecté sur %s" % (online_player["data"][0]["username"], online_player["data"][0]["meta"]["location"]["server"]) + full_text += s + "\n" + if not online_players: + full_text = "Personne n'est connecté :disappointed_relieved:" + full_text += "\n*Dernière mise à jour: %s*" % datetime.datetime.now().strftime("%d-%m-%Y %H:%M:%S") + online_channel = self.bot.get_channel(750100968766701708) + update_msg = await online_channel.fetch_message(await self.config.guild(guild).update_msg()) + await update_msg.edit(content=full_text) + except: if await self.config.log(): traceback.print_exc()