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 Using Databases in Python Gettin' CRUD-y With It CRUD: Search Function

Posted by sat pan
sat pan
Courses Plus Student 1,009 Points

Return a Object of Class model in ORM

I am using peewee ORM and trying to return a list of records in function "search_challenge()" for selecting the appropriate record ".where()" method query has been used but it gives an error

crud.py 
from models import Challenge


def create_challenge(name, language, steps=1):
    Challenge.create(name=name,
                     language=language,
                     steps=steps)

def search_challenges(name, language):

    return Challenge.select().where(Challenge.name==name,Challenge.language==language)

1 Answer

Kourosh Raeen
Kourosh Raeen
23,733 Points
Kourosh Raeen
Kourosh Raeen
23,733 Points

Hi Sat - You're very close. The name field should contain the name argument rather than being equal to it:

from models import Challenge

def create_challenge(name, language, steps=1):
    Challenge.create(name=name,
                     language=language,
                     steps=steps)

def search_challenges(name, language):
    return Challenge.select().where(Challenge.name.contains(name),Challenge.language==language)