diff --git a/planning/models.py b/planning/models.py index d04afc6..d97e80d 100644 --- a/planning/models.py +++ b/planning/models.py @@ -24,10 +24,14 @@ class Room(models.Model): def get_absolute_url(self): return reverse('list-rooms') + @property + def talks(self): + return self.talk_set.exclude(accepted=False) + @property def talks_by_date(self): - return self.talk_set.filter(start_date__isnull=False).order_by('start_date').all() + return self.talks.filter(start_date__isnull=False).order_by('start_date').all() @property def unscheduled_talks(self): - return self.talk_set.filter(start_date__isnull=True).all() + return self.talks.filter(start_date__isnull=True).all() diff --git a/planning/templates/planning/room_list.html b/planning/templates/planning/room_list.html index 07d6469..d5ece9e 100644 --- a/planning/templates/planning/room_list.html +++ b/planning/templates/planning/room_list.html @@ -21,7 +21,7 @@ {{ room.capacity }} {% trans "place" %}{{ room.capacity|pluralize }} {% if request|staff %} | - {{ room.talk_set.count }} {% trans "talk" %}{{ room.talk_set.count|pluralize }} + {{ room.talks.count }} {% trans "talk" %}{{ room.talks.count|pluralize }} {% if request|orga %} | {% bootstrap_icon "pencil" %} diff --git a/planning/views.py b/planning/views.py index d77b148..d593885 100644 --- a/planning/views.py +++ b/planning/views.py @@ -12,6 +12,7 @@ from .models import Room from .forms import RoomForm from .utils import Program + class RoomMixin(object): def get_queryset(self): return Room.objects.filter(site=get_current_site(self.request)).all()