Example #1
0
 def test_post_category_not_found(self):
     """ Assert that when a category gets searched on as a POST request,
     all results are returned if there are no results for this category. 
     """
     self.request_data.update({'process_search_btn':True})
     self.set_all_to_same_category()
     haystack_site.get_index(Coupon).update()
     self.response = self.client.post(reverse('all-coupons'),
         self.request_data)
     self.assert_all_returned()
Example #2
0
 def setUp(self):
     super(SearchCouponsTestCase, self).setUp()
     self.slot_list = SLOT_FACTORY.create_slots(create_count=5)
     haystack_site.get_index(Coupon).reindex()
     self.coupon_list = SLOT_FACTORY.get_active_coupons(
         slot_list=self.slot_list)
     self.coupon0 = self.coupon_list[0]
     self.all_coupons = ALL_COUPONS.get_all_coupons(
         Site.objects.get(id=2))[0]
     self.request_data = None
     self.response = None
Example #3
0
 def test_get_category_not_found(self):
     """ Assert for a GET search for a category without any coupons, all
     coupons are returned.
     """
     self.set_all_to_same_category()
     haystack_site.get_index(Coupon).update()
     self.response = self.client.get(reverse('all-coupons'),
         self.request_data)
     self.assertEqual(self.response.status_code, 200)
     LOG.debug("test_get_category_not_found = %s" % self.response.__dict__)
     self.assert_all_returned()
Example #4
0
 def test_get_both_not_found(self):
     """ Assert that when a category and query string gets searched on as a 
     GET request, all coupons are returned because there is not match 
     for this search set. 
     """
     self.request_data = {'q': 'Return all coupons', 'cat': '1'}
     self.set_all_to_same_category()
     haystack_site.get_index(Coupon).update()
     self.response = self.client.get(reverse('all-coupons'),
         self.request_data)
     self.assert_all_returned()
Example #5
0
 def test_get_both_one_found(self):
     """ Assert that when a category and query string gets searched on as a 
     GET request, all matched category results return because the string 
     was not found. 
     """
     self.request_data = {'q': 'No match string', 'cat': '1'}
     self.set_all_to_same_category()       
     self.coupon0.offer.business.categories = [1]
     haystack_site.get_index(Coupon).update()
     self.response = self.client.get(reverse('all-coupons'),
         self.request_data)
     self.assert_one_returned()
    def more_like_this(self, model_instance):
        from haystack.sites import site, NotRegistered

        index = site.get_index(model_instance.__class__)
        field_name = index.get_content_field()
        raw_results = self.conn.more_like_this("id:%s" % self.get_identifier(model_instance), field_name, fl="*,score")
        return self._process_results(raw_results)
Example #7
0
def reindex(instance):
    try:
        index = site.get_index(instance.__class__)
    except NotRegistered:
        return
    if isinstance(index, SearchIndex):
        res = index.enqueue_save(instance)
Example #8
0
def index_all_business_coupons(business):
    """
    Index all of this businesses coupons.
    """
    index = None
    for offer in business.offers.all():
        if not index:
            index = haystack_site.get_index(offer.coupons.model)
        for coupon in offer.coupons.all():
            index.update_object(coupon)
 def handle(self, **options):
     since = datetime.datetime.now() - REINDEX_INTERVAL 
     objects = {}
     for content_type_id, object_id in Visit.objects.filter(timestamp__gte=since).values_list("content_type_id", "object_id"):
         if content_type_id not in objects:
             objects[content_type_id] = [object_id]
         else:
             objects[content_type_id].append(object_id)
     
     for content_type_id, object_ids in objects.items():
         model = ContentType.objects.get(id=content_type_id).model_class()
         index = site.get_index(model)
         for ids in chunks(object_ids, getattr(settings, 'HAYSTACK_BATCH_SIZE', 1000)):
             qs = model.objects.filter(id__in=ids)
             index.backend.update(index, qs)
Example #10
0
    def handle_app(self, app, **options):
        # Cause the default site to load.
        from haystack import handle_registrations
        handle_registrations()
        
        from django.db.models import get_models
        from haystack.sites import site, NotRegistered

        for model in get_models(app):
            try:
                index = site.get_index(model)
            except NotRegistered:
                if self.verbosity >= 2:
                    print "Skipping '%s' - no index." % model
                continue

            extra_lookup_kwargs = {}
            updated_field = index.get_updated_field()
            
            if self.age:
                if updated_field:
                    extra_lookup_kwargs['%s__gte' % updated_field] = datetime.datetime.now() - datetime.timedelta(hours=self.age)
                else:
                    if self.verbosity >= 2:
                        print "No updated date field found for '%s' - not restricting by age." % model.__name__
            
            # DRL_TODO: .select_related() seems like a good idea here but
            #           can cause empty QuerySets. Why?
            qs = index.get_query_set().filter(**extra_lookup_kwargs).order_by(model._meta.pk.name)
            total = qs.count()

            if self.verbosity >= 1:
                print "Indexing %d %s." % (total, smart_str(model._meta.verbose_name_plural))

            for start in range(0, total, self.batchsize):
                end = min(start + self.batchsize, total)
                
                if self.verbosity >= 2:
                    print "  indexing %s - %d of %d." % (start+1, end, total)
                
                # Get a clone of the QuerySet so that the cache doesn't bloat up
                # in memory. Useful when reindexing large amounts of data.
                small_cache_qs = qs.all()
                index.backend.update(index, small_cache_qs[start:end])
Example #11
0
 def save(self, *args, **kwargs):
     super(Offer, self).save(*args, **kwargs)
     index = haystack_site.get_index(Coupon)
     for coupon in self.coupons.all():
         index.update_object(coupon)
     return self
Example #12
0
 def update_index(self):
     """ Update the coupon index for this specific coupon """        
     index = haystack_site.get_index(Coupon)
     index.update_object(self)