Added date selector to shuttles and traveller counts
This commit is contained in:
@@ -85,10 +85,10 @@ class Bus(models.Model):
|
||||
def __str__(self):
|
||||
return f"{self.route_name}"
|
||||
|
||||
def traveller_count(self):
|
||||
def traveller_count(self, date=None):
|
||||
count = 0
|
||||
for traveller in Traveller.objects.filter(bus_stops__bus=self):
|
||||
if traveller._is_active():
|
||||
if traveller._is_active(date):
|
||||
count += 1
|
||||
return count
|
||||
|
||||
|
||||
Reference in New Issue
Block a user