Add missed CountryQuerySet see HEA-651

Author: girumb

apps/common/models.py

@@ -490,6 +490,35 @@ def date_label(self):
         return date_text
 
 
+class CountryQuerySet(SearchQueryMixin, models.QuerySet):
+    """
+    Makes country searchable
+    """
+    def get_search_filter(self, search_term):
+        return (
+            Q(iso3166a2__iexact=search_term)
+            | Q(iso3166a3__iexact=search_term)
+            | Q(name__iexact=search_term)
+            | Q(iso_en_name__iexact=search_term)
+            | Q(iso_en_proper__iexact=search_term)
+            | Q(iso_en_ro_name__iexact=search_term)
+            | Q(iso_en_ro_proper__iexact=search_term)
+            | Q(iso_fr_name__iexact=search_term)
+            | Q(iso_fr_proper__iexact=search_term)
+            | Q(iso_es_name__iexact=search_term)
+        )
+
+class CountryManager(models.Manager):
+    """
+    Custom manager for Country model using CountryQuerySet.
+    """
+    def get_queryset(self):
+        qs = CountryQuerySet(self.model, using=self._db)
+        return qs
+
+    def search(self, search_term):
+        return self.get_queryset().search(search_term)
+
 class Country(models.Model):
     """
     A Country (or dependent territory or special area of geographical interest) included in ISO 3166.
@@ -546,6 +575,7 @@ class Country(models.Model):
         verbose_name=_("ISO Spanish name"),
         help_text=_("The name in Spanish of the Country approved by the ISO 3166 Maintenance Agency"),
     )
+    objects = CountryManager()
 
     def __str__(self):
         return self.iso_en_ro_name if self.iso_en_ro_name else ""