diff --git a/manage_events/api/filters.py b/manage_events/api/filters.py index aeb164c..08e7543 100644 --- a/manage_events/api/filters.py +++ b/manage_events/api/filters.py @@ -11,9 +11,9 @@ class EventFilter(filters.FilterSet): location = filters.CharFilter(field_name="venue__address", lookup_expr="icontains") category = filters.CharFilter(method="filter_category") start_date = filters.DateFilter(field_name="start_date", lookup_expr="gte") - end_date = filters.DateFilter(field_name="end_date", lookup_expr="lte") - price_from = filters.DateFilter(field_name="entry_fee", lookup_expr="gte") - price_to = filters.DateFilter(field_name="entry_fee", lookup_expr="lte") + # end_date = filters.DateFilter(field_name="end_date", lookup_expr="lte") + price_from = filters.NumberFilter(field_name="entry_fee", lookup_expr="gte") + price_to = filters.NumberFilter(field_name="entry_fee", lookup_expr="lte") age_group = filters.CharFilter(field_name="age_group", lookup_expr="icontains") class Meta: @@ -23,7 +23,7 @@ class EventFilter(filters.FilterSet): 'location', 'category', 'start_date', - 'end_date', + # 'end_date', 'price_from', 'price_to', 'age_group', diff --git a/manage_events/models.py b/manage_events/models.py index 54619dc..590f1cf 100644 --- a/manage_events/models.py +++ b/manage_events/models.py @@ -34,6 +34,15 @@ class Venue(BaseModel): def __str__(self): return self.title +class AgeGroups(BaseModel): + name = models.CharField(max_length=10, unique=True) + + class Meta: + db_table = "age_group" + + def __str__(self): + return self.name + class EventStatus(models.TextChoices): UPCOMING = "upcoming", "Upcoming"