AFPy/PonyConf
AFPy/PonyConf
7
2
Fork 3
Code Issues 10 Pull Requests 1 Projects Releases Wiki Activity

program: possibility to use filter

Browse Source
This commit is contained in:
Élie Bouttier 2016-10-18 22:54:36 +02:00
parent d1952324ce
commit af507c500c
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
planning/utils.py
View File

@ -17,13 +17,16 @@ Event = namedtuple('Event', ['talk', 'row', 'rowcount'])
class Program: class Program:
def __init__(self, site, empty_rooms=False): def __init__(self, site, empty_rooms=False, talk_filter=None):
self.talks = Talk.objects.\ self.talks = Talk.objects.\
filter(site=site, room__isnull=False, start_date__isnull=False).\ filter(site=site, room__isnull=False, start_date__isnull=False).\
filter(Q(duration__gt=0) | Q(event__duration__gt=0)).\ filter(Q(duration__gt=0) | Q(event__duration__gt=0)).\
exclude(accepted=False).\ exclude(accepted=False).\
order_by('start_date') order_by('start_date')
if talk_filter:
self.talks = self.talks.filter(talk_filter)
if empty_rooms: if empty_rooms:
self.rooms = Room.objects.filter(site=site) self.rooms = Room.objects.filter(site=site)
else: else: