mirror of
https://github.com/nottinghamtec/PyRIGS.git
synced 2026-01-16 21:12:13 +00:00
130 lines
4.1 KiB
Python
130 lines
4.1 KiB
Python
__author__ = 'Ghost'
|
|
from django import forms
|
|
from django.utils import formats
|
|
from django.conf import settings
|
|
from django.core import serializers
|
|
from django.contrib.auth.forms import AuthenticationForm, PasswordResetForm
|
|
from registration.forms import RegistrationFormUniqueEmail
|
|
from captcha.fields import ReCaptchaField
|
|
import simplejson
|
|
|
|
from RIGS import models
|
|
|
|
#Registration
|
|
class ProfileRegistrationFormUniqueEmail(RegistrationFormUniqueEmail):
|
|
captcha = ReCaptchaField()
|
|
|
|
class Meta:
|
|
model = models.Profile
|
|
fields = ('username','email','first_name','last_name','initials','phone')
|
|
|
|
def clean_initials(self):
|
|
"""
|
|
Validate that the supplied initials are unique.
|
|
"""
|
|
if models.Profile.objects.filter(initials__iexact=self.cleaned_data['initials']):
|
|
raise forms.ValidationError("These initials are already in use. Please supply different initials.")
|
|
return self.cleaned_data['initials']
|
|
|
|
# Login form
|
|
class LoginForm(AuthenticationForm):
|
|
captcha = ReCaptchaField(label='Captcha')
|
|
|
|
class PasswordReset(PasswordResetForm):
|
|
captcha = ReCaptchaField(label='Captcha')
|
|
|
|
# Events Shit
|
|
class EventForm(forms.ModelForm):
|
|
datetime_input_formats = formats.get_format_lazy("DATETIME_INPUT_FORMATS") + settings.DATETIME_INPUT_FORMATS
|
|
meet_at = forms.DateTimeField(input_formats=datetime_input_formats, required=False)
|
|
access_at = forms.DateTimeField(input_formats=datetime_input_formats, required=False)
|
|
|
|
items_json = forms.CharField()
|
|
|
|
items = {}
|
|
|
|
related_models = {
|
|
'person': models.Person,
|
|
'organisation': models.Organisation,
|
|
'venue': models.Venue,
|
|
'mic': models.Profile,
|
|
'checked_in_by': models.Profile,
|
|
}
|
|
|
|
@property
|
|
def _get_items_json(self):
|
|
items = {}
|
|
for item in self.instance.items.all():
|
|
data = serializers.serialize('json', [item])
|
|
struct = simplejson.loads(data)
|
|
items[item.pk] = simplejson.dumps(struct[0])
|
|
return simplejson.dumps(items)
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(EventForm, self).__init__(*args, **kwargs)
|
|
|
|
self.fields['items_json'].initial = self._get_items_json
|
|
self.fields['start_date'].widget.format = '%Y-%m-%d'
|
|
self.fields['end_date'].widget.format = '%Y-%m-%d'
|
|
|
|
self.fields['access_at'].widget.format = '%Y-%m-%dT%H:%M:%S'
|
|
self.fields['meet_at'].widget.format = '%Y-%m-%dT%H:%M:%S'
|
|
|
|
def init_items(self):
|
|
self.items = self.process_items_json()
|
|
return self.items
|
|
|
|
def process_items_json(self, event=None):
|
|
data = simplejson.loads(self.cleaned_data['items_json'])
|
|
items = {}
|
|
for key in data:
|
|
pk = int(key)
|
|
items[pk] = self._get_or_initialise_item(pk, data[key]['fields'], event)
|
|
|
|
return items
|
|
|
|
def _get_or_initialise_item(self, pk, data, event):
|
|
try:
|
|
item = models.EventItem.objects.get(pk=pk,event=event)
|
|
except models.EventItem.DoesNotExist:
|
|
item = models.EventItem()
|
|
|
|
item.name = data['name']
|
|
item.description = data['description']
|
|
item.quantity = data['quantity']
|
|
item.cost = data['cost']
|
|
item.order = data['order']
|
|
|
|
if (event):
|
|
item.event = event
|
|
item.full_clean()
|
|
else:
|
|
item.full_clean('event')
|
|
|
|
return item
|
|
|
|
def save(self, commit=True):
|
|
m = super(EventForm, self).save(commit=False)
|
|
|
|
if (commit):
|
|
m.save()
|
|
cur_items = m.items.all()
|
|
items = self.process_items_json(m)
|
|
# Delete any unneeded items
|
|
for item in cur_items:
|
|
if item.pk not in items:
|
|
item.delete()
|
|
|
|
for key in items:
|
|
items[key].save()
|
|
|
|
|
|
return m
|
|
|
|
class Meta:
|
|
model = models.Event
|
|
fields = ['is_rig', 'name', 'venue', 'start_time', 'end_date', 'start_date',
|
|
'end_time', 'meet_at', 'access_at', 'description', 'notes', 'mic',
|
|
'person', 'organisation', 'dry_hire', 'checked_in_by', 'status',
|
|
'collector','purchase_order']
|