Catch KeyError.

zsiciarz · Jan 24, 2013 · 2cbc197 · 2cbc197
1 parent d5d2e3b
commit 2cbc197
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/envelope/forms.py b/envelope/forms.py
@@ -194,6 +194,6 @@ def get_category_display(self):
         """
         try:
             category = int(self.cleaned_data['category'])
-        except (AttributeError, ValueError):
+        except (AttributeError, ValueError, KeyError):
             category = None
         return dict(self.get_category_choices()).get(category)