Small subscription fix

This commit is contained in:
Skia
2016-07-06 01:32:34 +02:00
parent 5355492c76
commit bab3a38a2e
2 changed files with 9 additions and 19 deletions

View File

@ -19,7 +19,7 @@ class SubscriptionForm(forms.ModelForm):
model = Subscription
fields = ['member', 'subscription_type', 'payment_method']
class NewSubscription(CanEditMixin, CreateView):
class NewSubscription(CanEditMixin, CreateView): # TODO: this must be able to create a new user if needed
template_name = 'subscription/subscription.jinja'
form_class = SubscriptionForm
@ -27,3 +27,11 @@ class NewSubscription(CanEditMixin, CreateView):
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)