dynamicweb/nosystemd/models.py

42 lines
1.2 KiB
Python
Raw Normal View History

from django.db import models
from membership.models import StripeCustomer, CustomUser
from utils.models import BillingAddress
class DonatorStatus(models.Model):
ACTIVE = 'active'
CANCELED = 'canceled'
STATUS_CHOICES = (
(ACTIVE, 'Active'),
(CANCELED, 'Canceled')
)
user = models.OneToOneField(CustomUser)
status = models.CharField(choices=STATUS_CHOICES, max_length=10, default=ACTIVE)
@classmethod
def create(cls, user):
cls.objects.get_or_create(user=user)
class Donation(models.Model):
donation = models.FloatField()
donator = models.ForeignKey(StripeCustomer)
created_at = models.DateTimeField(auto_now_add=True)
billing_address = models.ForeignKey(BillingAddress)
last4 = models.CharField(max_length=4)
cc_brand = models.CharField(max_length=10)
stripe_charge_id = models.CharField(max_length=100, null=True)
@classmethod
def create(cls, data):
obj = cls.objects.create(**data)
return obj
def set_stripe_charge(self, stripe_charge):
self.stripe_charge_id = stripe_charge.id
self.last4 = stripe_charge.source.last4
self.cc_brand = stripe_charge.source.brand
self.save