Only allow users to add NPC or Player characters.
[matthijs/projects/xerxes.git] / tools / forms.py
1 # Mostly based on code from Amelie, a project by Inter-Actief.
2
3 from django.forms.util import ErrorList
4 from django.forms.models import ModelForm
5 from django.forms.forms import Form
6
7 class ContextForm(Form):
8     def __init__(self, data=None, files=None, auto_id='id_%s', prefix=None,
9                  initial=None, request=None, error_class=ErrorList, label_suffix=':'):
10         self.request = request
11         if request and request.method == "POST":
12             if not data:
13                 data = request.POST
14             if not files:
15                 files = request.FILES
16         super(ContextForm, self).__init__(data, files, auto_id, prefix,
17                                        initial, request, error_class, label_suffix)
18
19 class ContextModelForm(ModelForm):
20     def __init__(self, data=None, files=None, auto_id='id_%s', prefix=None,
21                  initial=None, request=None, error_class=ErrorList, label_suffix=':', instance=None):
22         self.request = request
23         if request and request.method == "POST":
24             if not data:
25                 data = request.POST
26             if not files:
27                 files = request.FILES
28         super(ContextModelForm, self).__init__(data, files, auto_id, prefix, initial, 
29                                             error_class, label_suffix, instance)
30
31
32 # vim: set sts=4 sw=4 expandtab: