from django.shortcuts import render from django.views.generic.edit import UpdateView, CreateView from django.views.generic.base import View from django.utils.translation import ugettext_lazy as _ from django.core.exceptions import PermissionDenied, ValidationError from django.db import IntegrityError from django import forms from django.forms import Select from django.conf import settings from subscription.models import Subscriber, Subscription from core.views import CanEditMixin, CanEditPropMixin, CanViewMixin from core.models import User def get_subscriber(user): s = Subscriber.objects.filter(pk=user.pk).first() return s class SubscriptionForm(forms.ModelForm): class Meta: model = Subscription fields = ['member', 'subscription_type', 'payment_method'] def __init__(self, *args, **kwargs): super(SubscriptionForm, self).__init__(*args, **kwargs) # Add fields to allow basic user creation self.fields['last_name'] = forms.CharField(max_length=User._meta.get_field('last_name').max_length) self.fields['first_name'] = forms.CharField(max_length=User._meta.get_field('first_name').max_length) self.fields['email'] = forms.EmailField() self.fields.move_to_end('subscription_type') self.fields.move_to_end('payment_method') def clean(self): cleaned_data = super(SubscriptionForm, self).clean() if (cleaned_data.get("member") is None and "last_name" not in self.errors.as_data() and "first_name" not in self.errors.as_data() and "email" not in self.errors.as_data()): self.errors.pop("member", None) if Subscriber.objects.filter(email=cleaned_data.get("email")).first() is not None: self.add_error("email", ValidationError(_("A user with that email address already exists"))) else: u = Subscriber(last_name = self.cleaned_data.get("last_name"), first_name = self.cleaned_data.get("first_name"), email = self.cleaned_data.get("email")) u.generate_username() u.save() cleaned_data["member"] = u elif cleaned_data.get("member") is not None: self.errors.pop("last_name", None) self.errors.pop("first_name", None) self.errors.pop("email", None) if cleaned_data.get("member") is None: # This should be handled here, but it is done in the Subscription model's clean method # TODO investigate why! raise ValidationError(_("You must either choose an existing user or create a new one properly")) return cleaned_data class NewSubscription(CanEditMixin, CreateView): # TODO: this must be able to create a new user if needed template_name = 'subscription/subscription.jinja' form_class = SubscriptionForm def get_initial(self): if 'member' in self.request.GET.keys(): return {'member': self.request.GET['member']} return {} def form_valid(self, form): form.instance.subscription_start = Subscription.compute_start() form.instance.subscription_end = Subscription.compute_end( duration=settings.SITH_SUBSCRIPTIONS[form.instance.subscription_type]['duration'], start=form.instance.subscription_start ) return super(NewSubscription, self).form_valid(form)