khuxkm
/
badger
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Don't crash when the user trying to transmute has no badges

This commit is contained in:
Robert Miles 2020-06-23 19:38:22 -04:00
parent 72916a23a4
commit 3b42bcaa0f
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
badge.py
View File

@ -67,7 +67,7 @@ class BadgePopulation:
badge_names = list(badge_names)
for badge_name in badge_names:
found=False
for badge in self.badges[user]:
for badge in self.badges.get(user,[]):
if badge.name==badge_name: found=True
if not found:
raise UserDoesntHaveEnoughBadges(f"User {user} does not have a {badge_name}!")
@ -77,7 +77,7 @@ class BadgePopulation:
N = (((1-W)/len([x for x in self.population if x.normal]))+(W/len(self.population)))**-1
for badge_name in badge_names:
taken = False
for badge in self.badges[user]:
for badge in self.badges.get(user,[]):
if not taken and badge.name==badge_name:
taken=True
self.badges[user].remove(badge)