wbcore 1.59.3__tar.gz → 1.59.5__tar.gz

{wbcore-1.59.3 → wbcore-1.59.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wbcore
-Version: 1.59.3
+Version: 1.59.5
 Author-email: Christopher Wittlinger <c.wittlinger@stainly.com>
 Requires-Dist: boto3==1.35.*
 Requires-Dist: celery[redis]==5.*

{wbcore-1.59.3 → wbcore-1.59.5}/wbcore/contrib/agenda/models/calendar_item.py

@@ -15,6 +15,7 @@ from wbcore.contrib.icons.models import IconField
 from wbcore.models import WBModel
 from wbcore.utils.itertools import get_inheriting_subclasses
 from wbcore.utils.models import DeleteToDisableMixin
+from wbcore.workers import Queue
 
 
 class CalendarItem(DeleteToDisableMixin, WBModel):
@@ -219,7 +220,7 @@ def m2m_entities_changed(sender, instance, action, pk_set, **kwargs):
         transaction.on_commit(set_entity_list.s(instance.id).delay)
 
 
-@shared_task
+@shared_task(queue=Queue.HIGH_PRIORITY.value)
 def set_entity_list(calendar_item_id: int):
     frontend_list = []
     with suppress(

{wbcore-1.59.3 → wbcore-1.59.5}/wbcore/contrib/ai/llm/config.py

@@ -9,6 +9,8 @@ from langchain_core.messages import BaseMessage
 from langchain_openai import ChatOpenAI
 from pydantic import BaseModel
 
+from wbcore.workers import Queue
+
 from ..exceptions import APIStatusErrors, BadRequestErrors
 from .utils import run_llm
 
@@ -16,7 +18,7 @@ logger = logging.getLogger("llm")
 
 
 @shared_task(
-    queue="llm",
+    queue=Queue.BACKGROUND.value,
     autoretry_for=tuple(APIStatusErrors),
     retry_backoff=10,
     max_retries=5,  # retry 5 times maximum

{wbcore-1.59.3 → wbcore-1.59.5}/wbcore/contrib/ai/llm/mixins.py

@@ -1,8 +1,10 @@
 from celery import chain, shared_task
 from django.db.models import Model
 
+from wbcore.workers import Queue
 
-@shared_task(queue="llm")
+
+@shared_task(queue=Queue.BACKGROUND.value)
 def save_instance_as_task(instance):
     instance.save(_with_llm=False)
 

{wbcore-1.59.3 → wbcore-1.59.5}/wbcore/contrib/authentication/models/users_activities.py

@@ -9,6 +9,8 @@ from django.dispatch import receiver
 from django.utils import timezone
 from django.utils.translation import gettext_lazy as _
 
+from wbcore.workers import Queue
+
 from .users import User
 
 ACCESS_TOKEN_LIFETIME = settings.SIMPLE_JWT["ACCESS_TOKEN_LIFETIME"].seconds
@@ -97,7 +99,7 @@ def log_user_logged_in_success(sender, user, request, **kwargs):
         )
 
 
-@shared_task()
+@shared_task(queue=Queue.HIGH_PRIORITY.value)
 def refresh_user_activity(user_id, refresh_time, jti):
     """
     Signal triggerd whenever a refresh token is submit from the frontend. Update latest_refresh to the associated UserActivity.

{wbcore-1.59.3 → wbcore-1.59.5}/wbcore/contrib/authentication/tasks.py

@@ -5,12 +5,13 @@ from celery import shared_task
 from django.utils import timezone
 from dynamic_preferences.registries import global_preferences_registry
 
+from ...workers import Queue
 from .models import User
 
 logger = logging.getLogger(__name__)
 
 
-@shared_task
+@shared_task(queue=Queue.BACKGROUND.value)
 def delete_unregistered_user_account(prune_user_account_before_datetime: datetime | None = None):
     if not prune_user_account_before_datetime:
         global_preferences = global_preferences_registry.manager()

{wbcore-1.59.3 → wbcore-1.59.5}/wbcore/contrib/documents/models/documents.py

@@ -25,6 +25,7 @@ from wbcore.contrib.authentication.models import User
 from wbcore.contrib.guardian.models.mixins import PermissionObjectModelMixin
 from wbcore.models import WBModel
 from wbcore.utils.html import convert_html2text
+from wbcore.workers import Queue
 
 from .document_model_relationships import DocumentModelRelationship
 from .shareable_links import ShareableLink
@@ -238,7 +239,7 @@ class Document(PermissionObjectModelMixin, WBModel):
         verbose_name_plural = _("Documents")
 
 
-@shared_task
+@shared_task(queue=Queue.HIGH_PRIORITY.value)
 def send_email_as_task(
     document_id,
     to_emails: List | str,

{wbcore-1.59.3 → wbcore-1.59.5}/wbcore/contrib/example_app/models.py

@@ -28,6 +28,7 @@ from wbcore.enums import RequestType
 from wbcore.metadata.configs.buttons import ActionButton
 from wbcore.models.orderable import OrderableModel
 from wbcore.utils.models import ActiveObjectManager, ComplexToStringMixin
+from wbcore.workers import Queue
 
 
 def upload_to_profile_images(instance, filename):
@@ -592,7 +593,7 @@ class Match(ComplexToStringMixin, CalendarItem):
         permissions = [("change_match_status", "Change Match Status")]
 
 
-@shared_task
+@shared_task(queue=Queue.DEFAULT.value)
 def start_match(match_id: int):
     """Sets the match status from scheduled to ongoing.
 

{wbcore-1.59.3 → wbcore-1.59.5}/wbcore/contrib/guardian/models/mixins.py

@@ -11,6 +11,7 @@ from wbcore.contrib.authentication.models.users import User
 from wbcore.contrib.guardian.utils import reload_permissions
 from wbcore.permissions.mixins import PermissionMixin
 from wbcore.utils.itertools import get_inheriting_subclasses
+from wbcore.workers import Queue
 
 
 class PermissionObjectModelMixin(PermissionMixin):
@@ -113,7 +114,7 @@ def post_save_user(sender, instance, created, **kwargs):
         transaction.on_commit(lambda: assign_object_permissions_for_user_as_task.delay(instance.id))  # type: ignore
 
 
-@shared_task
+@shared_task(queue=Queue.DEFAULT.value)
 def assign_user_permissions_for_object_as_task(
     content_type_id: int, instance_id: int, prune_existing: bool | None = True
 ):
@@ -127,7 +128,7 @@ def assign_user_permissions_for_object_as_task(
             instance.reload_permissions(prune_existing=prune_existing)  # type: ignore
 
 
-@shared_task
+@shared_task(queue=Queue.DEFAULT.value)
 def assign_object_permissions_for_user_as_task(user_id: int):
     """
     Utility function to create object permission from user decoupling from the main thread

{wbcore-1.59.3 → wbcore-1.59.5}/wbcore/contrib/guardian/tasks.py

@@ -1,9 +1,10 @@
 from celery import shared_task
 from wbcore.contrib.guardian.models.mixins import PermissionObjectModelMixin
 from wbcore.utils.itertools import get_inheriting_subclasses
+from wbcore.workers import Queue
 
 
-@shared_task
+@shared_task(queue=Queue.EXTENDED_BACKGROUND.value)
 def reload_permissions_as_task(prune_existing: bool | None = True, force_pruning: bool | None = False):
     for subclass in get_inheriting_subclasses(PermissionObjectModelMixin):
         for instance in subclass.objects.iterator():

{wbcore-1.59.3 → wbcore-1.59.5}/wbcore/contrib/i18n/translation.py

@@ -9,6 +9,7 @@ from langchain_core.messages import BaseMessage, HumanMessage, SystemMessage
 from modeltrans.fields import TranslatedVirtualField
 
 from wbcore.contrib.ai.llm.utils import run_llm
+from wbcore.workers import Queue
 
 if TYPE_CHECKING:
     from django.db.models import Model
@@ -123,7 +124,7 @@ def translate_model(model: "Model", override: bool = False):
     model.save()
 
 
-@shared_task
+@shared_task(queue=Queue.DEFAULT.value)
 def translate_model_as_task(content_type_id: int, pk: Any, override: bool = False):
     """
     Celery task for asynchronous translation of a model instance.

{wbcore-1.59.3 → wbcore-1.59.5}/wbcore/contrib/io/models.py

@@ -40,6 +40,7 @@ from wbcore.contrib.notifications.utils import create_notification_type
 from wbcore.utils.importlib import import_from_dotted_path
 from wbcore.utils.models import ComplexToStringMixin
 
+from ...workers import Queue
 from .enums import ExportFormat, get_django_import_export_format
 from .signals import post_import
 
@@ -615,7 +616,7 @@ def pre_delete_source(sender, instance, **kwargs):
     ImportSource.objects.filter(source=instance).update(source=None)
 
 
-@shared_task(queue="importexport")
+@shared_task(queue=Queue.BACKGROUND.value)
 def trigger_workflow_as_task(source_id: int, execution_time: datetime | None = None, **kwargs):
     """
     Call the `import_source` as a celery task
@@ -624,7 +625,7 @@ def trigger_workflow_as_task(source_id: int, execution_time: datetime | None = N
     source.trigger_workflow(execution_time=execution_time, **kwargs)
 
 
-@shared_task(queue="importexport")
+@shared_task(queue=Queue.BACKGROUND.value)
 def generate_import_sources_as_task(source_id: int, execution_time: datetime, **kwargs) -> list[int]:
     """
     Call the `import_source` as a celery task
@@ -636,7 +637,7 @@ def generate_import_sources_as_task(source_id: int, execution_time: datetime, **
     return import_source_ids
 
 
-@shared_task(queue="importexport")
+@shared_task(queue=Queue.BACKGROUND.value)
 def process_import_sources_as_task(import_source_ids: list[int]):
     """
     Call the `import_source` as a celery task
@@ -936,7 +937,7 @@ class ImportSource(ImportExportSource):
         return "wbcore:io:importsourcerepresentation-list"
 
 
-@shared_task(queue="importexport")
+@shared_task(queue=Queue.BACKGROUND.value)
 def import_data_as_task(import_source_id: int, **kwargs):
     """
     Call `import_data` as a celery task
@@ -946,7 +947,7 @@ def import_data_as_task(import_source_id: int, **kwargs):
     import_source.import_data(**kwargs)
 
 
-@shared_task(queue="importexport")
+@shared_task(queue=Queue.DEFAULT.value)
 def export_data_as_task(export_source_id: int, **kwargs):
     """
     Call `import_data` as a celery task

{wbcore-1.59.3 → wbcore-1.59.5}/wbcore/contrib/io/tasks.py

@@ -5,8 +5,10 @@ from django.db import connection
 from django.utils import timezone
 from dynamic_preferences.registries import global_preferences_registry
 
+from wbcore.workers import Queue
 
-@shared_task()
+
+@shared_task(queue=Queue.BACKGROUND.value)
 def clean_up_import_source(today: date | None = None):
     if not today:
         today = timezone.now()

{wbcore-1.59.3 → wbcore-1.59.5}/wbcore/contrib/notifications/dispatch.py

@@ -12,6 +12,7 @@ from rest_framework.reverse import reverse
 
 from wbcore.shares.signals import handle_widget_sharing
 
+from ...workers import Queue
 from .models import Notification, NotificationType, NotificationTypeSetting
 from .tasks import send_notification_task
 
@@ -64,7 +65,7 @@ def send_notification(
             )
 
 
-@shared_task()
+@shared_task(queue=Queue.HIGH_PRIORITY.value)
 def send_notification_as_task(code, title, body, user_id, **kwargs):
     if not isinstance(user_id, list):
         user_id = [user_id]

{wbcore-1.59.3 → wbcore-1.59.5}/wbcore/contrib/notifications/tasks.py

@@ -9,6 +9,7 @@ from wbcore.contrib.notifications.models.notification_types import (
     NotificationTypeSetting,
 )
 from wbcore.contrib.notifications.models.notifications import Notification
+from wbcore.workers import Queue
 
 
 def send_notification_email(notification: Notification):
@@ -35,7 +36,7 @@ def send_notification_email(notification: Notification):
     msg.send()
 
 
-@shared_task
+@shared_task(queue=Queue.HIGH_PRIORITY.value)
 def send_notification_task(notification_pk: int):
     """A celery task to send out a notification via email, web or mobile
 

{wbcore-1.59.3 → wbcore-1.59.5}/wbcore/contrib/pandas/fields.py

@@ -43,6 +43,14 @@ class CharField(_BaseField):
         return super().to_representation(value).astype("string", errors="ignore")
 
 
+@dataclass
+class DateTimeField(_BaseField):
+    type: str = "datetime"
+
+    def to_representation(self, value: pd.Series) -> pd.Series:
+        return pd.to_datetime(super().to_representation(value)).dt.strftime("%Y-%m-%d %H:%M:%S")
+
+
 @dataclass
 class DateField(_BaseField):
     type: str = "date"

{wbcore-1.59.3 → wbcore-1.59.5}/wbcore/contrib/workflow/models/step.py

@@ -22,6 +22,7 @@ from wbcore.contrib.workflow.utils import get_model_serializer_class_for_instanc
 from wbcore.models import WBModel
 from wbcore.utils.html import convert_html2text
 from wbcore.utils.string_loader import StringSourceLoader
+from wbcore.workers import Queue
 
 
 class Step(WBModel):
@@ -661,7 +662,7 @@ class FinishStep(Step):
         verbose_name_plural = _("Finish Steps")
 
 
-@shared_task()
+@shared_task(queue=Queue.DEFAULT.value)
 def activate_step(step_id: int, process_id: UUID):
     step = Step.objects.get(pk=step_id).get_casted_step()
     process = Process.objects.get(pk=process_id)
@@ -739,7 +740,7 @@ def _check_previous_steps_failed(step: Step, process: Process, check_all: bool =
                     _check_previous_steps_failed(step, process)
 
 
-@shared_task()
+@shared_task(queue=Queue.DEFAULT.value)
 def process_can_finish(process_id: UUID):
     """Checks if all critical steps for a process can still be reached. Fails the process if this is not the case.
 

{wbcore-1.59.3 → wbcore-1.59.5}/wbcore/metadata/configs/display/list_display.py

@@ -54,7 +54,8 @@ class Field:
     lock_position: Literal["left"] | Literal["right"] | None = None
     movable: bool = True
     resizable: bool = True
-    auto_size: bool = True
+    suppress_auto_size: bool = False
+    auto_size: bool = False
     menu: bool = True
     size_to_fit: bool = True
 
@@ -110,7 +111,10 @@ class Field:
         if self.lock_position:
             repr["lock_position"] = self.lock_position
 
-        if not self.auto_size:
+        if self.suppress_auto_size:
+            repr["suppress_auto_size"] = self.suppress_auto_size
+
+        if self.auto_size:
             repr["auto_size"] = self.auto_size
 
         if not self.menu:
@@ -214,6 +218,7 @@ class ListDisplay:
     condensed: bool = False
 
     auto_height: bool = False
+    auto_size_columns: bool = False
 
     tree: bool = False
     tree_group_parent_pointer: str | None = (
@@ -242,6 +247,7 @@ class ListDisplay:
         yield "ordering_field", self.ordering_field
 
         yield "auto_height", self.auto_height
+        yield "auto_size_columns", self.auto_size_columns
 
         if self.condensed:
             yield "condensed", self.condensed

{wbcore-1.59.3 → wbcore-1.59.5}/wbcore/models/base.py

@@ -12,6 +12,7 @@ from igraph import Graph
 
 from wbcore.signals.models import get_dependant_dynamic_fields_instances
 
+from ..workers import Queue
 from .fields import DynamicDecimalField, DynamicFloatField
 
 
@@ -136,7 +137,7 @@ class DynamicModel(models.Model, metaclass=DynamicMetaClass):
         abstract = True
 
 
-@shared_task
+@shared_task(queue=Queue.DEFAULT.value)
 def set_dynamic_field_as_task(content_type_id, object_id, **kwargs):
     content_type = ContentType.objects.get(id=content_type_id)
     instance = content_type.model_class().objects.get(id=object_id)
@@ -173,7 +174,7 @@ class WBModel(models.Model):
         return errors
 
 
-@shared_task
+@shared_task(queue=Queue.DEFAULT.value)
 def merge_as_task(content_type_id: int, main_object_id: int, merged_object_id: int):
     content_type = ContentType.objects.get(id=content_type_id)
     with suppress(content_type.model_class().DoesNotExist):

{wbcore-1.59.3 → wbcore-1.59.5}/wbcore/pandas/__init__.py

@@ -12,6 +12,7 @@ from wbcore.contrib.pandas.fields import (  # noqa
     PKField,
     CharField,
     DateField,
+    DateTimeField,
     DateRangeField,
     BooleanField,
     TextField,

{wbcore-1.59.3 → wbcore-1.59.5}/wbcore/pandas/fields.py

@@ -10,6 +10,7 @@ from wbcore.contrib.pandas.fields import (  # noqa
     PKField,
     CharField,
     DateField,
+    DateTimeField,
     DateRangeField,
     BooleanField,
     TextField,

{wbcore-1.59.3 → wbcore-1.59.5}/wbcore/tasks.py

@@ -11,9 +11,10 @@ from wbcore.cache.registry import periodic_cache_registry
 from wbcore.models import DynamicModel
 from wbcore.utils.itertools import get_inheriting_subclasses
 from wbcore.utils.models import ComplexToStringMixin, DeleteToDisableMixin
+from wbcore.workers import Queue
 
 
-@shared_task
+@shared_task(queue=Queue.EXTENDED_BACKGROUND.value)
 def recompute_latest_modified_dynamic_fields(from_datetime=None, to_datetime=None, timedelta_days=1):
     """
     Daily (expected frequency) recompute of all dynamic model objects that falls within the specified time frame (default to the last 24h).
@@ -39,7 +40,7 @@ def recompute_latest_modified_dynamic_fields(from_datetime=None, to_datetime=Non
             DynamicModel.update_dynamic_fields_of_dependant_objects(already_updated_object)
 
 
-@shared_task
+@shared_task(queue=Queue.EXTENDED_BACKGROUND.value)
 def recompute_computed_str(debug: bool = False):
     """
     When this task is executed, it will loop over all objects that inherit from ComplexToStringMixin and compare their current computed_str value with the expected one.
@@ -66,7 +67,7 @@ def recompute_computed_str(debug: bool = False):
                 subclass.objects.bulk_update(objs, ["computed_str"])
 
 
-@shared_task
+@shared_task(queue=Queue.EXTENDED_BACKGROUND.value)
 def clean_deleted_objects():
     """
     Periodically arise deleted objects that have passed the retention period from the database.
@@ -80,7 +81,7 @@ def clean_deleted_objects():
                 ).delete()
 
 
-@shared_task
+@shared_task(queue=Queue.EXTENDED_BACKGROUND.value)
 def refetch_pandas_api_view():
     for cache_api_view in periodic_cache_registry.classes:
         cache_api_view.fetch_cache()

wbcore-1.59.5/wbcore/workers.py

@@ -0,0 +1,8 @@
+import enum
+
+
+class Queue(enum.Enum):
+    HIGH_PRIORITY = "high_priority"  # High priority (for urgent, short tasks)
+    DEFAULT = "default"  # for normal tasks
+    BACKGROUND = "background"  # Less urgent, medium - length tasks (few minutes)
+    EXTENDED_BACKGROUND = "extended_background"  # Very long-lasting background tasks (tasks lasting hours)

wbcore-1.59.3/wbcore/utils/task.py

@@ -1,6 +0,0 @@
-from celery import shared_task
-
-
-@shared_task
-def save_as_task(obj):
-    obj.save()