From 44798159e54aa6759ce4db18fdc72bf1e39ed3ee Mon Sep 17 00:00:00 2001
From: Kálmán Viktor <kviktor@cloud.bme.hu>
Date: Wed, 11 Sep 2013 13:17:38 +0200
Subject: [PATCH] network: filter records by type

---
 circle/network/templates/network/record-list.html |  8 ++++++++
 circle/network/views.py                           | 13 +++++++++++++
 2 files changed, 21 insertions(+)

diff --git a/circle/network/templates/network/record-list.html b/circle/network/templates/network/record-list.html
index 8bd4f30..df9c723 100644
--- a/circle/network/templates/network/record-list.html
+++ b/circle/network/templates/network/record-list.html
@@ -15,6 +15,14 @@
     </h1>
 </div>
 
+<ul class="nav nav-pills" style="margin: 5px 0 20px 0;">
+    <li class="disabled"><a href="#">{% trans "Filter by type" %}</a></li>
+    <li {% if not request.GET.type %} class="active"{% endif %}><a href="{{ request.path }}">{% trans "ALL" %}</a></li>
+    {% for type in types %}
+        <li{% if request.GET.type == type.0 %} class="active"{% endif %}><a href="?type={{ type.0 }}">{{ type.0 }}</a></li>
+    {% endfor %}
+</ul>
+
 <div class="table-responsive">
     {% render_table table %}
 </div>
diff --git a/circle/network/views.py b/circle/network/views.py
index d7a6aa3..c9d1800 100644
--- a/circle/network/views.py
+++ b/circle/network/views.py
@@ -420,6 +420,19 @@ class RecordList(SingleTableView):
     template_name = "network/record-list.html"
     table_pagination = False
 
+    def get_context_data(self, **kwargs):
+        context = super(RecordList, self).get_context_data(**kwargs)
+        context['types'] = Record.CHOICES_type
+        return context
+
+    def get_table_data(self):
+        type_id = self.request.GET.get('type')
+        if type_id:
+            data = Record.objects.filter(type=type_id).select_related()
+        else:
+            data = Record.objects.select_related()
+        return data
+
 
 class RecordDetail(UpdateView, SuccessMessageMixin):
     model = Record
--
libgit2 0.26.0