app.py

import os
from flask import Flask, render_template, request, redirect, current_app, abort
import ConfigParser
import stripe
from functools import wraps
from urlparse import urlparse
import json

def secure_required_in_production(fn):
    @wraps(fn)
    def decorated_view(*args, **kwargs):
        if request.is_secure or urlparse(request.url).netloc.startswith('localhost:') and IS_TEST_MODE:
                return fn(*args, **kwargs)
        else:
                return redirect(request.url.replace("http://", "https://"))
        
        abort(500)
            
    return decorated_view



config = ConfigParser.SafeConfigParser()
config.read('settings.conf')

stripe_sk = config.get('stripe_keys', 'secret')
stripe_pk = config.get('stripe_keys', 'publishable')


assert stripe_sk[3:7] == stripe_pk[3:7]
assert stripe_sk[3:7] == 'test' or  stripe_sk[3:7] == 'live'
IS_TEST_MODE = False
if stripe_sk[3:7] == 'test':
  IS_TEST_MODE = True

stripe.api_key = stripe_sk
        
app = Flask(__name__)

@app.context_processor
def inject_user():
    return dict(IS_TEST_MODE=IS_TEST_MODE)
    
#redirect for now since the flyer has the signup URL as our main URL 
@app.route('/signup')
def bounce():
    return redirect('http://appletonmakerspace.org')

@app.route('/signup2')
@secure_required_in_production
def index():
    return render_template('signup.html', key=stripe_pk)

@app.route('/signup_result', methods=['POST'])
@secure_required_in_production
def charge():

    if 'plan' not in request.form or request.form['plan'] == '':
      plan=None
      description='no plan, but a customer account has been created.  Email the treasurer to continue with whatever you are doing.'
    elif request.form['plan'] == 'onetime':
      if 'amount' not in request.form:
        return render_template('error.html', what='server side error parsing amount')
      try:
        stripeToken = json.loads(request.form['stripeToken'])
        a=int(float(request.form['amount'])*100);
        stripe.Charge.create(
          amount=a,
          currency="usd",
            card=stripeToken['id'],
              description="donation/payment: "+request.form['comment']
              )    
        return render_template('signup_complete.html', description="donation/payment of $%.2f sent"%(a/100))
      except:
        return render_template('error.html', what='error charging donation/payment')
    else:
      try:
        plan = stripe.Plan.retrieve(request.form['plan'])
        description='%s at a recurring cost of $%d / %s'%(plan.name, plan.amount/100, plan.interval)
      except:
        return render_template('error.html', what='requested membership plan "%s" not found'%request.form['plan'])

    try:
      stripeToken = json.loads(request.form['stripeToken'])

      customer = stripe.Customer.create(
        email=stripeToken['email'],
        card=stripeToken['id'],
        plan=plan
        )
      if not plan:
        description += ' reference customer# %s'%customer.id
    except:
      return render_template('error.html', what='error adding new customer')

#    # Amount in cents
#    amount = 500
#    charge = stripe.Charge.create(
#        customer=customer.id,
#        amount=amount,
#        currency='usd',
#        description='Flask Charge'
#    )



    return render_template('signup_complete.html', description='are signed up for '+description)

if __name__ == '__main__':
    app.run(debug=True)