Add constants for model field choices.

[matthijs/projects/xerxes.git] / influences / views.py

diff --git a/influences/views.py b/influences/views.py
index 334bfc70d076f71455dc7329e1282c5d1a928d84..8b2961bcb9a5a23b19e806995df18b6366ff4dd3 100644 (file)
--- a/influences/views.py
+++ b/influences/views.py
@@ -20,9 +20,9 @@ from xerxes.tools.misc import make_choices
 def add_influence(request, character_id=None):
     initial = {}
     # Get the current user's characters
-    my_chars = request.user.character_set.all().filter(type__in=['P', 'N'])
+    my_chars = request.user.character_set.all().filter(type__in=[Character.PLAYER, Character.NPC])
     # Get all chars
-    all_chars = Character.objects.all().filter(type__in=['P', 'N'])
+    all_chars = Character.objects.all().filter(type__in=[Character.PLAYER, Character.NPC])
 
     # If a character_id was specified in the url, or there is only one
     # character, preselect it.