Skip to content

Commit e1b6611

Browse files
committed
feedback
1 parent deae2ba commit e1b6611

File tree

2 files changed

+8
-7
lines changed

2 files changed

+8
-7
lines changed

codeforlife/user/filters/user.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55

66
import typing as t
77

8+
from django.db.models import Q # isort: skip
89
from django.db.models.query import QuerySet # isort: skip
910
from django_filters import ( # type: ignore[import-untyped] # isort: skip
1011
rest_framework as filters,
@@ -40,8 +41,9 @@ def name_method(
4041
# TODO: use PostgreSQL specific lookup
4142
# https://docs.djangoproject.com/en/5.0/ref/contrib/postgres/lookups/#std-fieldlookup-trigram_similar
4243
return queryset.filter(
43-
first_name__icontains=first_name
44-
) | queryset.filter(last_name__icontains=last_name)
44+
Q(first_name__icontains=first_name)
45+
| Q(last_name__icontains=last_name)
46+
)
4547

4648
def only_teachers__method(
4749
self: FilterSet, queryset: QuerySet[User], _: str, value: bool

codeforlife/user/models/teacher.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99

1010
from common.models import Teacher, TeacherModelManager
1111
from django.db import models
12+
from django.db.models import Q
1213

1314
from .klass import Class
1415
from .school import School
@@ -100,18 +101,16 @@ def school_users(self):
100101
# pylint: disable-next=import-outside-toplevel
101102
from .user import User
102103

103-
return (
104-
# student-users
105-
User.objects.filter(
104+
return User.objects.filter(
105+
Q( # student-users
106106
new_teacher__isnull=True,
107107
**(
108108
{"new_student__class_field__teacher__school": self.school}
109109
if self.is_admin
110110
else {"new_student__class_field__teacher": self}
111111
)
112112
)
113-
# school-teacher-users
114-
| User.objects.filter(
113+
| Q( # school-teacher-users
115114
new_student__isnull=True,
116115
new_teacher__school=self.school,
117116
)

0 commit comments

Comments
 (0)