Filter Alerts by roles

IoT.py

from django.db import models
from django.db.models import Q
from wirelesstag.models import TagManager
from guardian.shortcuts import get_objects_for_user


class AlertQuerySet(models.QuerySet):
    def for_user(self, user):
        if user.groups.filter(name='Company administrators').exists():
            alerts = self.filter(
                sensor_log__tag__manager__company__administrator=user,
                energy_log__tag__manager__company__administrator=user
            )
        else:
            alerts = self.filter(
                Q(sensor_log__tag__manager__in=get_objects_for_user(
                    user,
                    (
                        'view_tagmanager',
                        'change_tagmanager',
                        'delete_tagmanager'
                    ),
                    TagManager,
                    any_perm=True
                )
            ) | Q(
                energy_log__tag__manager__in=get_objects_for_user(
                    user,
                    (
                        'view_tagmanager',
                        'change_tagmanager',
                        'delete_tagmanager'
                    ),
                    TagManager,
                    any_perm=True
                )
            ))
        return alerts