Fixes KeyError when there isn't an event in the registry.

pinax · Jun 1, 2012 · a1f25cb · a1f25cb
1 parent fdeb3f4
commit a1f25cb
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/brabeion/internals.py b/brabeion/internals.py
@@ -26,8 +26,9 @@ def register(self, badge):
             self._event_registry.setdefault(event, []).append(badge)
 
     def possibly_award_badge(self, event, **state):
-        for badge in self._event_registry[event]:
-            badge.possibly_award(**state)
+        if event in self._event_registry:
+            for badge in self._event_registry[event]:
+                badge.possibly_award(**state)
 
 
 badges = BadgeCache()
diff --git a/docs/models.txt b/docs/models.txt
@@ -3,7 +3,7 @@ Models in ``brabeion``
 
 .. module:: brabeion.models
 
-.. class:: BadgeAwarded(models.Model)
+.. class:: BadgeAward(models.Model)
 
     .. attribute:: user
 

diff --git a/docs/usage.txt b/docs/usage.txt
@@ -100,7 +100,7 @@ to the user:
 
 By default badges will be awarded at the current time, if you need to overide
 the award time of the badge you can pass a ``force_timestamp`` keyword argument
-to ``possible_award_badge()``.
+to ``possibly_award_badge()``.
 
 Asynchronous Badges
 -------------------