refac

backend/open_webui/models/calendar.py

@@ -4,10 +4,12 @@ from typing import Optional
 from uuid import uuid4
 
 from open_webui.internal.db import Base, get_async_db_context
+from open_webui.constants import ERROR_MESSAGES
 from open_webui.models.access_grants import AccessGrantModel, AccessGrants
 from open_webui.models.groups import Groups
 from open_webui.models.users import User, UserModel, UserResponse
-from pydantic import BaseModel, ConfigDict, Field
+from open_webui.utils.automations import rrule_interval_seconds
+from pydantic import BaseModel, ConfigDict, Field, field_validator
 from sqlalchemy import (
     JSON,
     BigInteger,
@@ -26,6 +28,7 @@ from sqlalchemy import (
 from sqlalchemy.ext.asyncio import AsyncSession
 
 log = logging.getLogger(__name__)
+MIN_CALENDAR_RRULE_INTERVAL_SECONDS = 24 * 60 * 60
 
 
 ####################
@@ -191,6 +194,20 @@ class CalendarEventForm(BaseModel):
     meta: Optional[dict] = None
     attendees: Optional[list[dict]] = None
 
+    @field_validator('rrule')
+    @classmethod
+    def reject_sub_daily_rrule(cls, value: Optional[str]) -> Optional[str]:
+        if value:
+            try:
+                interval = rrule_interval_seconds(value)
+            except ValueError:
+                raise
+            except Exception as e:
+                raise ValueError(ERROR_MESSAGES.AUTOMATION_INVALID_RRULE(e))
+            if interval is not None and interval < MIN_CALENDAR_RRULE_INTERVAL_SECONDS:
+                raise ValueError(ERROR_MESSAGES.CALENDAR_RRULE_TOO_FREQUENT)
+        return value
+
 
 class CalendarEventUpdateForm(BaseModel):
     calendar_id: Optional[str] = None
@@ -207,6 +224,20 @@ class CalendarEventUpdateForm(BaseModel):
     is_cancelled: Optional[bool] = None
     attendees: Optional[list[dict]] = None
 
+    @field_validator('rrule')
+    @classmethod
+    def reject_sub_daily_rrule(cls, value: Optional[str]) -> Optional[str]:
+        if value:
+            try:
+                interval = rrule_interval_seconds(value)
+            except ValueError:
+                raise
+            except Exception as e:
+                raise ValueError(ERROR_MESSAGES.AUTOMATION_INVALID_RRULE(e))
+            if interval is not None and interval < MIN_CALENDAR_RRULE_INTERVAL_SECONDS:
+                raise ValueError(ERROR_MESSAGES.CALENDAR_RRULE_TOO_FREQUENT)
+        return value
+
 
 class RSVPForm(BaseModel):
     status: str  # 'accepted' | 'declined' | 'tentative' | 'pending'