Welcome to the Treehouse Community

Want to collaborate on code errors? Have bugs you need feedback on? Looking for an extra set of eyes on your latest project? Get support with fellow developers, designers, and programmers of all backgrounds and skill levels here with the Treehouse Community! While you're at it, check out some resources Treehouse students have shared here.

Looking to learn something new?

Treehouse offers a seven day free trial for new students. Get access to thousands of hours of content and join thousands of Treehouse students and alumni in the community today.

Start your free trial

Python Build a Social Network with Flask How to Win Friends Follow/Unfollow views

Posted by Shalom Weberman
Shalom Weberman
3,564 Points

Not sure what this code isn't working. It keeps telling me to try again.

I have tried to debug this. Maybe I am over doing it. I don't see what is wrong with the code. Please help me out.

import datetime

from flask import Flask, g, render_template, flash, redirect, url_for from flask.ext.bcrypt import check_password_hash from flask.ext.login import LoginManager, login_user, current_user, login_required, logout_user

import forms import models

app = Flask(name) app.secret_key = 'this is our super secret key. do not share it with anyone!' login_manager = LoginManager() login_manager.init_app(app) login_manager.login_view = 'login'

@login_manager.user_loader def load_user(userid): try: return models.User.select().where( models.User.id == int(userid) ).get() except models.DoesNotExist: return None

@app.before_request def before_request(): g.db = models.DATABASE g.db.connect() g.user = current_user

@app.after_request def after_request(response): g.db.close() return response

@app.route('/register', methods=('GET', 'POST')) def register(): form = forms.SignUpInForm() if form.validate_on_submit(): models.User.new( email=form.email.data, password=form.password.data ) flash("Thanks for registering!") return render_template('register.html', form=form)

@app.route('/login', methods=('GET', 'POST')) def login(): form = forms.SignUpInForm() if form.validate_on_submit(): try: user = models.User.get( models.User.email == form.email.data ) if check_password_hash(user.password, form.password.data): login_user(user) flash("You're now logged in!") else: flash("No user with that email/password combo") except models.DoesNotExist: flash("No user with that email/password combo") return render_template('register.html', form=form)

@app.route('/secret') @login_required def secret(): return "I should only be visible to logged-in users"

@app.route('/logout') def logout(): logout_user() return redirect(url_for('login'))

@app.route('/') def index(): return render_template('index.html')

@app.route('/order', methods=('GET', 'POST')) def order_lunch(): form = forms.LunchOrderForm() if form.validate_on_submit(): models.LunchOrder.create( user=g.user._get_current_object(), date=form.date.data, order=form.order.data.strip() ) return render_template('lunch.html', form=form)

@app.route('/today') @login_required def today(): order = models.LunchOrder.select().where( models.LunchOrder.date == datetime.date.today() & models.LunchOrder.user == g.user._get_current_object() ).get() return render_template('today.html', order=order)

@app.route('/cancel_order/<int:order_id>') @login_required def cancel_order(order_id): try: order = models.LunchOrder.select().where( id=order_id, user=g.user._get_current_object() ).get() except models.DoesNotExist: pass else: order.delete_instance() return redirect(url_for('index'))

@app.route('/follow/<int:user_id>') @login_required def follow(user_id): try: to_user = models.User.get(models.User.user_id) except models.DoesNotExist: pass else: try: models.Relationship.create( from_user = g.user._get_current_object(), to_user = to_user ) except models.IntegrityError: pass else: flash("great") return redirect(url_for('index'))

Nathan Tallack
Nathan Tallack
22,159 Points
Nathan Tallack
Nathan Tallack
22,159 Points

Formats correctly like this: For others that will help you.

import datetime

from flask import Flask, g, render_template, flash, redirect, url_for
from flask.ext.bcrypt import check_password_hash 
from flask.ext.login import LoginManager, login_user, current_user, login_required, logout_user

import forms 
import models

app = Flask(__name__) 
app.secret_key = 'this is our super secret key. do not share it with anyone!' 
login_manager = LoginManager() 
login_manager.init_app(app) 
login_manager.login_view = 'login'

@login_manager.user_loader 
def load_user(userid): 
    try: 
        return models.User.select().where( models.User.id == int(userid) ).get() 
    except models.DoesNotExist: 
        return None

@app.before_request 
def before_request(): 
    g.db = models.DATABASE 
    g.db.connect() 
    g.user = current_user

@app.after_request 
def after_request(response): 
    g.db.close() 
    return response

@app.route('/register', methods=('GET', 'POST')) 
def register(): 
    form = forms.SignUpInForm() 
    if form.validate_on_submit(): 
        models.User.new( email=form.email.data, password=form.password.data ) 
        flash("Thanks for registering!") 
        return render_template('register.html', form=form)

@app.route('/login', methods=('GET', 'POST')) 
def login(): 
    form = forms.SignUpInForm() 
    if form.validate_on_submit(): 
        try: 
            user = models.User.get( models.User.email == form.email.data ) 
            if check_password_hash(user.password, form.password.data): 
                login_user(user) flash("You're now logged in!") 
            else: 
                flash("No user with that email/password combo") 
        except models.DoesNotExist: 
            flash("No user with that email/password combo") 
            return render_template('register.html', form=form)

@app.route('/secret') 
@login_required 
def secret(): 
    return "I should only be visible to logged-in users"

@app.route('/logout') 
def logout(): 
    logout_user() return redirect(url_for('login'))

@app.route('/') 
def index(): 
    return render_template('index.html')

@app.route('/order', methods=('GET', 'POST')) 
def order_lunch(): 
    form = forms.LunchOrderForm() 
    if form.validate_on_submit(): 
        models.LunchOrder.create( user=g.user._get_current_object(), 
                                  date=form.date.data, order=form.order.data.strip() ) 
        return render_template('lunch.html', form=form)

@app.route('/today') 
@login_required 
def today(): 
    order = models.LunchOrder.select().where( models.LunchOrder.date == datetime.date.today() 
                                              & models.LunchOrder.user == g.user._get_current_object() ).get() 
    return render_template('today.html', order=order)

@app.route('/cancel_order/') 
@login_required 
def cancel_order(order_id): 
    try: 
        order = models.LunchOrder.select().where( id=order_id, user=g.user._get_current_object() ).get() 
    except models.DoesNotExist: 
        pass 
    else: order.delete_instance() 
    return redirect(url_for('index'))

@app.route('/follow/') 
@login_required 
def follow(user_id): 
    try: 
        to_user = models.User.get(models.User.user_id) 
    except models.DoesNotExist: 
        pass 
    else: 
        try: 
            models.Relationship.create( from_user = g.user._get_current_object(), to_user = to_user ) 
        except models.IntegrityError: 
            pass 
        else: flash("great") 
    return redirect(url_for('index'))

Sit tight and let me see if I can work out what is wrong. :)

2 Answers

Shalom Weberman
Shalom Weberman
3,564 Points
Shalom Weberman
Shalom Weberman
3,564 Points

I figured it out. In the route I added, I was not finding the correct user. Here is how it should be written:

@app.route("/follow/") 
@login_required
def follow(user_id):
    try:
      to_user = models.User.get(models.User.id==user_id)
    except models.DoesNotExist:
      pass
    else:
      try:
        models.Relationship.create(
          from_user=g.user._get_current_object(),
          to_user= to_user
        )
      except models.IntegrityError:
        pass
    return redirect(url_for('index'))```
Nathan Tallack
Nathan Tallack
22,159 Points
Nathan Tallack
Nathan Tallack
22,159 Points

Not sure if this matters:

Where you have:

@login_manager.user_loader
def load_user(userid):
    try:
        return models.User.select().where( models.User.id == int(userid) ).get()
    except models.DoesNotExist:
        return None

I have:

@login_manager.user_loader
def load_user(userid):
    try:
        return models.User.get(models.User.id == userid)
    except models.DoesNotExist:
        return None

Also, again not sure if this matters, but I am adding a username attribute along with my email and password on my signup (registration) form where below you are only adding email and password. How are you setting your username? I am also sending them to index if they register but back to register if they fail to validate something.

I have:

@app.route('/register', methods=('GET', 'POST'))
def register():
    form = forms.RegisterForm()
    if form.validate_on_submit():
        flash("Yay, you registered!", "success")
        models.User.create_user(
            username=form.username.data,
            email=form.email.data,
            password=form.password.data
        )
        return redirect(url_for('index'))
    return render_template('register.html', form=form)

You have:

@app.route('/register', methods=('GET', 'POST'))
def register():
    form = forms.SignUpInForm()
    if form.validate_on_submit():
        models.User.new( email=form.email.data, password=form.password.data )
        flash("Thanks for registering!")
        return render_template('register.html', form=form)
Shalom Weberman
Shalom Weberman
3,564 Points

Thanks for the reformat. All I did was add the new route at the bottom of what was given to me in the task. I only added from @app.route('/follow/') and below.