def register_post():
if current_user.is_authenticated():
return redirect(url_for('front_page'))
form = RegistrationForm(request.form)
if form.validate():
user_dict = {
"email": form.email.data,
"password": form.password.data
}
u = register_user(**user_dict)
u.created_at=datetime.datetime.utcnow()
u.modified_at=datetime.datetime.utcnow()
u.first_name=form.first_name.data
u.last_name=form.last_name.data
u.ran_through_first_run_wizard=False
custom_questions = dict()
for c in form.custom_fields():
custom_questions[c.id] = c.data
u.custom_questions_json = custom_questions
db.session.add(u)
db.session.commit()
g.user = u
login_user(u, force=True)
flash('You were successfully logged in')
return redirect(url_for('front_page'))
else:
return render('register.html', form=form)
def status_post():
status_update = request.args.get('status')
s = StatusUpdate()
s.status_update = status_update
s.user = current_user
db.session.add(s)
db.session.commit()
db.session.flush()
return render("hootenflaten_status/status.html", status=s)
def register_facebook_account():
if current_user.is_authenticated():
return redirect(url_for('front_page'))
auth_id = session['fb_auth']
auth = FacebookUser.query.filter_by(id=auth_id).first()
registration_form = RegistrationForm( first_name = auth.first_name,
last_name = auth.last_name,
email = auth.email )
return render('facebook_auth/confirm_fb.html', facebook_user=auth, form=registration_form)
def login():
"""View function for login view"""
form = LoginForm()
if form.validate_on_submit():
login_user(form.user, remember=form.remember.data)
after_this_request(_commit)
if not request.json:
return redirect(get_post_login_redirect())
return render('login.html',
login_user_form=form)
def status_comment():
# TODO: Right now if you know the id of a status update and you have an account you can comment on it.
# TODO: ie, there's no real privacy security around this
id = request.args.get('id')
s = StatusUpdate.query.filter_by(id=id).first()
if s is not None:
c = Comment()
c.comment = request.args.get('comment')
c.user = current_user
db.session.add(c)
s.comments.append(c)
db.session.add(s)
db.session.commit()
return render("hootenflaten_status/comment.html", comment=c)
else:
return jsonify(status="FAIL")
def user_profile(user_id):
user = User.query.filter_by(id=user_id).first()
return render('hootenflaten_user_profile/main.html',
profile_user=user)
def front_page():
if current_user.is_authenticated():
return redirect(url_for('main'))
return render('base.html')
def main():
# TODO: Abstract out things that need to be on the front page from apps
# TODO: We now get ALL status updates, we have to cross this against friend lists (or make this a configable option)
statuses = StatusUpdate.query.filter_by(active=True).order_by('-created_at').limit(100)
return render("logged_in_base.html", statuses=statuses)
def register():
if current_user.is_authenticated():
return redirect(url_for('front_page'))
return render('register.html', form=RegistrationForm())
first_name = user_details.get('first_name'),
last_name = user_details.get('last_name'),
facebook_id = user_details.get('id'),
user_name = user_details.get('username'),
email = user_details.get('email'))
db.session.add(auth)
db.session.commit()
if auth.user is not None:
g.user = auth.user
login_user(auth.user, force=True)
flash('You were successfully logged in')
return redirect(url_for('front_page'))
session['fb_auth'] = auth.id
return redirect(url_for('.register_facebook_account'))
except Exception, e:
current_app.logger.exception("Error getting facebook access token: %s" % e)
abort(500)
return render('facebook_auth/confirm_fb.html')
@login_required
@fb_auth.route('/test', methods=['GET'])
def test():
return "Hello you are logged in! -- %s" % current_user.username
