from django.views.generic import TemplateView, CreateView, FormView, DetailView, UpdateView,\ ListView from django.http import HttpResponseRedirect from django.shortcuts import render from django.core.urlresolvers import reverse_lazy, reverse from django.contrib.auth import authenticate, login from django.contrib.auth.mixins import LoginRequiredMixin from django.conf import settings from django.contrib import messages from membership.models import CustomUser, StripeCustomer from utils.stripe_utils import StripeUtils from utils.views import PasswordResetViewMixin, PasswordResetConfirmViewMixin from utils.forms import PasswordResetRequestForm from utils.mailer import BaseEmail from .forms import LoginForm, SignupForm, DonationForm, DonationBillingForm from .models import Donation, DonatorStatus class LandingView(TemplateView): template_name = "nosystemd/landing.html" def get_context_data(self, *args, **kwargs): total_donations_amount = Donation.get_total_donations_amount() context = { 'total_donations_amount': total_donations_amount } return context class LoginView(FormView): template_name = "nosystemd/login.html" form_class = LoginForm success_url = reverse_lazy('nosystemd:landing') def get_success_url(self): next_url = self.request.session.get('next', self.success_url) return next_url def form_valid(self, form): email = form.cleaned_data.get('email') password = form.cleaned_data.get('password') auth_user = authenticate(email=email, password=password) if auth_user: login(self.request, auth_user) return HttpResponseRedirect(self.get_success_url()) return HttpResponseRedirect(self.get_success_url()) def get(self, request, *args, **kwargs): if self.request.user.is_authenticated(): return HttpResponseRedirect(reverse('nosystemd:landing')) return super(LoginView, self).get(request, *args, **kwargs) class SignupView(CreateView): template_name = 'nosystemd/signup.html' model = CustomUser form_class = SignupForm def get_success_url(self): next_url = self.request.POST.get('next', reverse('nosystemd:login')) return next_url def form_valid(self, form): name = form.cleaned_data.get('name') email = form.cleaned_data.get('email') password = form.cleaned_data.get('password') CustomUser.register(name, password, email) auth_user = authenticate(email=email, password=password) login(self.request, auth_user) return HttpResponseRedirect(self.get_success_url()) class PasswordResetView(PasswordResetViewMixin): template_name = 'nosystemd/reset_password.html' success_url = reverse_lazy('nosystemd:login') form_class = PasswordResetRequestForm template_email_path = 'nosystemd/emails/' class PasswordResetConfirmView(PasswordResetConfirmViewMixin): template_name = 'nosystemd/confirm_reset_password.html' success_url = reverse_lazy('nosystemd:login') class DonationView(LoginRequiredMixin, FormView): template_name = 'nosystemd/donation.html' form_class = DonationBillingForm success_url = reverse_lazy('nosystemd:make_donation') def get_login_url(self): return "%s?next=%s" % (reverse('nosystemd:signup'), reverse('nosystemd:make_donation')) def get_context_data(self, **kwargs): context = super(DonationView, self).get_context_data(**kwargs) context.update({ 'stripe_key': settings.STRIPE_API_PUBLIC_KEY }) return context def get(self, request, *args, **kwargs): if DonatorStatus.objects.filter(user=self.request.user).exists(): messages.success(self.request, 'Your already are a monthly contributor') return HttpResponseRedirect(reverse_lazy('nosystemd:donator_status')) return self.render_to_response(self.get_context_data()) def post(self, request, *args, **kwargs): form = self.get_form() if form.is_valid(): context = self.get_context_data() token = form.cleaned_data.get('token') donation_amount = form.cleaned_data.get('donation_amount') # Get or create stripe customer customer = StripeCustomer.get_or_create(email=self.request.user.email, token=token) if not customer: form.add_error("__all__", "Invalid credit card") return self.render_to_response(self.get_context_data(form=form)) # Create Billing Address billing_address = form.save() # Make stripe charge to a customer stripe_utils = StripeUtils() stripe_utils.CURRENCY = 'usd' charge_response = stripe_utils.make_charge(amount=donation_amount, customer=customer.stripe_id) charge = charge_response.get('response_object') # Check if the payment was approved if not charge: context.update({ 'paymentError': charge_response.get('error'), 'form': form }) return render(request, self.template_name, context) # Create a donation charge = charge_response.get('response_object') donation_data = request.POST.copy() donation_data.update({ 'cc_brand': charge.source.brand, 'stripe_charge_id': charge.id, 'last4': charge.source.last4, 'billing_address': billing_address.id, 'donator': customer.id, 'donation': donation_amount }) donation_form = DonationForm(donation_data) if donation_form.is_valid(): donation = donation_form.save() context = { 'donation': donation, 'base_url': "{0}://{1}".format(request.scheme, request.get_host()) } email_data = { 'subject': 'Your donation have been charged', 'to': request.user.email, 'context': context, 'template_name': 'donation_charge', 'template_path': 'nosystemd/emails/' } email = BaseEmail(**email_data) email.send() return HttpResponseRedirect(reverse('nosystemd:donations', kwargs={'pk': donation.id})) else: self.form_invalid(donation_form) else: return self.form_invalid(form) class DonationDetailView(LoginRequiredMixin, DetailView): template_name = "nosystemd/donation_detail.html" context_object_name = "donation" login_url = reverse_lazy('nosystemd:login') model = Donation class DonationListView(LoginRequiredMixin, ListView): template_name = "nosystemd/donations.html" context_object_name = "donations" login_url = reverse_lazy('nosystemd:login') model = Donation def get_queryset(self): queryset = super(DonationListView, self).get_queryset() queryset = queryset.filter(donator__user=self.request.user) return queryset class DonatorStatusDetailView(LoginRequiredMixin, TemplateView): template_name = "nosystemd/donator_status.html" login_url = reverse_lazy('nosystemd:login') model = DonatorStatus def get_context_data(self, **kwargs): context = super(DonatorStatusDetailView, self).get_context_data(**kwargs) context.update({ 'donator_status': self.request.user.donatorstatus if self.request.user.donatorstatus else None }) return context def get(self, request, *args, **kwargs): if not request.user.donatorstatus: HttpResponseRedirect('nosystemd:landing') return super(DonatorStatusDetailView, self).get(request, *args, **kwargs) class ChangeDonatorStatusDetailView(LoginRequiredMixin, UpdateView): template_name = "nosystemd/donator_status.html" context_object_name = "donator_status" login_url = reverse_lazy('nosystemd:login') model = DonatorStatus def get_object(self, queryset=None): return self.request.user.donatorstatus def post(self, *args, **kwargs): donator_status = self.get_object() donator_status.status = DonatorStatus.ACTIVE \ if donator_status.status == DonatorStatus.CANCELED else DonatorStatus.CANCELED donator_status.save() messages.success(self.request, 'Your monthly donation status has been changed.') return HttpResponseRedirect(reverse_lazy('nosystemd:donator_status'))