Implement full text search into the Dashboard

Fix #226

jarbas/dashboard/admin.py

@@ -4,6 +4,8 @@
 from brazilnum.cnpj import format_cnpj
 from brazilnum.cpf import format_cpf
 from django.contrib.admin import SimpleListFilter
+from django.contrib.postgres.search import SearchQuery, SearchRank
+from django.db.models import F
 from django.forms.widgets import Widget
 from simple_history.admin import SimpleHistoryAdmin
 
@@ -247,16 +249,7 @@ class ReimbursementModelAdmin(SimpleHistoryAdmin):
         # 'still_available',
     )
 
-    search_fields = (
-        'applicant_id',
-        'cnpj_cpf',
-        'congressperson_name',
-        'document_id',
-        'party',
-        'state',
-        'supplier',
-        'subquota_description',
-    )
+    search_fields = ('search_vector',)
 
     list_filter = (
         SuspiciousListFilter,
@@ -376,5 +369,18 @@ def formfield_for_dbfield(self, db_field, **kwargs):
             kwargs['widget'] = widgets.get(db_field.name)
         return super().formfield_for_dbfield(db_field, **kwargs)
 
+    def get_search_results(self, request, queryset, search_term):
+        if not search_term:
+            return super(ReimbursementModelAdmin, self) \
+                .get_search_results(request, queryset, search_term)
+
+        query = SearchQuery(search_term, config='portuguese')
+        rank = SearchRank(F('search_vector'), query)
+        queryset = Reimbursement.objects.annotate(rank=rank) \
+            .filter(search_vector=query) \
+            .order_by('-rank')
+
+        return queryset, False
+
 
 dashboard.register(Reimbursement, ReimbursementModelAdmin)