Welcome to the Treehouse Community

The Treehouse Community is a meeting place for developers, designers, and programmers of all backgrounds and skill levels to get support. Collaborate here on code errors or bugs that you need feedback on, or asking for an extra set of eyes on your latest project. Join thousands of Treehouse students and alumni in the community today. (Note: Only Treehouse students can comment or ask questions, but non-students are welcome to browse our conversations.)

Looking to learn something new?

Treehouse offers a seven day free trial for new students. Get access to thousands of hours of content and a supportive community. Start your free trial today.

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,732 Points
Kourosh Raeen
Kourosh Raeen
23,732 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)