def update(): current_plan = current_user.subscription.plan active_plan = Subscription.get_plan_by_id(current_plan) new_plan = Subscription.get_new_plan(request.form.keys()) plan = Subscription.get_plan_by_id(new_plan) # Guard against an invalid, missing or identical plan. is_same_plan = new_plan == active_plan['id'] if ((new_plan is not None and plan is None) or is_same_plan) and \ request.method == 'POST': return redirect(url_for('billing.update')) form = UpdateSubscriptionForm(coupon_code=current_user.subscription.coupon) if form.validate_on_submit(): subscription = Subscription() updated = subscription.update(user=current_user, coupon_code=request.form.get( 'coupon_code'), plan=plan.get('id')) if updated: flash(_('Your subscription has been updated.'), 'success') return redirect(url_for('user.settings')) else: flash(_('That coupon code is not valid.'), 'error') return redirect(url_for('billing.update')) return render_template('billing/pricing.html', form=form, plans=settings.STRIPE_PLANS, active_plan=active_plan)
def test_no_credit_change_for_subscription_downgrade(self): """ Same credits for a subscription downgrade. """ credits = 20 current_plan = Subscription.get_plan_by_id('pro') new_plan = Subscription.get_plan_by_id('standard') credits = add_subscription_credits(credits, current_plan, new_plan, None) assert credits == 20
def test_add_credits_to_subscription_upgrade(self): """ Add credits to a subscription upgrade. """ credits = 20 current_plan = Subscription.get_plan_by_id('standard') new_plan = Subscription.get_plan_by_id('pro') credits = add_subscription_credits(credits, current_plan, new_plan, None) assert credits == 45
def test_no_credit_change_for_same_subscription(self): """ Same credits for the same subscription. """ credits = 20 current_plan = Subscription.get_plan_by_id('pro') new_plan = Subscription.get_plan_by_id('pro') may_29_2015 = datetime.datetime(2015, 5, 29, 0, 0, 0) may_29_2015 = pytz.utc.localize(may_29_2015) credits = add_subscription_credits(credits, current_plan, new_plan, may_29_2015) assert credits == 20
def event(): if not request.json: return render_json(406, {'error': 'Mime-type is not application/json'}) if request.json.get('id') is None: return render_json(406, {'error': 'Invalid Stripe event'}) try: safe_event = PaymentEvent.retrieve(request.json.get('id')) parsed_event = Invoice.parse_from_event(safe_event) user = Invoice.prepare_and_save(parsed_event) if parsed_event.get('total') > 0: plan = Subscription.get_plan_by_id(user.subscription.plan) user.add_credits(plan) except InvalidRequestError as e: # We could not parse the event. return render_json(422, {'error': str(e)}) except Exception as e: # Return a 200 because something is really wrong and we want Stripe to # stop trying to fulfill this webhook request. return render_json(200, {'error': str(e)}) return render_json(200, {'success': True})
def create(): if current_user.subscription: flash(_('You already have an active subscription.'), 'info') return redirect(url_for('user.settings')) plan = request.args.get('plan') subscription_plan = Subscription.get_plan_by_id(plan) # Guard against an invalid or missing plan. if subscription_plan is None and request.method == 'GET': flash(_('Sorry, that plan did not exist.'), 'error') return redirect(url_for('billing.pricing')) stripe_key = current_app.config.get('STRIPE_PUBLISHABLE_KEY') form = SubscriptionForm(stripe_key=stripe_key, plan=plan) if form.validate_on_submit(): subscription = Subscription() created = subscription.create(user=current_user, name=request.form.get('name'), plan=request.form.get('plan'), coupon_code=request.form.get( 'coupon_code'), token=request.form.get('stripe_token')) if created: flash(_('Awesome, thanks for subscribing!'), 'success') else: flash(_('You must enable JavaScript for this request.'), 'warning') return redirect(url_for('user.settings')) return render_template('billing/payment_method.html', form=form, plan=subscription_plan)
def update_payment_method(): if not current_user.credit_card: flash(_('You do not have a payment method on file.'), 'error') return redirect(url_for('user.settings')) active_plan = Subscription.get_plan_by_id( current_user.subscription.plan) card = current_user.credit_card stripe_key = current_app.config.get('STRIPE_PUBLISHABLE_KEY') form = SubscriptionForm(stripe_key=stripe_key, plan=active_plan, name=current_user.name) if form.validate_on_submit(): subscription = Subscription() updated = subscription.update_payment_method(user=current_user, credit_card=card, name=request.form.get( 'name'), token=request.form.get( 'stripe_token')) if updated: flash(_('Your payment method has been updated.'), 'success') else: flash(_('You must enable JavaScript for this request.'), 'warning') return redirect(url_for('user.settings')) return render_template('billing/payment_method.html', form=form, plan=active_plan, card_last4=card.last4)