dynamicweb/digitalglarus/mixins.py

from django.core.urlresolvers import reverse_lazy
from django.contrib.auth.mixins import LoginRequiredMixin
from django.views.generic import UpdateView
from django.db import models
from django.http import HttpResponseRedirect
from django.contrib import messages


from membership.models import StripeCustomer
from utils.models import BillingAddress


class MembershipRequiredMixin(object):
    membership_redirect_url = None

    def dispatch(self, request, *args, **kwargs):
        from .models import Membership
        if not Membership.is_digitalglarus_active_member(request.user):
            return HttpResponseRedirect(self.membership_redirect_url)

        return super(MembershipRequiredMixin, self).dispatch(request, *args, **kwargs)


class IsNotMemberMixin(object):
    already_member_redirect_url = None

    def dispatch(self, request, *args, **kwargs):
        from .models import Membership
        if Membership.is_digitalglarus_active_member(request.user):
            return HttpResponseRedirect(self.already_member_redirect_url)

        return super(IsNotMemberMixin, self).dispatch(request, *args, **kwargs)


class Ordereable(models.Model):
    customer = models.ForeignKey(StripeCustomer)
    amount = models.FloatField()
    billing_address = models.ForeignKey(BillingAddress)
    created_at = models.DateTimeField(auto_now_add=True)
    approved = models.BooleanField(default=False)
    last4 = models.CharField(max_length=4, blank=True)
    cc_brand = models.CharField(max_length=10, blank=True)
    stripe_charge_id = models.CharField(max_length=100, null=True)

    class Meta:
        abstract = True

    @classmethod
    def create(cls, data):
        stripe_charge = data.pop('stripe_charge', None)
        # del(data['status'])
        instance = cls.objects.create(**data)
        if not stripe_charge:
            return instance
        instance.stripe_charge_id = stripe_charge.id
        instance.last4 = stripe_charge.source.last4
        instance.cc_brand = stripe_charge.source.brand
        instance.save()
        return instance


class ChangeMembershipStatusMixin(LoginRequiredMixin, UpdateView):
    success_message = None
    success_url = reverse_lazy('digitalglarus:membership_orders_list')
    login_url = reverse_lazy('digitalglarus:login')
    fields = '__all__'

    def get_object(self):
        from .models import MembershipOrder
        membership_order = MembershipOrder.objects.\
            filter(customer__user=self.request.user).last()
        if not membership_order:
            raise AttributeError("Membership does not exists")
        membership = membership_order.membership
        return membership

    def post(self, *args, **kwargs):
        membership = self.get_object()
        membership.activate()

        messages.add_message(self.request, messages.SUCCESS, self.success_message)

        return HttpResponseRedirect(self.success_url)