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 Related Users

Anyone with an idea on where i'm getting wrong

Add a new model named Relationship. It should have a ForeignKeyField, related to User. The field should be named from_user with a related_name of "relationships"..

models.py
import datetime

from flask.ext.bcrypt import generate_password_hash
from flask.ext.login import UserMixin
from peewee import *

DATABASE = SqliteDatabase(':memory:')


class User(UserMixin, Model):
    email = CharField(unique=True)
    password = CharField(max_length=100)
    join_date = DateTimeField(default=datetime.datetime.now)
    bio = CharField(default='')
    user = ForeignKeyField(
        rel_model=User,
        related_name='new'
    )

    content = TextField()


    class Meta:
        database = DATABASE
        order_by('timestamps')

class Relationship(Model):
    from_user = ForeignKeyField(User, related_name='relationships')
    to_user = ForeignKeyField(User, related_name='new')

    @classmethod
    def new(cls, email, password):
        cls.create(
            email=email,
            password=generate_password_hash(password)
        )


class LunchOrder(Model):
    order = TextField()
    date = DateField()
    user = ForeignKeyField(User, related_name="orders")


def initialize():
    DATABASE.connect()
    DATABASE.create_tables([User, LunchOrder], safe=True)
    DATABASE.close()

1 Answer

Chris Freeman
MOD
Chris Freeman
Treehouse Moderator 68,454 Points

You were on the right track. A typo fixes Task 2. Also, unrelated classmethod should be removed.

class Relationship(Model):
    from_user = ForeignKeyField(User, related_name='relationships')  # Task 1 correct
    to_user = ForeignKeyField(User, related_name='related_to')  # changed related_name to 'related_to' fixes Task 2

    # This block should not be included in this class.
    # @classmethod
    # def new(cls, email, password):
    #     cls.create(
    #         email=email,
    #         password=generate_password_hash(password)
    #     )

Post back if you need more help. Good Luck on Task 3!