def base_context():
context = {}
context['API_KEY'] = api_key
context['user'] = None
if 'username' in session:
context['user'] = users.find_one(username=session['username'])
return context
def register():
context = base_context()
if context['user'] is not None:
return redirect(url_for('index'))
if request.method == 'POST':
form = request.form
if not form.has_key('license'):
first = form['first']
last = form['last']
email = form['email']
password = form['pass']
if users.find_one(username=email) is None:
user = users.insert(first_name=first, last_name=last,
username=email, password=password, is_owner=False, ad_type=0)
session['username'] = user.username
return redirect(url_for('index'))
else:
context['error'] = 'Email already registered in the system'
else:
first = form['first']
last = form['last']
email = form['email']
license = form['license']
password = form['pass']
if users.find_one(username=email) is None:
user = users.insert(first_name=first, last_name=last,
username=email, password=password, licenses=[license],
is_owner=True)
session['username'] = user.username
return redirect(url_for('index'))
return render_template('register.html', **context)
def login():
context = base_context()
if context['user'] is not None:
return redirect(url_for('index'))
if request.method == 'POST':
form = request.form
user = users.find_one(username=form['email'], password=form['pass'])
if user is not None:
session['username'] = user.username
return redirect(url_for('index'))
else:
context['error'] = 'Email and password combination is incorrect'
return render_template('login.html', **context)
def ads_page():
context = base_context()
if context['user'] is None:
redirect(url_for('index'))
if request.method == 'POST':
email = request.form['stripeEmail']
stripe.Charge.create(
amount=20,
currency='usd',
card=request.form['stripeToken'],
description='Charge for ' + email
)
owner = users.find_one(username=email)
owner.ad_type = 1
owner.save()
return redirect(url_for('profile'))
return render_template('ads.html', **context)
def user_profile(uid):
context = base_context()
context['target_user'] = users.find_one(_id=uid)
return render_template('user_profile.html', **context)
