Produits Entr'ouvert » Chrono

    def get_opening_hours(self, date):
        if hasattr(self, 'prefetched_timeperiods'):
            timeperiods = self.prefetched_timeperiods
        else:
            timeperiods = self.timeperiod_set.all()

        openslots = IntervalSet()
        for timeperiod in timeperiods:
            if timeperiod.weekday != date.weekday():
                continue
            start_datetime = make_aware(datetime.datetime.combine(date, timeperiod.start_time))
            end_datetime = make_aware(datetime.datetime.combine(date, timeperiod.end_time))
            openslots.add(start_datetime, end_datetime)

        if hasattr(self, 'prefetched_exceptions'):
            timeexceptions = self.prefetched_exceptions
        else:
            timeexceptions = self.timeperiodexception_set.all()

        aware_date = make_aware(datetime.datetime(date.year, date.month, date.day))
        exceptions = IntervalSet()
        aware_next_date = aware_date + datetime.timedelta(days=1)
        for exception in timeexceptions:
            if exception.end_datetime < aware_date:
                continue
            if exception.start_datetime > aware_next_date:
                continue
            exceptions.add(exception.start_datetime, exception.end_datetime)

        return [OpeningHour(*time_range) for time_range in (openslots - exceptions)]

J'ai utilisé des prefetch pour les time periods et les exceptions, quitte à devoir filtrer à la main plutôt que passer par un queryset filtrant:
sur la vue month, on appelle cette méthode pour chaque jour affiché, ce qui fait autant de querysets.
(les exceptions étaient déjà prefetch dans la vue, autant les réutiliser dans la méthode get_opening_hours)

Je sais bien, mais j'aime bien en général poser mes prefetch dans un to_attr: c'est plus explicite, à la lecture du code on sait de suite d'où ça vient.
Je voulais éviter qu'on soit dans le futur tentés de remettre un queryset filtré dans la méthode get_opening_hours: si on voit que c'est prefetché alors on sait qu'on ne doit pas y toucher sans y avoir mûrement réfléchi.
Mais, ça se tient, sans to_attr on peut reduire un peu le code :) Du coup j'ai posé des commentaires.

mauvaises pièces jointes :/

Lauréline Guerin a écrit :

Je voulais éviter qu'on soit dans le futur tentés de remettre un queryset filtré dans la méthode

get_opening_hours: si on voit que c'est prefetché alors on sait qu'on ne doit pas y toucher sans y avoir mûrement réfléchi.

Les tests sur le nombre de query évite facilement ce genre d'accident.

commit 1119909e34a94a1adeb282ad3757f34ba21e16e7
Author: Lauréline Guérin <zebuline@entrouvert.com>
Date:   Mon May 25 14:52:08 2020 +0200

    agendas: reduce querysets on day/month views (#43306)

commit 06ee841b111cbc9f5819091f3d518605dd5b0513
Author: Lauréline Guérin <zebuline@entrouvert.com>
Date:   Mon May 25 14:47:11 2020 +0200

    agendas: add tests on queryset for day/month views (#43306)