Source code for shuup.campaigns.management.commands.rebuild_campaign_caches

# -*- coding: utf-8 -*-
# This file is part of Shuup.
#
# Copyright (c) 2012-2021, Shuup Commerce Inc. All rights reserved.
#
# This source code is licensed under the OSL-3.0 license found in the
# LICENSE file in the root directory of this source tree.
from django.core.management.base import BaseCommand
from itertools import chain

from shuup.campaigns.models import (
    BasketCampaign,
    CatalogCampaign,
    CatalogFilter,
    CategoryFilter,
    ProductFilter,
    ProductTypeFilter,
)
from shuup.campaigns.models.matching import update_matching_catalog_filters


class Command(BaseCommand):
    def handle(self, *args, **options):
        self.resave_campaigns()
        self.rebuild_cache()

    def rebuild_cache(self):
        filters = list(
            chain(
                ProductTypeFilter.objects.all(),
                ProductFilter.objects.all(),
                CategoryFilter.objects.all(),
                CatalogFilter.objects.all(),
            )
        )

        entry_count = len(filters)

        for i, entry in enumerate(filters):
            update_matching_catalog_filters(entry)
            print("Recaching filter %d / %d..." % (i + 1, entry_count))  # noqa

    def resave_campaigns(self):
        campaigns = list(chain(BasketCampaign.objects.all(), CatalogCampaign.objects.all()))

        entry_count = len(campaigns)

        for i, entry in enumerate(campaigns):
            entry.save()
            print("Recaching campaign %d / %d..." % (i + 1, entry_count))  # noqa