wbcore 1.46.0__py2.py3-none-any.whl → 1.58.2__py2.py3-none-any.whl

wbcore/contrib/geography/tests/test_viewsets.py

@@ -1,6 +1,100 @@
 import pytest
+from rest_framework import status
+
+from wbcore.contrib.geography.factories import ContinentFactory
+from wbcore.contrib.geography.models import Geography
+from wbcore.contrib.geography.serializers import GeographyModelSerializer
+from wbcore.contrib.geography.viewsets import (
+    GeographyModelViewSet,
+    GeographyRepresentationViewSet,
+)
 
 
 @pytest.mark.django_db
-class TestSpecificSViewsets:
-    pass
+@pytest.mark.with_db
+class TestViewsets:
+    @pytest.fixture
+    def continents(self):
+        return ContinentFactory.create_batch(3)
+
+    @pytest.fixture
+    def continent(self, continents):
+        return continents[0]
+
+    @pytest.fixture
+    def continent_build(self):
+        instance = ContinentFactory.build()
+        return GeographyModelSerializer(instance).data
+
+    @pytest.mark.parametrize("viewset", [GeographyModelViewSet, GeographyRepresentationViewSet])
+    def test_get(self, viewset, request_factory, super_user, continents):
+        # Arrange
+        request = request_factory.get("")
+        request.user = super_user
+        viewset = viewset.as_view({"get": "list"})
+        # Act
+        response = viewset(request)
+        # Assert
+        assert len(response.data["results"]) == 3
+        assert response.status_code == status.HTTP_200_OK
+
+    @pytest.mark.parametrize("viewset", [GeographyModelViewSet, GeographyRepresentationViewSet])
+    def test_retrieve(self, viewset, request_factory, super_user, continent):
+        # Arrange
+        request = request_factory.get("")
+        request.user = super_user
+        viewset = viewset.as_view({"get": "retrieve"})
+        # Act
+        response = viewset(request, pk=continent.id)
+        instance = response.data.get("instance")
+        # Assert
+        assert instance is not None
+        assert instance["id"] == continent.id
+        assert response.status_code == status.HTTP_200_OK
+
+    def test_create(self, request_factory, super_user, continent_build):
+        # Arrange
+        request = request_factory.post("", data=continent_build, format="json")
+        request.user = super_user
+        viewset = GeographyModelViewSet.as_view({"post": "create"})
+        # Act
+        response = viewset(request)
+        # Assert
+        assert response.status_code == status.HTTP_201_CREATED
+
+    def test_delete(self, request_factory, super_user, continents):
+        # Arrange
+        entry_id = continents[1].id
+        request = request_factory.delete("", args=entry_id)
+        request.user = super_user
+        viewset = GeographyModelViewSet.as_view({"delete": "destroy"})
+        # Act
+        response = viewset(request, pk=entry_id)
+        # Assert
+        assert response.status_code == status.HTTP_204_NO_CONTENT
+        assert Geography.objects.count() == 2
+        assert not Geography.objects.filter(id=entry_id).exists()
+
+    def test_put(self, request_factory, super_user, continent, continent_build):
+        # Arrange
+        continent_build["id"] = continent.id
+        request = request_factory.put("", data=continent_build, format="json")
+        request.user = super_user
+        viewset = GeographyModelViewSet.as_view({"put": "update"})
+        # Act
+        response = viewset(request, pk=continent.id)
+        # Assert
+        assert response.status_code == status.HTTP_200_OK
+
+    def test_patch(self, request_factory, super_user, continent):
+        # Arrange
+        new_field_data = "Foo Bar"
+        request = request_factory.patch("", data={"name": new_field_data})
+        request.user = super_user
+        viewset = GeographyModelViewSet.as_view({"patch": "partial_update"})
+        # Act
+        response = viewset(request, pk=continent.id)
+        continent.refresh_from_db()
+        # Assert
+        assert response.status_code == status.HTTP_200_OK
+        assert continent.name == new_field_data

wbcore/contrib/guardian/tests/test_model_mixins.py

@@ -4,8 +4,7 @@ from wbcore.contrib.guardian.models.mixins import PermissionObjectModelMixin
 
 @pytest.fixture
 def mocked_permission_object_model_mixin(mocker):
-    def __init__(self):
-        ...
+    def __init__(self): ...  # noqa
 
     PermissionObjectModelMixin.__init__ = __init__
     PermissionObjectModelMixin.objects = mocker.Mock()
@@ -22,8 +21,8 @@ class TestPermissionObjectModelMixin:
     def test_save_run_assign_permissions(self, mocker, mocked_permission_object_model_mixin):
         mocker.patch("django.db.models.Model.save")
         on_commit = mocker.patch("wbcore.contrib.guardian.models.mixins.transaction.on_commit")
-        ContentType = mocker.patch("wbcore.contrib.guardian.models.mixins.ContentType")
-        ContentType.objects.get_for_model.return_value = mocker.Mock()
+        content_type_class = mocker.patch("wbcore.contrib.guardian.models.mixins.ContentType")
+        content_type_class.objects.get_for_model.return_value = mocker.Mock()
 
         mocked_permission_object_model_mixin().save()
 

wbcore/contrib/guardian/tests/test_tasks.py

@@ -8,39 +8,39 @@ from wbcore.contrib.guardian.models.mixins import (
 
 class TestAssignUserPermissionsForObjectAsTask:
     def test_called(self, mocker):
-        ContentType = mocker.patch("wbcore.contrib.guardian.models.mixins.ContentType")
+        content_type_class = mocker.patch("wbcore.contrib.guardian.models.mixins.ContentType")
         content_type = mocker.Mock()
         model_class = mocker.Mock()
         instance = mocker.Mock()
         model_class.objects.get.return_value = instance
         content_type.model_class.return_value = model_class
-        ContentType.objects.get.return_value = content_type
+        content_type_class.objects.get.return_value = content_type
 
         assign_user_permissions_for_object_as_task(1, 1)
 
         instance.reload_permissions.assert_called_once_with(prune_existing=True)
 
     def test_called_with_prune_existing(self, mocker):
-        ContentType = mocker.patch("wbcore.contrib.guardian.models.mixins.ContentType")
+        content_type_class = mocker.patch("wbcore.contrib.guardian.models.mixins.ContentType")
         content_type = mocker.Mock()
         model_class = mocker.Mock()
         instance = mocker.Mock()
         model_class.objects.get.return_value = instance
         content_type.model_class.return_value = model_class
-        ContentType.objects.get.return_value = content_type
+        content_type_class.objects.get.return_value = content_type
 
         assign_user_permissions_for_object_as_task(1, 1, prune_existing=False)
 
         instance.reload_permissions.assert_called_once_with(prune_existing=False)
 
     def test_not_called_without_model_class(self, mocker):
-        ContentType = mocker.patch("wbcore.contrib.guardian.models.mixins.ContentType")
+        content_type_class = mocker.patch("wbcore.contrib.guardian.models.mixins.ContentType")
         content_type = mocker.Mock()
         model_class = mocker.Mock()
         instance = mocker.Mock()
         model_class.objects.get.return_value = instance
         content_type.model_class.return_value = None
-        ContentType.objects.get.return_value = content_type
+        content_type_class.objects.get.return_value = content_type
 
         assign_user_permissions_for_object_as_task(1, 1)
 
@@ -63,13 +63,13 @@ class TestAssignUserPermissionsForObjectAsTask:
 
 class TestAssignObjectPermissionsForUserAsTask:
     def test_called(self, mocker):
-        User = mocker.patch("wbcore.contrib.guardian.models.mixins.User")
+        user_class = mocker.patch("wbcore.contrib.guardian.models.mixins.User")
         user = mocker.Mock()
 
-        User.objects.get.return_value = user
+        user_class.objects.get.return_value = user
 
         get_inheriting_subclasses = mocker.patch("wbcore.contrib.guardian.models.mixins.get_inheriting_subclasses")
-        get_inheriting_subclasses.return_value = [User]
+        get_inheriting_subclasses.return_value = [user_class]
         reload_permissions = mocker.patch("wbcore.contrib.guardian.models.mixins.reload_permissions")
 
         assign_object_permissions_for_user_as_task(user_id=1)

wbcore/contrib/guardian/tests/test_viewsets.py

@@ -10,7 +10,7 @@ class TestPivotUserObjectPermissionModelViewSet:
     def test_cached_property_permissions(self, mocker, request):
         mocked_filter = mocker.patch("wbcore.contrib.guardian.viewsets.viewsets.Permission.objects.filter")
         view = PivotUserObjectPermissionModelViewSet(request=request, kwargs={"content_type_id": 1})
-        view.permissions
+        assert view.permissions
 
         mocked_filter.assert_called_once_with(
             Q(content_type=1)
@@ -24,7 +24,7 @@ class TestPivotUserObjectPermissionModelViewSet:
         mocked_get_object_for_this_type = mocker.Mock()
         mocked_contrib_type.get.return_value = mocked_get_object_for_this_type
         view = PivotUserObjectPermissionModelViewSet(request=request, kwargs={"content_type_id": 1, "object_pk": 1})
-        view.linked_object
+        assert view.linked_object
 
         mocked_get_object_for_this_type.get_object_for_this_type.assert_called_once()
 

wbcore/contrib/guardian/viewsets/configs/__init__.py

@@ -1,4 +1,4 @@
-from .buttons import add_object_permission_button
+from .buttons import add_object_permission_button, PivotUserObjectPermissionButtonViewConfig
 from .displays import PivotUserObjectPermissionDisplayViewConfig
 from .endpoints import PivotUserObjectPermissionEndpointViewConfig
 from .titles import PivotUserObjectPermissionTitleViewConfig

wbcore/contrib/guardian/viewsets/configs/buttons.py

@@ -4,10 +4,19 @@ from django.dispatch import receiver
 from rest_framework.reverse import reverse
 from wbcore.contrib.guardian.models.mixins import PermissionObjectModelMixin
 from wbcore.contrib.icons.icons import WBIcon
+from wbcore.metadata.configs.buttons import ButtonViewConfig
 from wbcore.metadata.configs.buttons.buttons import WidgetButton
+from wbcore.metadata.configs.buttons.enums import Button
 from wbcore.signals.instance_buttons import add_extra_button
 
 
+class PivotUserObjectPermissionButtonViewConfig(ButtonViewConfig):
+    def get_create_buttons(self):
+        return {
+            Button.SAVE_AND_CLOSE.value,
+        }
+
+
 @receiver(add_extra_button)
 def add_object_permission_button(sender, instance, request, view, pk=None, **kwargs):
     with suppress(AttributeError):

wbcore/contrib/guardian/viewsets/configs/endpoints.py

@@ -3,6 +3,13 @@ from wbcore.metadata.configs.endpoints import EndpointViewConfig
 
 
 class PivotUserObjectPermissionEndpointViewConfig(EndpointViewConfig):
+    def get_instance_endpoint(self, **kwargs):
+        return reverse(
+            "wbcore:guardian:pivoteduserobjectpermission-list",
+            args=[self.view.kwargs.get("content_type_id", None), self.view.kwargs.get("object_pk", None)],
+            request=self.request,
+        )
+
     def get_update_endpoint(self, **kwargs):
         return reverse(
             "wbcore:guardian:pivoteduserobjectpermission-list",

wbcore/contrib/guardian/viewsets/viewsets.py

@@ -8,6 +8,7 @@ from wbcore.contrib.authentication.models.users import Permission, User
 from wbcore.contrib.authentication.serializers import UserRepresentationSerializer
 from wbcore.contrib.guardian.models import UserObjectPermission
 from wbcore.contrib.guardian.viewsets.configs import (
+    PivotUserObjectPermissionButtonViewConfig,
     PivotUserObjectPermissionDisplayViewConfig,
     PivotUserObjectPermissionEndpointViewConfig,
     PivotUserObjectPermissionTitleViewConfig,
@@ -17,6 +18,7 @@ from wbcore.contrib.guardian.viewsets.configs import (
 class PivotUserObjectPermissionModelViewSet(viewsets.ModelViewSet):
     queryset = UserObjectPermission.objects.all()
 
+    button_config_class = PivotUserObjectPermissionButtonViewConfig
     display_config_class = PivotUserObjectPermissionDisplayViewConfig
     endpoint_config_class = PivotUserObjectPermissionEndpointViewConfig
     title_config_class = PivotUserObjectPermissionTitleViewConfig

wbcore/contrib/i18n/__init__.py

@@ -0,0 +1,2 @@
+from .buttons import add_auto_translate_action_button
+from .viewsets import ModelTranslateMixin

wbcore/contrib/i18n/buttons.py

@@ -0,0 +1,33 @@
+from django.dispatch import receiver
+from django.urls import resolve
+from rest_framework.request import Request
+from rest_framework.reverse import reverse
+
+from wbcore import serializers
+from wbcore.contrib.icons.icons import WBIcon
+from wbcore.enums import RequestType
+from wbcore.metadata.configs.buttons import ActionButton
+from wbcore.metadata.configs.buttons.enums import ButtonDefaultColor
+from wbcore.metadata.configs.display.instance_display.shortcuts import create_simple_display
+from wbcore.signals.instance_buttons import add_extra_button
+
+
+class AutoTranslateSerializer(serializers.Serializer):
+    override_existing_data = serializers.BooleanField(default=False)
+
+
+@receiver(add_extra_button)
+def add_auto_translate_action_button(sender, instance, request: Request, view, pk=None, **kwargs):
+    if instance and pk and view and hasattr(view, "auto_translate"):
+        url = reverse(resolve(request.path).view_name.replace("detail", "auto-translate"), args=[pk], request=request)
+        return ActionButton(
+            method=RequestType.POST,
+            icon=WBIcon.DEAL.icon,
+            color=ButtonDefaultColor.SUCCESS,
+            endpoint=url,
+            label="Auto Translate",
+            action_label="Auto Translate",
+            description_fields="You will automatically translate all fields available for translation.",
+            serializer=AutoTranslateSerializer,
+            instance_display=create_simple_display([["override_existing_data"]]),
+        )

wbcore/contrib/i18n/serializers/fields.py

@@ -0,0 +1,20 @@
+from django.conf import settings
+
+from wbcore.serializers.fields import JSONField
+
+
+class TranslationJSONField(JSONField):
+    field_type = "translation_field"
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.read_only = False
+        self.required = False
+        self.allow_null = True
+
+    def get_representation(self, request, field_name) -> tuple[str, dict]:
+        key, representation = super().get_representation(request, field_name)
+        representation["default_language"] = settings.LANGUAGE_CODE
+        representation["languages"] = settings.MODELTRANS_AVAILABLE_LANGUAGES
+        representation["fields"] = self.parent.Meta.model._meta.get_field("i18n").fields
+        return key, representation

wbcore/contrib/i18n/serializers/mixins.py

@@ -0,0 +1,13 @@
+from wbcore import serializers
+from wbcore.contrib.i18n.serializers.fields import TranslationJSONField
+
+
+class ModelTranslateSerializerMixin(serializers.ModelSerializer):
+    _i18n = TranslationJSONField(source="i18n")
+
+    def update(self, instance, validated_data):
+        i18n_data = validated_data.pop("i18n", {}) or {}
+        if instance.i18n is None:
+            instance.i18n = {}
+        instance.i18n.update(i18n_data)
+        return super().update(instance, validated_data)

wbcore/contrib/i18n/tests/conftest.py

@@ -0,0 +1,11 @@
+import pytest
+from django.apps import apps
+from django.db.backends.base.base import BaseDatabaseWrapper
+from django.db.models.signals import pre_migrate
+from pytest_factoryboy import register
+from wbcore.contrib.authentication.factories import InternalUserFactory, UserFactory
+from wbcore.contrib.geography.tests.signals import app_pre_migration
+from wbcore.tests.conftest import *
+
+
+pre_migrate.connect(app_pre_migration, sender=apps.get_app_config("geography"))

wbcore/contrib/i18n/tests/test_viewsets.py

@@ -0,0 +1,67 @@
+from unittest.mock import MagicMock, patch
+
+import pytest
+from wbcore.contrib.i18n.viewsets import ModelTranslateMixin
+
+
+class TestModel:
+    """A simple test model for testing the ModelTranslateMixin."""
+
+    id = 1
+
+
+@pytest.fixture
+def viewset():
+    """Create a viewset instance for testing."""
+
+    class TestViewSet(ModelTranslateMixin):
+        def get_object(self):
+            return TestModel()
+
+    return TestViewSet()
+
+
+@patch("wbcore.contrib.i18n.viewsets.translate_model_as_task")
+@patch("wbcore.contrib.i18n.viewsets.ContentType.objects.get_for_model")
+def test_auto_translate_success(mock_get_for_model, mock_translate_task, viewset, rf):
+    """Test that auto_translate initiates a translation task and returns a success response."""
+    # Setup
+    mock_content_type = MagicMock()
+    mock_content_type.id = 123
+    mock_get_for_model.return_value = mock_content_type
+
+    mock_translate_task.delay = MagicMock()
+
+    request = rf.post("/auto-translate/")
+    request.data = {}
+
+    # Execute
+    response = viewset.auto_translate(request)
+
+    # Assert
+    mock_get_for_model.assert_called_once()
+    mock_translate_task.delay.assert_called_once_with(123, 1, False)
+
+    assert response.status_code == 200
+    assert response.data == {"__notification": "The translation started in the background."}
+
+
+@patch("wbcore.contrib.i18n.viewsets.translate_model_as_task")
+def test_auto_translate_no_object(mock_translate_task, viewset, rf):
+    """Test that auto_translate returns an error response when no object is found."""
+    # Setup
+    viewset.get_object = MagicMock(return_value=None)
+    mock_translate_task.delay = MagicMock()
+
+    request = rf.post("/auto-translate/")
+    request.data = {}
+    # Execute
+    response = viewset.auto_translate(request)
+
+    # Assert
+    mock_translate_task.delay.assert_not_called()
+
+    # The viewset code uses status.HTTP_400_BAD_REQUEST
+    assert response.status_code == 400
+    assert "non_field_errors" in response.data
+    assert response.data["non_field_errors"] == ["The URL was malformatted."]

wbcore/contrib/i18n/translation.py

@@ -0,0 +1,140 @@
+from copy import deepcopy
+from typing import TYPE_CHECKING, Any, Iterator
+
+from celery import shared_task
+from django.conf import settings
+from django.contrib.contenttypes.models import ContentType
+from django.db.models import Field
+from langchain_core.messages import BaseMessage, HumanMessage, SystemMessage
+from modeltrans.fields import TranslatedVirtualField
+
+from wbcore.contrib.ai.llm.utils import run_llm
+
+if TYPE_CHECKING:
+    from django.db.models import Model
+
+
+def get_base_prompt(from_language: str, to_language: str) -> list["BaseMessage"]:
+    """
+    Creates a base prompt for translation with system instructions.
+
+    Args:
+        from_language: ISO2 code of the source language
+        to_language: ISO2 code of the target language
+
+    Returns:
+        list[BaseMessage]: A list containing the system message for translation
+    """
+    return [
+        SystemMessage(f"""
+    You are a highly skilled translator with expertise in accurately translating content between languages.
+    Your task is to translate the provided content from ISO2={from_language} to ISO2={to_language}.
+
+    Please adhere to the following guidelines:
+    - Maintain the original meaning, tone, and style of the content.
+    - Preserve any formatting, including line breaks and special characters.
+    - Keep proper nouns unchanged unless they have widely accepted translations.
+    - Do not add, remove, or alter any information in the content.
+    - Provide only the translated text without any additional explanations or notes.
+    """)
+    ]
+
+
+def _is_translated_field(field: Field | TranslatedVirtualField) -> bool:
+    """
+    Determines if a field is a translated field that needs processing.
+
+    Args:
+        field: The field to check
+
+    Returns:
+        bool: True if the field is a TranslatedVirtualField with a non-default language
+    """
+    return (
+        isinstance(field, TranslatedVirtualField)
+        and field.language is not None
+        and field.language != settings.LANGUAGE_CODE
+    )
+
+
+def _get_translation_fields(
+    fields: list[Field | TranslatedVirtualField | Any],
+) -> Iterator[TranslatedVirtualField]:
+    """
+    Filters a list of fields to only include translated fields.
+
+    Args:
+        fields: List of model fields to filter
+
+    Returns:
+        Iterator[TranslatedVirtualField]: Iterator yielding only the translated virtual fields
+    """
+    yield from filter(_is_translated_field, fields)  # type: ignore
+
+
+def translate_string(content: str, language: str) -> str:
+    prompt = get_base_prompt(settings.LANGUAGE_CODE, language)
+    prompt.append(HumanMessage(content))
+    return str(run_llm(prompt)[0])
+
+
+def translate_dict(content: dict, language: str) -> dict:
+    new_content = deepcopy(content)
+
+    for key, value in content.items():
+        if isinstance(value, dict):
+            new_content[key] = translate_dict(value, language)
+        elif isinstance(value, str):
+            new_content[key] = translate_string(value, language)
+
+    return new_content
+
+
+def translate_model(model: "Model", override: bool = False):
+    """
+    Translates all translatable fields in a model instance.
+
+    Processes all fields in the model that are marked for translation,
+    generates translations using an LLM, and saves the model with
+    the translated content.
+
+    Args:
+        model: The Django model instance to translate
+        override: If True, do not set the translated value if there is already content
+    """
+    for field in _get_translation_fields(model._meta.get_fields()):
+        original_content = getattr(model, field.original_field.name)
+        if field.language is None or (field is not None and not override):
+            continue
+
+        if custom_translation := getattr(model, f"_translate_{field.original_field.name}", None):
+            setattr(model, field.name, custom_translation(field.language))
+        else:
+            match original_content:
+                case str():
+                    setattr(
+                        model,
+                        field.name,
+                        translate_string(original_content, field.language),
+                    )
+                case dict():
+                    setattr(model, field.name, translate_dict(original_content, field.language))
+
+    model.save()
+
+
+@shared_task
+def translate_model_as_task(content_type_id: int, pk: Any, override: bool = False):
+    """
+    Celery task for asynchronous translation of a model instance.
+
+    Retrieves the model class from the content type ID, loads the
+    specific instance by primary key, and translates its fields.
+
+    Args:
+        content_type_id: ID of the ContentType for the model to translate
+        pk: Primary key of the specific model instance to translate
+    """
+    if mc := ContentType.objects.get(id=content_type_id).model_class():
+        model = mc.objects.get(pk=pk)
+        translate_model(model, override)

wbcore/contrib/i18n/viewsets.py

@@ -0,0 +1,36 @@
+from django.contrib.contenttypes.models import ContentType
+from rest_framework import status
+from rest_framework.decorators import action
+from rest_framework.request import Request
+from rest_framework.response import Response
+
+from wbcore import viewsets
+from wbcore.contrib.i18n.translation import translate_model_as_task
+
+
+class ModelTranslateMixin(viewsets.ModelViewSet):
+    @action(methods=["POST"], detail=True)
+    def auto_translate(self, request: Request, *args, **kwargs):
+        """Initiates an asynchronous translation task for the specified model instance.
+
+        This method retrieves the model instance based on the provided URL parameters,
+        starts a background translation task, and returns a notification response.
+
+        Args:
+            request (Request): The HTTP request object containing the data for the translation.
+            *args: Additional positional arguments.
+            **kwargs: Additional keyword arguments.
+            override_existing_data (bool): If True, existing content will not be overridden.
+
+        Returns:
+            Response: A response indicating the status of the translation initiation.
+        """
+        override_existing_data = request.data.get("override_existing_data", "false") == "true"
+
+        obj = self.get_object()
+        if obj:
+            ct = ContentType.objects.get_for_model(obj)  # type: ignore
+            translate_model_as_task.delay(ct.id, obj.id, override_existing_data)
+            return Response({"__notification": "The translation started in the background."})
+
+        return Response({"non_field_errors": ["The URL was malformatted."]}, status=status.HTTP_400_BAD_REQUEST)

wbcore/contrib/icons/backends/default.py

@@ -15,6 +15,7 @@ class IconBackend(AbstractBackend):
     APPROVE = "wb-icon-thumb-up"
     BANK = "wb-icon-bank"
     BIRTHDAY = "wb-icon-cake-1"
+    BROADCAST = "wb-icon-send"
     BOOKMARK = "wb-icon-bookmark"
     CALENDAR = "wb-icon-calendar-1"
     CHART_AREA = "wb-icon-chart-area"

wbcore/contrib/icons/backends/material.py

@@ -19,6 +19,7 @@ class IconBackend(AbstractBackend):
     BANK = "account_balance"
     BIRTHDAY = "cake"
     BOOKMARK = "bookmark_border"
+    BROADCAST = "arrow_split"
     CALENDAR = "calendar_month"
     CHART_AREA = "stacked_line_chart"
     CHART_BARS_HORIZONTAL = "align_horizontal_center"

wbcore/contrib/icons/icons.py

@@ -15,20 +15,16 @@ FALLBACK_ICON_VALUE = "FALLBACK_ICON"
 class WBIconMeta(ChoicesMeta):
     """A metaclass for creating a enum choices."""
 
-    def __new__(metacls, classname, bases, classdict, **kwds):
+    def __new__(cls, classname, bases, classdict, **kwds):
         dict.__setitem__(
             classdict, FALLBACK_ICON_VALUE, (FALLBACK_ICON_VALUE, "Fallback Icon")
         )  # add a default member that represent the fallback in case it's not yet implement in the imported backend
-        cls = super().__new__(metacls, classname, bases, classdict, **kwds)
+        cls = super().__new__(cls, classname, bases, classdict, **kwds)
         with suppress(ModuleNotFoundError):
-            setattr(
-                cls,
-                "icon_backend",
-                import_from_dotted_path(getattr(settings, "WBCORE_ICON_BACKEND", DEFAULT_ICON_BACKEND)),
-            )
+            cls.icon_backend = import_from_dotted_path(getattr(settings, "WBCORE_ICON_BACKEND", DEFAULT_ICON_BACKEND))
             icon_backend = cls.icon_backend
             # For each enumeration values, attached an "icon" property to its members
-            for member, value in zip(cls.__members__.values(), cls.values):
+            for member, value in zip(cls.__members__.values(), cls.values, strict=False):
                 member._icon_ = getattr(icon_backend, value, icon_backend.fallback_icon)
         return enum.unique(cls)
 
@@ -54,6 +50,7 @@ class WBIcon(TextChoices, metaclass=WBIconMeta):
     BANK = "BANK", "Bank"
     BIRTHDAY = "BIRTHDAY", "Birthday"
     BOOKMARK = "BOOKMARK", "Bookmark"
+    BROADCAST = "BROADCAST", "Broadcast"
     CALENDAR = "CALENDAR", "Calendar"
     CHART_AREA = "CHART_AREA", "Chart area"
     CHART_BARS_HORIZONTAL = "CHART_BARS_HORIZONTAL", "Chart bars horizonal"

wbcore/contrib/io/admin.py

@@ -133,6 +133,7 @@ class ImportSourceModelAdmin(admin.ModelAdmin):
                     ("parser_handler", "source", "origin"),
                     ("file", "save_data", "progress_index"),
                     ("creator",),
+                    ("resource_kwargs",),
                     ("data",),
                     ("log",),
                     ("errors_log",),