You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
32 lines
876 B
32 lines
876 B
from django.contrib.auth.models import User
|
|
from django.db import models
|
|
|
|
|
|
class SubscriptionManager(models.Manager):
|
|
|
|
def for_user(self, user):
|
|
try:
|
|
sub = self.get(user_id=user.id)
|
|
except Subscription.DoesNotExist:
|
|
sub = Subscription(user=user)
|
|
sub.save()
|
|
|
|
return sub
|
|
|
|
|
|
class Subscription(models.Model):
|
|
user = models.OneToOneField(User, blank=True, null=True)
|
|
customer_id = models.CharField(max_length=36, blank=True)
|
|
payment_method_token = models.CharField(max_length=35, blank=True)
|
|
subscription_id = models.CharField(max_length=10, blank=True)
|
|
plan_id = models.CharField(max_length=10, blank=True)
|
|
|
|
objects = SubscriptionManager()
|
|
|
|
def price(self):
|
|
if self.plan_id == "P5":
|
|
return 5
|
|
elif self.plan_id == "P20":
|
|
return 20
|
|
|
|
return 0
|