Sith/subscription/views.py

30 lines
975 B
Python
Raw Normal View History

from django.shortcuts import render
from django.views.generic.edit import UpdateView, CreateView
2016-02-05 15:59:42 +00:00
from django.views.generic.base import View
from django.core.exceptions import PermissionDenied
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
2016-02-05 15:59:42 +00:00
from core.models import User
2016-03-22 10:42:00 +00:00
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']
class NewSubscription(CanEditMixin, CreateView):
2016-03-22 10:42:00 +00:00
template_name = 'subscription/subscription.jinja'
form_class = SubscriptionForm
2016-03-22 10:42:00 +00:00
def get_initial(self):
if 'member' in self.request.GET.keys():
return {'member': self.request.GET['member']}
return {}