added prof roles

cogs/wynncraft/wynncraftcog.py

@@ -1,4 +1,6 @@
 import asyncio
+import itertools
+
 from redbot.core import commands
 from redbot.core import Config, checks
 from discord import TextChannel
@@ -36,7 +38,65 @@ class WynncraftCog(commands.Cog):
                 "archer": 750097050867204246,
                 "warrior": 750097044433272833
             },
-            "update_msg": 750124413529358467
+            "update_msg": 750124413529358467,
+            "role_professions": {
+                "farming": [
+                    743544309059551292,  # 0-9
+                    750103671077077004,  # 10-19
+                    750103673178423397,  # 20-29
+                    750103675195752528,  # 30-39
+                    750103677011755080,  # 40-49
+                    750103681046675527,  # 50-59
+                    750103683349479452,  # 60-69
+                    750103692182552667,  # 70-79
+                    750103687589920789,  # 80-89
+                    750103685295767734,  # 90-99
+                    750103689536077966,  # 100-109
+                    750103679314558980,  # 110
+                ],
+                "mining": [
+                    743544348821553305,  # 0-9
+                    750105003464851508,  # 10-19
+                    750105011173851176,  # 20-29
+                    750105019260469288,  # 30-39
+                    750104996204380211,  # 40-49
+                    750105001522888874,  # 50-59
+                    750105005759135805,  # 60-69
+                    750105016433377380,  # 70-79
+                    750105008644554783,  # 80-89
+                    750105013749153792,  # 90-99
+                    750104998670762075,  # 100-109
+                    750104994392309801,  # 110
+                ],
+                "fishing": [
+                    743544397152518146,  # 0-9
+                    750105742127923303,  # 10-19
+                    750105731482779779,  # 20-29
+                    750105744774529124,  # 30-39
+                    750105734615662672,  # 40-49
+                    750105754739933345,  # 50-59
+                    750105729284964472,  # 60-69
+                    750105749899837551,  # 70-79
+                    750105752374345769,  # 80-89
+                    750105747219546122,  # 90-99
+                    750105737254011051,  # 100-109
+                    750105739716067488,  # 110
+                ],
+                "woodcutting": [
+                    743920688607264818,  # 0-9
+                    750106425174261771,  # 10-19
+                    750106428253012048,  # 20-29
+                    750106431033835570,  # 30-39
+                    750106433785430126,  # 40-49
+                    750106436419452949,  # 50-59
+                    750106438931579032,  # 60-69
+                    750106441557475430,  # 70-79
+                    750106444493488129,  # 80-89
+                    750106446883979326,  # 90-99
+                    750106449790631937,  # 100-109
+                    750106452277985301,  # 110
+                ]
+            }
         }
         default_global = {
             "log": True
@@ -97,6 +157,7 @@ class WynncraftCog(commands.Cog):
             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()
+                roles_professions = await self.config.guild(guild).role_professions()
                 guild_id = guild.id
                 online_players = []
                 await self._log("Loop id %s" % guild_id)
@@ -111,8 +172,10 @@ class WynncraftCog(commands.Cog):
                         r = requests.get("https://api.wynncraft.com/v2/player/%s/stats" % member_name).json()
 
                         max_combat_lvl_class = None
+                        best_professions = dict.fromkeys(list(roles_professions.keys()), 0)
+
                         if not r["data"]:
-                            await self._log("%s had data %s" % (member_name, r["data"]))
+                            await self._log("%s had empty data %s" % (member_name, r["data"]))
                             continue
                         for cl in r["data"][0]["classes"]:
                             if guild_id not in last_status:
@@ -140,6 +203,11 @@ class WynncraftCog(commands.Cog):
                                 # await self._log("cl: %s, max_combat_lvl_class: %s" % (cl, max_combat_lvl_class))
                                 max_combat_lvl_class = cl
 
+                            # set max_lvl_professions
+                            for prof_name in list(roles_professions.keys()):
+                                if cl["professions"][prof_name]["level"] > best_professions[prof_name]:
+                                    best_professions[prof_name] = cl["professions"][prof_name]["level"]
+
                         if r["data"][0]["meta"]["location"]["online"]:
                             await self._log("%s is online" % member_name)
                             online_players.append(r)
@@ -192,6 +260,27 @@ class WynncraftCog(commands.Cog):
                                 await self._log("remove roles: %s" % roles_to_remove)
                                 await discord_member.remove_roles(*roles_to_remove)
 
+                            # update professions
+                            role_ids_to_add = []
+                            for prof_name in best_professions:
+                                prof_level = best_professions[prof_name]
+                                if prof_level < 5: continue  # do not add roles for professions < lvl 5
+                                prof_role_index = prof_level // 10
+                                if prof_role_index > 10: prof_role_index = 10
+                                role_ids_to_add.append(roles_professions[prof_name][prof_role_index])
+
+                            roles_to_add = []
+                            for role_id_to_add in role_ids_to_add:
+                                roles_to_add.append(guild.get_role(role_id_to_add))
+                            await self._log("Add role: %s" % roles_to_add)
+                            await discord_member.add_roles(*roles_to_add)
+
+                            all_prof_ids = list(itertools.chain.from_iterable([roles_professions[b] for b in roles_professions]))
+                            roles_to_remove = list(filter(lambda role: role.id in all_prof_ids and role.id not in role_id_to_add, 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))
 
@@ -203,7 +292,7 @@ class WynncraftCog(commands.Cog):
                         full_text += s + "\n"
                     if not online_players:
                         full_text = "Personne n'est connecté :disappointed_relieved:\n"
-                    full_text += "\n*Dernière mise à jour: %s*" % datetime.datetime.now().strftime("%d-%m-%Y %H:%M:%S")
+                    full_text += "\n*Dernière mise à jour: %s*\n*Les vraies informations peuvent prendre jusqu'à 5 minutes de plus*" % 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)