次のコードがあります
#!/usr/bin/env python
"""doc"""
import peewee
db = peewee.SqliteDatabase(":memory:")
class BaseModel(peewee.Model): # pylint: disable=W0232
"""base model"""
class Meta: # pylint: disable=C0111,W0232,R0903
database = db
class Student(BaseModel):
"""doc"""
name = peewee.CharField()
class Course(BaseModel):
"""doc"""
name = peewee.CharField()
class StudentCourse(BaseModel):
"""doc"""
student = peewee.ForeignKeyField(Student)
course = peewee.ForeignKeyField(Course)
Student.create_table()
Course.create_table()
StudentCourse.create_table()
s1 = Student(name="Student1")
s1.save()
s2 = Student(name="Student2")
s2.save()
s3 = Student(name="Student3")
s3.save()
s4 = Student(name="Student4")
s4.save()
c1 = Course(name="course1")
c1.save()
c2 = Course(name="course2")
c2.save()
c3 = Course(name="course3")
c3.save()
sc21 = StudentCourse(student=s2, course=c1)
sc21.save()
sc22 = StudentCourse(student=s2, course=c2)
sc22.save()
sc23 = StudentCourse(student=s2, course=c3)
sc23.save()
sc31 = StudentCourse(student=s3, course=c1)
sc31.save()
sc32 = StudentCourse(student=s3, course=c2)
sc32.save()
sc41 = StudentCourse(student=s4, course=c1)
sc41.save()
query = (
StudentCourse
.select(StudentCourse, Student, Course)
.join(Course)
.switch(StudentCourse)
.join(Student)
.order_by(Student.name))
last = None
for student_course in query:
student = student_course.student
if student != last:
last = student
print "Student: %s" % student.name
print " - %s" % student_course.course.name
次の出力が生成されます。
Student: Student2
- course1
- course2
- course3
Student: Student3
- course1
- course2
Student: Student4
- course1
これは基本的にpeewee docsの例です
結果をフィルタリングして、特定のサブセットからすべてのコースを受講している学生のみを選択する必要があります。
courses = ["course1", "course2"]
IN演算子を使用して、選択クエリを次のように変更しようとしました。
query = (
StudentCourse
.select(StudentCourse, Student, Course)
.join(Course)
.switch(StudentCourse)
.join(Student)
.where(Course.name << courses)
.order_by(Student.name))
しかし、それは間違った結果をもたらします。つまり、私が間違っていることを意味します:
Student: Student2
- course1
- course2
Student: Student3
- course1
- course2
Student: Student4
- course1
ここには 2 つの間違いがあります。
- Student2のcourse3が印刷されていません。必要です。これは、course3がコースリストにないためだと思います。
- Student4は選択されますが、選択されるべきではありません。 Student4にはcourse1とcourse2の両方が含まれていないにもかかわらず、course1がコースリストに含まれているためです。
たった 1 つのクエリで目的を達成することは可能ですか、それとも学生を反復処理し、それぞれに対してSELECTクエリを作成し、コースのリストをコースフィルターと照合する必要がありますか?
私の目標は、この出力を取得することです:
Student: Student2
- course1
- course2
- course3
Student: Student3
- course1
- course2