Adding specific filters to search #23 #24

Luan
1 parent 006fe4a1
Showing 3 changed files with 146 additions and 68 deletions Show diff stats
src/search/forms.py
src/search/views.py
src/templates/search/search.html
@@ -8,13 +8,35 @@ from django.utils.translation import ugettext_lazy as _
 from haystack.forms import SearchForm
 from accounts.models import User
-from super_archives.models import Message
+from super_archives.models import Message, MailingList
 class ColabSearchForm(SearchForm):
     q = forms.CharField(label=_('Search'), required=False)
     order = forms.CharField(widget=forms.HiddenInput(), required=False)
     type = forms.CharField(required=False, label=_(u'Type'))
+    author = forms.CharField(required=False, label=_(u'Author'))
+    # ticket status
+    tag = forms.CharField(required=False, label=_(u'Status'))
+    # mailinglist tag
+    list = forms.MultipleChoiceField(
+        required=False,
+        label=_(u'Mailinglist'),
+        choices=[(v, v) for v in MailingList.objects.values('name')
+                 for (v, v) in v.items()]
+    )
+    milestone = forms.CharField(required=False, label=_(u'Milestone'))
+    priority = forms.CharField(required=False, label=_(u'Priority'))
+    component = forms.CharField(required=False, label=_(u'Component'))
+    severity = forms.CharField(required=False, label=_(u'Severity'))
+    reporter = forms.CharField(required=False, label=_(u'Reporter'))
+    keywords = forms.CharField(required=False, label=_(u'Keywords'))
+    collaborators = forms.CharField(required=False, label=_(u'Collaborators'))
+    repository_name = forms.CharField(required=False, label=_(u'Repository'))
+    username = forms.CharField(required=False, label=_(u'Username'))
+    name = forms.CharField(required=False, label=_(u'Name'))
+    institution = forms.CharField(required=False, label=_(u'Institution'))
+    role = forms.CharField(required=False, label=_(u'Role'))
     def search(self):
         if not self.is_valid():
@@ -33,25 +55,44 @@ class ColabSearchForm(SearchForm):
             types = self.cleaned_data['type']
             sqs = sqs.filter(type__in=types.split())
-
         if self.cleaned_data['order']:
             for option, dict_order in settings.ORDERING_DATA.items():
                 if self.cleaned_data['order'] == option:
                     if dict_order['fields']:
                         sqs = sqs.order_by(*dict_order['fields'])
-           # if self.cleaned_data['type'] == 'user':
-           #     sqs = self.searchqueryset.models(User)
-           # elif self.cleaned_data['type'] in ['message', 'thread']:
-           #     sqs = self.searchqueryset.models(Message)
-           # elif self.cleaned_data['type'] == 'wiki':
-           #     sqs = self.searchqueryset.models(Wiki)
-           # elif self.cleaned_data['type'] == 'changeset':
-           #     sqs = self.searchqueryset.models(Changeset)
-           # elif self.cleaned_data['type'] == 'ticket':
-           #     sqs = self.searchqueryset.models(Ticket)
-           # else:
-           #     sqs = self.searchqueryset.all()
+        if self.cleaned_data['author']:
+            sqs = sqs.filter(author=self.cleaned_data['author'])
+
+        if self.cleaned_data['milestone']:
+            sqs = sqs.filter(milestone=self.cleaned_data['milestone'])
+        if self.cleaned_data['priority']:
+            sqs = sqs.filter(priority=self.cleaned_data['priority'])
+        if self.cleaned_data['severity']:
+            sqs = sqs.filter(severity=self.cleaned_data['severity'])
+        if self.cleaned_data['reporter']:
+            sqs = sqs.filter(reporter=self.cleaned_data['reporter'])
+        if self.cleaned_data['keywords']:
+            sqs = sqs.filter(keywords=self.cleaned_data['keywords'])
+        if self.cleaned_data['collaborators']:
+            sqs = sqs.filter(collaborators=self.cleaned_data['collaborators'])
+        if self.cleaned_data['repository_name']:
+            sqs = sqs.filter(
+                repository_name=self.cleaned_data['repository_name']
+            )
+        if self.cleaned_data['username']:
+            sqs = sqs.filter(username=self.cleaned_data['username'])
+        if self.cleaned_data['name']:
+            sqs = sqs.filter(name=self.cleaned_data['name'])
+        if self.cleaned_data['institution']:
+            sqs = sqs.filter(institution=self.cleaned_data['institution'])
+        if self.cleaned_data['role']:
+            sqs = sqs.filter(role=self.cleaned_data['role'])
+        if self.cleaned_data['tag']:
+            sqs = sqs.filter(tag=self.cleaned_data['tag'])
+
+        if self.cleaned_data['list']:
+            sqs = sqs.filter(tag__in=self.cleaned_data['list'])
         if self.load_all:
             sqs = sqs.load_all()
 # -*- coding:utf-8 -*-
 from django.conf import settings
+from django.utils.translation import ugettext as _
+
 from haystack.views import SearchView
 class ColabSearchView(SearchView):
     def extra_context(self, *args, **kwargs):
-        # Retornar todos os campos de cada tipo a serem filtrados
-        # retornar os nomes dos campos
-        # retornar os ícones dos tipos
-
-        # a critical point on the system
         types = {
             'wiki': {
                 'icon': 'file',
-                'fields': [
-                    'title', 'description', 'author', 'collaborators',
-                    'created', 'modified',
-                ],
+                'name': _(u'Wiki'),
+                'fields': {'author': _(u'Author')},
             },
-            'discussion': {
+            'thread': {
                 'icon': 'thread',
-                'fields': [
-                    'title', 'description', 'created', 'modified', 'author',
-                    'tag',
-                ],
+                'name': _(u'Discussion'),
+                'fields': {'author': _(u'Author'), 'list': _(u'Mailinglist')},
             },
             'ticket': {
                 'icon': 'ticket',
-                'fields': [
-                    'title', 'description', 'milestone', 'priority',
-                    'component', 'version', 'severity', 'reporter', 'author',
-                    'status', 'keywords', 'collaborators', 'created',
-                    'modified',
-                ],
+                'name': _(u'Ticket'),
+                'fields': {
+                    'milestone': _(u'Milestone'), 'priority': _(u'Priority'),
+                    'component': _(u'Component'), 'severity': _(u'Severity'),
+                    'reporter': _(u'Reporter'), 'author': _(u'Author'),
+                    'tag': _(u'Status'), 'keywords': _(u'Keywords'),
+                    'collaborators': _(u'Collaborators'),
+                },
             },
             'changeset': {
                 'icon': 'changeset',
-                'fields': [
-                    'title', 'author', 'description', 'repository_name',
-                    'created', 'modified',
-                ],
+                'name': _(u'Changeset'),
+                'fields': {'author': _(u'Author'), 'repository_name': _(u'Repository')},
             },
             'user': {
                 'icon': 'user',
-                'fields': [
-                    'title', 'description', 'username', 'name',
-                    'email', 'institution', 'role', 'google_talk', 'webpage',
-                ],
+                'name': _(u'User'),
+                'fields': {'username': _(u'Username'), 'name': _(u'Name'), 'institution': _(u'Institution'), 'role': _(u'Role')},
             },
         }
-        types = self.form.cleaned_data['type']
+
+        try:
+            type_chosen = self.form.cleaned_data.get('type')
+        except AttributeError:
+            type_chosen = ''
+
         return dict(
-            types=types.split(),
-            types_str=types,
+            filters=types.get(type_chosen),
+            type_chosen=type_chosen,
             order_data=settings.ORDERING_DATA
         )
@@ -17,6 +17,45 @@
     <div id="filters" class="hidden-xs hidden-sm col-md-2 col-lg-2">
       <h3>{% trans "Filters" %}</h3>
+      {% if filters %}
+        <ul class="unstyled-list">
+          <li class="selected" title="{% trans "Remove filter" %}">
+            <span class="glyphicon glyphicon-remove"></span>
+              <a href="{% pop_from_get type=type_chosen %}">
+                  <span class="glyphicon glyphicon-{{ filters.icon }}"></span>
+                  {{ filters.name }}
+              </a>
+          </li>
+        </ul>
+        <hr />
+        <form role="form">
+          <input type="hidden" name="q" value="{{ request.GET.q }}" />
+          <input type="hidden" name="type" value="{{ type_chosen }}" />
+
+          {% for field_lookup, field_display in filters.fields.items %}
+            <div class="form-group">
+              <label for="{{ field_lookup }}">{{ field_display }}</label>
+              {% ifequal field_lookup "list" %}
+                <select name="{{ field_lookup }}" class="form-control" multiple>
+                  {% for value, option in form.fields.list.choices %}
+                    <option value="{{ value }}">{{ option }}</option>
+                  {% endfor %}
+                </select>
+              {% else %}
+              <div class="input-group">
+                <span class="input-group-addon">
+                  <input type="checkbox" name="{{ field_lookup }}_check">
+                </span>
+                <input type="text" class="form-control" placeholder="{{ field_display }}" name="{{ field_lookup }}" value="">
+              </div>
+              {% endifequal %}
+            </div>
+          {% endfor %}
+          <input type="submit" class="btn btn-default pull-right" value="{% trans 'Filter' %}" />
+        </form>
+      <br /><hr />
+      {% endif %}
+
       <h4>{% trans "Sort by" %}</h4>
       <ul class="unstyled-list">
         {% for option, dict_order in order_data.items %}
@@ -33,30 +72,32 @@
         {% endfor %}
       </ul>
-      <h4>{% trans "Types" %}</h4>
+      {% if not request.GET.type %}
+        <h4>{% trans "Types" %}</h4>
-      <ul class="unstyled-list">
-        <li>
-          <span class="glyphicon glyphicon-file"></span>
-          <a href="{% append_to_get type='wiki' %}">{% trans "Wiki" %}</a>
-        </li>
-        <li>
-          <span class="glyphicon glyphicon-envelope"></span>
-          <a href="{% append_to_get type='thread' %}">{% trans "Discussion" %}</a>
-        </li>
-        <li>
-          <span class="glyphicon glyphicon-tag"></span>
-          <a href="{% append_to_get type='ticket' %}">{% trans "Ticket" %}</a>
-        </li>
-        <li>
-          <span class="glyphicon glyphicon-align-right"></span>
-          <a href="{% append_to_get type='changeset' %}">{% trans "Changeset" %}</a>
-        </li>
-        <li>
-          <span class="glyphicon glyphicon-user"></span>
-          <a href="{% append_to_get type='user' %}">{% trans "User" %}</a>
-        </li>
-      </ul>
+        <ul class="unstyled-list">
+          <li>
+            <span class="glyphicon glyphicon-file"></span>
+            <a href="{% append_to_get type='wiki' %}">{% trans "Wiki" %}</a>
+          </li>
+          <li>
+            <span class="glyphicon glyphicon-envelope"></span>
+            <a href="{% append_to_get type='thread' %}">{% trans "Discussion" %}</a>
+          </li>
+          <li>
+            <span class="glyphicon glyphicon-tag"></span>
+            <a href="{% append_to_get type='ticket' %}">{% trans "Ticket" %}</a>
+          </li>
+          <li>
+            <span class="glyphicon glyphicon-align-right"></span>
+            <a href="{% append_to_get type='changeset' %}">{% trans "Changeset" %}</a>
+          </li>
+          <li>
+            <span class="glyphicon glyphicon-user"></span>
+            <a href="{% append_to_get type='user' %}">{% trans "User" %}</a>
+          </li>
+        </ul>
+      {% endif %}
     </div>
     <div class="col-lg-10">
	@@ -8,13 +8,35 @@ from django.utils.translation import ugettext_lazy as _		@@ -8,13 +8,35 @@ from django.utils.translation import ugettext_lazy as _
8	from haystack.forms import SearchForm	8	from haystack.forms import SearchForm
9		9
10	from accounts.models import User	10	from accounts.models import User
11	-from super_archives.models import Message	11	+from super_archives.models import Message, MailingList
12		12
13		13
14	class ColabSearchForm(SearchForm):	14	class ColabSearchForm(SearchForm):
15	q = forms.CharField(label=_('Search'), required=False)	15	q = forms.CharField(label=_('Search'), required=False)
16	order = forms.CharField(widget=forms.HiddenInput(), required=False)	16	order = forms.CharField(widget=forms.HiddenInput(), required=False)
17	type = forms.CharField(required=False, label=_(u'Type'))	17	type = forms.CharField(required=False, label=_(u'Type'))
		18	+ author = forms.CharField(required=False, label=_(u'Author'))
		19	+ # ticket status
		20	+ tag = forms.CharField(required=False, label=_(u'Status'))
		21	+ # mailinglist tag
		22	+ list = forms.MultipleChoiceField(
		23	+ required=False,
		24	+ label=_(u'Mailinglist'),
		25	+ choices=[(v, v) for v in MailingList.objects.values('name')
		26	+ for (v, v) in v.items()]
		27	+ )
		28	+ milestone = forms.CharField(required=False, label=_(u'Milestone'))
		29	+ priority = forms.CharField(required=False, label=_(u'Priority'))
		30	+ component = forms.CharField(required=False, label=_(u'Component'))
		31	+ severity = forms.CharField(required=False, label=_(u'Severity'))
		32	+ reporter = forms.CharField(required=False, label=_(u'Reporter'))
		33	+ keywords = forms.CharField(required=False, label=_(u'Keywords'))
		34	+ collaborators = forms.CharField(required=False, label=_(u'Collaborators'))
		35	+ repository_name = forms.CharField(required=False, label=_(u'Repository'))
		36	+ username = forms.CharField(required=False, label=_(u'Username'))
		37	+ name = forms.CharField(required=False, label=_(u'Name'))
		38	+ institution = forms.CharField(required=False, label=_(u'Institution'))
		39	+ role = forms.CharField(required=False, label=_(u'Role'))
18		40
19	def search(self):	41	def search(self):
20	if not self.is_valid():	42	if not self.is_valid():
	@@ -33,25 +55,44 @@ class ColabSearchForm(SearchForm):		@@ -33,25 +55,44 @@ class ColabSearchForm(SearchForm):
33	types = self.cleaned_data['type']	55	types = self.cleaned_data['type']
34	sqs = sqs.filter(type__in=types.split())	56	sqs = sqs.filter(type__in=types.split())
35		57
36	-
37	if self.cleaned_data['order']:	58	if self.cleaned_data['order']:
38	for option, dict_order in settings.ORDERING_DATA.items():	59	for option, dict_order in settings.ORDERING_DATA.items():
39	if self.cleaned_data['order'] == option:	60	if self.cleaned_data['order'] == option:
40	if dict_order['fields']:	61	if dict_order['fields']:
41	sqs = sqs.order_by(*dict_order['fields'])	62	sqs = sqs.order_by(*dict_order['fields'])
42	- # if self.cleaned_data['type'] == 'user':
43	- # sqs = self.searchqueryset.models(User)
44	- # elif self.cleaned_data['type'] in ['message', 'thread']:
45	- # sqs = self.searchqueryset.models(Message)
46	- # elif self.cleaned_data['type'] == 'wiki':
47	- # sqs = self.searchqueryset.models(Wiki)
48	- # elif self.cleaned_data['type'] == 'changeset':
49	- # sqs = self.searchqueryset.models(Changeset)
50	- # elif self.cleaned_data['type'] == 'ticket':
51	- # sqs = self.searchqueryset.models(Ticket)
52	- # else:
53	- # sqs = self.searchqueryset.all()
54		63
		64	+ if self.cleaned_data['author']:
		65	+ sqs = sqs.filter(author=self.cleaned_data['author'])
		66	+
		67	+ if self.cleaned_data['milestone']:
		68	+ sqs = sqs.filter(milestone=self.cleaned_data['milestone'])
		69	+ if self.cleaned_data['priority']:
		70	+ sqs = sqs.filter(priority=self.cleaned_data['priority'])
		71	+ if self.cleaned_data['severity']:
		72	+ sqs = sqs.filter(severity=self.cleaned_data['severity'])
		73	+ if self.cleaned_data['reporter']:
		74	+ sqs = sqs.filter(reporter=self.cleaned_data['reporter'])
		75	+ if self.cleaned_data['keywords']:
		76	+ sqs = sqs.filter(keywords=self.cleaned_data['keywords'])
		77	+ if self.cleaned_data['collaborators']:
		78	+ sqs = sqs.filter(collaborators=self.cleaned_data['collaborators'])
		79	+ if self.cleaned_data['repository_name']:
		80	+ sqs = sqs.filter(
		81	+ repository_name=self.cleaned_data['repository_name']
		82	+ )
		83	+ if self.cleaned_data['username']:
		84	+ sqs = sqs.filter(username=self.cleaned_data['username'])
		85	+ if self.cleaned_data['name']:
		86	+ sqs = sqs.filter(name=self.cleaned_data['name'])
		87	+ if self.cleaned_data['institution']:
		88	+ sqs = sqs.filter(institution=self.cleaned_data['institution'])
		89	+ if self.cleaned_data['role']:
		90	+ sqs = sqs.filter(role=self.cleaned_data['role'])
		91	+ if self.cleaned_data['tag']:
		92	+ sqs = sqs.filter(tag=self.cleaned_data['tag'])
		93	+
		94	+ if self.cleaned_data['list']:
		95	+ sqs = sqs.filter(tag__in=self.cleaned_data['list'])
55		96
56	if self.load_all:	97	if self.load_all:
57	sqs = sqs.load_all()	98	sqs = sqs.load_all()
	@@ -17,6 +17,45 @@		@@ -17,6 +17,45 @@
17	<div id="filters" class="hidden-xs hidden-sm col-md-2 col-lg-2">	17	<div id="filters" class="hidden-xs hidden-sm col-md-2 col-lg-2">
18	<h3>{% trans "Filters" %}</h3>	18	<h3>{% trans "Filters" %}</h3>
19		19
		20	+ {% if filters %}
		21	+ <ul class="unstyled-list">
		22	+ <li class="selected" title="{% trans "Remove filter" %}">
		23	+ <span class="glyphicon glyphicon-remove"></span>
		24	+ <a href="{% pop_from_get type=type_chosen %}">
		25	+ <span class="glyphicon glyphicon-{{ filters.icon }}"></span>
		26	+ {{ filters.name }}
		27	+ </a>
		28	+ </li>
		29	+ </ul>
		30	+ <hr />
		31	+ <form role="form">
		32	+ <input type="hidden" name="q" value="{{ request.GET.q }}" />
		33	+ <input type="hidden" name="type" value="{{ type_chosen }}" />
		34	+
		35	+ {% for field_lookup, field_display in filters.fields.items %}
		36	+ <div class="form-group">
		37	+ <label for="{{ field_lookup }}">{{ field_display }}</label>
		38	+ {% ifequal field_lookup "list" %}
		39	+ <select name="{{ field_lookup }}" class="form-control" multiple>
		40	+ {% for value, option in form.fields.list.choices %}
		41	+ <option value="{{ value }}">{{ option }}</option>
		42	+ {% endfor %}
		43	+ </select>
		44	+ {% else %}
		45	+ <div class="input-group">
		46	+ <span class="input-group-addon">
		47	+ <input type="checkbox" name="{{ field_lookup }}_check">
		48	+ </span>
		49	+ <input type="text" class="form-control" placeholder="{{ field_display }}" name="{{ field_lookup }}" value="">
		50	+ </div>
		51	+ {% endifequal %}
		52	+ </div>
		53	+ {% endfor %}
		54	+ <input type="submit" class="btn btn-default pull-right" value="{% trans 'Filter' %}" />
		55	+ </form>
		56	+ <br /><hr />
		57	+ {% endif %}
		58	+
20	<h4>{% trans "Sort by" %}</h4>	59	<h4>{% trans "Sort by" %}</h4>
21	<ul class="unstyled-list">	60	<ul class="unstyled-list">
22	{% for option, dict_order in order_data.items %}	61	{% for option, dict_order in order_data.items %}
	@@ -33,30 +72,32 @@		@@ -33,30 +72,32 @@
33	{% endfor %}	72	{% endfor %}
34	</ul>	73	</ul>
35		74
36	- <h4>{% trans "Types" %}</h4>	75	+ {% if not request.GET.type %}
		76	+ <h4>{% trans "Types" %}</h4>
37		77
38	- <ul class="unstyled-list">
39	- <li>
40	- <span class="glyphicon glyphicon-file"></span>
41	- <a href="{% append_to_get type='wiki' %}">{% trans "Wiki" %}</a>
42	- </li>
43	- <li>
44	- <span class="glyphicon glyphicon-envelope"></span>
45	- <a href="{% append_to_get type='thread' %}">{% trans "Discussion" %}</a>
46	- </li>
47	- <li>
48	- <span class="glyphicon glyphicon-tag"></span>
49	- <a href="{% append_to_get type='ticket' %}">{% trans "Ticket" %}</a>
50	- </li>
51	- <li>
52	- <span class="glyphicon glyphicon-align-right"></span>
53	- <a href="{% append_to_get type='changeset' %}">{% trans "Changeset" %}</a>
54	- </li>
55	- <li>
56	- <span class="glyphicon glyphicon-user"></span>
57	- <a href="{% append_to_get type='user' %}">{% trans "User" %}</a>
58	- </li>
59	- </ul>	78	+ <ul class="unstyled-list">
		79	+ <li>
		80	+ <span class="glyphicon glyphicon-file"></span>
		81	+ <a href="{% append_to_get type='wiki' %}">{% trans "Wiki" %}</a>
		82	+ </li>
		83	+ <li>
		84	+ <span class="glyphicon glyphicon-envelope"></span>
		85	+ <a href="{% append_to_get type='thread' %}">{% trans "Discussion" %}</a>
		86	+ </li>
		87	+ <li>
		88	+ <span class="glyphicon glyphicon-tag"></span>
		89	+ <a href="{% append_to_get type='ticket' %}">{% trans "Ticket" %}</a>
		90	+ </li>
		91	+ <li>
		92	+ <span class="glyphicon glyphicon-align-right"></span>
		93	+ <a href="{% append_to_get type='changeset' %}">{% trans "Changeset" %}</a>
		94	+ </li>
		95	+ <li>
		96	+ <span class="glyphicon glyphicon-user"></span>
		97	+ <a href="{% append_to_get type='user' %}">{% trans "User" %}</a>
		98	+ </li>
		99	+ </ul>
		100	+ {% endif %}
60	</div>	101	</div>
61		102
62	<div class="col-lg-10">	103	<div class="col-lg-10">