update class role

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()