def charge():
if session.get('logged_in'):
user = helpers.get_user()
try:
amount = 1000 # amount in cents
customer = stripe.Customer.create(
email=user.email, source=request.form['stripeToken'])
stripe.Charge.create(customer=customer.id,
amount=amount,
currency='usd',
description='Discount Optimizer Charge')
helpers.change_user(payment=helpers.payment_token())
user.active = True
return render_template('home.html', user=user)
except stripe.error.StripeError:
return render_template('error.html')
def login():
if not session.get('logged_in'):
form = forms.LoginForm(request.form)
if request.method == 'POST':
username = request.form['username'].lower()
password = request.form['password']
if form.validate():
if helpers.credentials_valid(username, password):
session['logged_in'] = True
session['username'] = username
return json.dumps({'status': 'Login successful'})
return json.dumps({'status': 'Invalid user/pass'})
return json.dumps({'status': 'Both fields required'})
return render_template('login.html', form=form)
user = helpers.get_user()
user.active = user.payment == helpers.payment_token()
user.key = stripe_keys['publishable_key']
return render_template('home.html', user=user)
