added prof roles

This commit is contained in:
Amazed 2020-09-01 02:08:49 +02:00
parent d4b1586072
commit bca6e5639c

View File

@ -1,4 +1,6 @@
import asyncio import asyncio
import itertools
from redbot.core import commands from redbot.core import commands
from redbot.core import Config, checks from redbot.core import Config, checks
from discord import TextChannel from discord import TextChannel
@ -36,7 +38,65 @@ class WynncraftCog(commands.Cog):
"archer": 750097050867204246, "archer": 750097050867204246,
"warrior": 750097044433272833 "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 = { default_global = {
"log": True "log": True
@ -97,6 +157,7 @@ class WynncraftCog(commands.Cog):
for guild in self.bot.guilds: for guild in self.bot.guilds:
roles_combat_level = await self.config.guild(guild).role_levels() roles_combat_level = await self.config.guild(guild).role_levels()
roles_classes = await self.config.guild(guild).role_class() roles_classes = await self.config.guild(guild).role_class()
roles_professions = await self.config.guild(guild).role_professions()
guild_id = guild.id guild_id = guild.id
online_players = [] online_players = []
await self._log("Loop id %s" % guild_id) 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() r = requests.get("https://api.wynncraft.com/v2/player/%s/stats" % member_name).json()
max_combat_lvl_class = None max_combat_lvl_class = None
best_professions = dict.fromkeys(list(roles_professions.keys()), 0)
if not r["data"]: 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 continue
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:
@ -140,6 +203,11 @@ class WynncraftCog(commands.Cog):
# await self._log("cl: %s, max_combat_lvl_class: %s" % (cl, max_combat_lvl_class)) # await self._log("cl: %s, max_combat_lvl_class: %s" % (cl, max_combat_lvl_class))
max_combat_lvl_class = cl 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"]: if r["data"][0]["meta"]["location"]["online"]:
await self._log("%s is online" % member_name) await self._log("%s is online" % member_name)
online_players.append(r) online_players.append(r)
@ -192,6 +260,27 @@ class WynncraftCog(commands.Cog):
await self._log("remove roles: %s" % roles_to_remove) await self._log("remove roles: %s" % roles_to_remove)
await discord_member.remove_roles(*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: else:
await self._log("max_combat_lvl_class: %s, discord_member: %s (name: %s)" % (max_combat_lvl_class, discord_member, member_name)) 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" full_text += s + "\n"
if not online_players: if not online_players:
full_text = "Personne n'est connecté :disappointed_relieved:\n" 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) online_channel = self.bot.get_channel(750100968766701708)
update_msg = await online_channel.fetch_message(await self.config.guild(guild).update_msg()) update_msg = await online_channel.fetch_message(await self.config.guild(guild).update_msg())
await update_msg.edit(content=full_text) await update_msg.edit(content=full_text)