Improved get_or_create_plan and subscribe_customer_to_plan

This commit is contained in:
PCoder 2017-08-17 13:01:05 +05:30
parent 1a2b0e4e29
commit 783ab5714c

View file

@ -1,8 +1,10 @@
import stripe import stripe
from django.conf import settings from django.conf import settings
from datacenterlight.models import StripePlan from datacenterlight.models import StripePlan
import logging
stripe.api_key = settings.STRIPE_API_PRIVATE_KEY stripe.api_key = settings.STRIPE_API_PRIVATE_KEY
logger = logging.getLogger(__name__)
def handleStripeError(f): def handleStripeError(f):
@ -57,6 +59,8 @@ class StripeUtils(object):
CURRENCY = 'chf' CURRENCY = 'chf'
INTERVAL = 'month' INTERVAL = 'month'
SUCCEEDED_STATUS = 'succeeded' SUCCEEDED_STATUS = 'succeeded'
PLAN_ALREADY_EXISTS = 'Plan already exists'
PLAN_EXISTS_ERROR_MSG = 'Plan {} exists already.\nCreating a local StripePlan now.'
def __init__(self): def __init__(self):
self.stripe = stripe self.stripe = stripe
@ -147,6 +151,7 @@ class StripeUtils(object):
try: try:
stripe_plan_db_obj = StripePlan.objects.get(stripe_plan_id=stripe_plan_id) stripe_plan_db_obj = StripePlan.objects.get(stripe_plan_id=stripe_plan_id)
except StripePlan.DoesNotExist: except StripePlan.DoesNotExist:
try:
stripe_plan = self.stripe.Plan.create( stripe_plan = self.stripe.Plan.create(
amount=amount, amount=amount,
interval=self.INTERVAL, interval=self.INTERVAL,
@ -157,26 +162,33 @@ class StripeUtils(object):
stripe_plan_db_obj = StripePlan.objects.create(stripe_plan_id=stripe_plan_id) stripe_plan_db_obj = StripePlan.objects.create(stripe_plan_id=stripe_plan_id)
else: else:
stripe_plan_db_obj = None stripe_plan_db_obj = None
except stripe.error.InvalidRequestError as e:
if self.PLAN_ALREADY_EXISTS in str(e):
logger.debug(self.PLAN_EXISTS_ERROR_MSG.format(stripe_plan_id))
stripe_plan_db_obj = StripePlan.objects.create(stripe_plan_id=stripe_plan_id)
return stripe_plan_db_obj return stripe_plan_db_obj
@handleStripeError @handleStripeError
def subscribe_customer_to_plan(self, customer, plans): def subscribe_customer_to_plan(self, customer, plans):
""" """
Subscribes the given customer to the list of plans Subscribes the given customer to the list of given plans
:param customer: The stripe customer identifier :param customer: The stripe customer identifier
:param plans: A list of stripe plan id strings to which the customer needs to be subscribed to :param plans: A list of stripe plans. Ref: https://stripe.com/docs/api/python#create_subscription-items
e.g.
plans = [
{
"plan": "dcl-v1-cpu-2-ram-5gb-ssd-10gb",
},
]
:return: The subscription StripeObject :return: The subscription StripeObject
""" """
stripe.Subscription.create(
subscription_result = self.stripe.Subscription.create(
customer=customer, customer=customer,
items=[ items=plans,
{
"plan": "silver-elite-429",
},
],
itemss=[{"plan": plans[index] for index, item in enumerate(plans)}]
) )
return subscription_result
@handleStripeError @handleStripeError
def make_payment(self, customer, amount, token): def make_payment(self, customer, amount, token):