<reponame>melisayu/wagtail
from wagtail.admin.views.bulk_action import BulkAction
from wagtail.core import hooks


class BulkActionRegistry:
    def __init__(self):
        self.actions = dict()  # {app_name: {model_name: {action_name: action_class]}}
        self.has_scanned_for_bulk_actions = False

    def _scan_for_bulk_actions(self):
        if not self.has_scanned_for_bulk_actions:
            for action_class in hooks.get_hooks('register_bulk_action'):
                if not issubclass(action_class, BulkAction):
                    raise Exception("{} is not a subclass of {}".format(action_class.__name__, BulkAction.__name__))
                for model in action_class.models:
                    self.actions.setdefault(model._meta.app_label, {})
                    self.actions[model._meta.app_label].setdefault(model._meta.model_name, {})
                    self.actions[model._meta.app_label][model._meta.model_name][action_class.action_type] = action_class
            self.has_scanned_for_bulk_actions = True

    def get_bulk_actions_for_model(self, app_label, model_name):
        self._scan_for_bulk_actions()
        return self.actions.get(app_label, {}).get(model_name, {}).values()

    def get_bulk_action_class(self, app_label, model_name, action_type):
        self._scan_for_bulk_actions()
        return self.actions.get(app_label, {}).get(model_name, {}).get(action_type, None)


bulk_action_registry = BulkActionRegistry()

from wagtail.admin.views.bulk_action import BulkAction
from wagtail.core import hooks


class BulkActionRegistry:
    def __init__(self):
        self.actions = dict()  # {app_name: {model_name: {action_name: action_class]}}
        self.has_scanned_for_bulk_actions = False

    def _scan_for_bulk_actions(self):
        if not self.has_scanned_for_bulk_actions:
            for action_class in hooks.get_hooks('register_bulk_action'):
                if not issubclass(action_class, BulkAction):
                    raise Exception("{} is not a subclass of {}".format(action_class.__name__, BulkAction.__name__))
                for model in action_class.models:
                    self.actions.setdefault(model._meta.app_label, {})
                    self.actions[model._meta.app_label].setdefault(model._meta.model_name, {})
                    self.actions[model._meta.app_label][model._meta.model_name][action_class.action_type] = action_class
            self.has_scanned_for_bulk_actions = True

    def get_bulk_actions_for_model(self, app_label, model_name):
        self._scan_for_bulk_actions()
        return self.actions.get(app_label, {}).get(model_name, {}).values()

    def get_bulk_action_class(self, app_label, model_name, action_type):
        self._scan_for_bulk_actions()
        return self.actions.get(app_label, {}).get(model_name, {}).get(action_type, None)


bulk_action_registry = BulkActionRegistry()
# {app_name: {model_name: {action_name: action_class]}}