Rename roster methods in portal view

This commit is contained in:
2023-10-17 22:46:42 +02:00
parent 9164ba494f
commit c7fa54e90e
2 changed files with 10 additions and 9 deletions

View File

@@ -3,7 +3,7 @@ from django.urls import path
from portal.views import (
GetData,
GetRoster,
GetRosterFiltered,
GetObjectsFiltered,
GetFlatpage,
GetRollingStock,
GetConsist,
@@ -14,7 +14,7 @@ from portal.views import (
Types,
Books,
GetBook,
SearchRoster,
SearchObjects,
)
urlpatterns = [
@@ -95,22 +95,22 @@ urlpatterns = [
path("bookshelf/book/<uuid:uuid>", GetBook.as_view(), name="book"),
path(
"search",
SearchRoster.as_view(http_method_names=["post"]),
SearchObjects.as_view(http_method_names=["post"]),
name="search",
),
path(
"search/<str:search>/<int:page>",
SearchRoster.as_view(),
SearchObjects.as_view(),
name="search_pagination",
),
path(
"<str:_filter>/<str:search>",
GetRosterFiltered.as_view(),
GetObjectsFiltered.as_view(),
name="filtered",
),
path(
"<str:_filter>/<str:search>/<int:page>",
GetRosterFiltered.as_view(),
GetObjectsFiltered.as_view(),
name="filtered_pagination",
),
path("<uuid:uuid>", GetRollingStock.as_view(), name="rolling_stock"),

View File

@@ -83,7 +83,7 @@ class GetRoster(GetData):
queryset = RollingStock.objects.order_by(*order_by_fields())
class SearchRoster(View):
class SearchObjects(View):
def run_search(self, request, search, _filter, page=1):
site_conf = get_site_conf()
if _filter is None:
@@ -215,11 +215,12 @@ class SearchRoster(View):
return self.get(request, search, page)
class GetRosterFiltered(View):
class GetObjectsFiltered(View):
def run_filter(self, request, search, _filter, page=1):
site_conf = get_site_conf()
if _filter == "type":
title = RollingStockType.objects.get(slug__iexact=search)
title = get_object_or_404(RollingStockType, slug__iexact=search)
query = Q(rolling_class__type__slug__iexact=search)
elif _filter == "company":
title = get_object_or_404(Company, slug__iexact=search)