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=request.form.get('coupon_code'), plan=plan.get('id')) if updated: flash('Your subscription has been updated.', 'success') return redirect(url_for('user.settings')) return render_template('billing/pricing.html', form=form, plans=settings.STRIPE_PLANS, active_plan=active_plan)
def test_no_coin_change_for_subscription_downgrade(self): """ Same coins for a subscription downgrade. """ coins = 100 current_plan = Subscription.get_plan_by_id('gold') new_plan = Subscription.get_plan_by_id('bronze') coins = add_subscription_coins(coins, current_plan, new_plan, None) assert coins == 100
def test_add_coins_to_subscription_upgrade(self): """ Add coins to a subscription upgrade. """ coins = 100 current_plan = Subscription.get_plan_by_id('bronze') new_plan = Subscription.get_plan_by_id('gold') coins = add_subscription_coins(coins, current_plan, new_plan, None) assert coins == 590
def test_no_coin_change_for_same_subscription(self): """ Same coins for the same subscription. """ coins = 100 current_plan = Subscription.get_plan_by_id('gold') new_plan = Subscription.get_plan_by_id('gold') may_29_2015 = datetime.datetime(2015, 5, 29, 0, 0, 0) may_29_2015 = pytz.utc.localize(may_29_2015) coins = add_subscription_coins(coins, current_plan, new_plan, may_29_2015) assert coins == 100
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_coins(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=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)
def test_subscribed_user_receives_more_coins(self, users): """ Subscribed user receives more coins. """ user = User.find_by_identity('*****@*****.**') user.add_coins(Subscription.get_plan_by_id('bronze')) assert user.coins == 210