CodeMirror/Brandon209-Red-bot-Cogs
1
0
Fork 0
mirror of https://github.com/brandons209/Red-bot-Cogs.git synced 2024-05-19 11:52:55 +12:00
Code Issues Projects Releases Wiki Activity

fix some key errors

Browse source
This commit is contained in:
brandons209 2020-02-09 04:06:08 -05:00
parent a38fcec133
commit 17c0a80633
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
costmanager/manager.py
View file

@ -76,12 +76,12 @@ class CostManager(commands.Cog):
return 0
command_data = command_data[command]
charged_roles = set(command_data["role_ids"].keys())
charged_roles = set(command_data.get("role_ids", {}).keys())
found_roles = charged_roles & member_roles
cost = 0
# check user cost
if str(member.id) in command_data["user_ids"].keys():
if str(member.id) in command_data.get("user_ids", {}).keys():
cost = command_data["user_ids"][str(member.id)]
# check role cost, choose lowest cost if mutliple roles found.
elif found_roles:
@ -346,14 +346,14 @@ class CostManager(commands.Cog):
for command_name, data in commands.items():
msg += f"\t{command_name}: {data['cost']}\n"
msg += "\t\tRole Costs:\n"
for role_id, cost in data["role_ids"].items():
for role_id, cost in data.get("role_ids", {}).items():
role = guild.get_role(int(role_id))
if not role:
continue
msg += f"\t\t\t{role.name}: {cost}\n"
msg += "\t\tUser Costs:\n"
for user_id, cost in data["user_ids"].items():
for user_id, cost in data.get("user_ids", {}).items():
user = guild.get_member(int(user_id))
if not user:
continue