Split out online and retail store sections.

This commit is contained in:
2013-04-07 13:38:44 +01:00
parent 42c92b4881
commit f9cc17671d
5 changed files with 96 additions and 2 deletions

View File

@@ -1,4 +1,4 @@
from .stores import StoreListView, StoreDetailView, StoreUpdateView, StoreCreateView
from .stores import StoreListView, OnlineStoreListView, RetailStoreListView, StoreDetailView, StoreUpdateView, StoreCreateView
from .chains import ChainListView, ChainDetailView
from .search import DistanceSearchView
from .claims import ClaimCreateView

View File

@@ -27,6 +27,21 @@ class StoreListView(HaystackSearchListMixin, ListView):
return qs.filter(active=True).select_related('address', 'address__country')
class OnlineStoreListView(StoreListView):
template_name_suffix = '_online_list'
def get_queryset(self):
qs = super(OnlineStoreListView, self).get_queryset()
return qs.filter(store_type__in=[Store.STORE_TYPE_ONLINE, Store.STORE_TYPE_BOTH])
class RetailStoreListView(StoreListView):
def get_queryset(self):
qs = super(RetailStoreListView, self).get_queryset()
return qs.filter(store_type__in=[Store.STORE_TYPE_ONLINE, Store.STORE_TYPE_BOTH])
class StoreDetailView(EditorCheckMixin, DetailView):
model = Store