from django.contrib.auth import get_user_model
from django.contrib.auth.models import Group
from django.db import transaction
from django.db.models.signals import m2m_changed
from django.dispatch import receiver
from shuup.admin.modules.orders.receivers import handle_custom_payment_return_requests
from shuup.admin.signals import object_saved
from shuup.core import cache
from shuup.core.models import Product, ShopProduct
from shuup.core.order_creator.signals import order_creator_finished
from shuup.core.tasks import run_task
[docs]
@receiver(m2m_changed, sender=get_user_model().groups.through)
def on_user_groups_change(instance, action, model, **kwargs):
# TODO : Create a test for this receiver to ensure it behaves as expected.
from shuup.admin.utils.permissions import USER_PERMISSIONS_CACHE_NAMESPACE
# a group has changed it's users relation through group.users.set()
# then we need to bump the entire cache
if isinstance(instance, Group):
cache.bump_version(USER_PERMISSIONS_CACHE_NAMESPACE)
# bump only the user's permission cache
elif isinstance(instance, get_user_model()):
cache.bump_version(f"{USER_PERMISSIONS_CACHE_NAMESPACE}:{instance.pk}")
[docs]
@receiver(object_saved)
def on_object_saved(sender, object, **kwargs):
# make sure to index the prices of the product when a product is saved
if isinstance(object, ShopProduct):
transaction.on_commit(
lambda: run_task("shuup.core.catalog.tasks.index_shop_product", shop_product_id=object.pk)
)
if isinstance(object, Product):
transaction.on_commit(lambda: run_task("shuup.core.catalog.tasks.index_product", product_id=object.pk))
order_creator_finished.connect(
handle_custom_payment_return_requests,
dispatch_uid="shuup.admin.handle_cash_payments",
)