ewilliams28/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2026-08-13 01:02:25 -06:00
Code Issues Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2026-07-27 03:54:05 -04:00
parent b7394c63fa
commit 2d928df304
1 changed files with 30 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/open_webui/utils/automations.py
+30 -14
View File
@@ -25,6 +25,7 @@ from typing import Optional
from uuid import uuid4 from uuid import uuid4
from zoneinfo import ZoneInfo from zoneinfo import ZoneInfo
from dateutil import parser as date_parser
from dateutil.rrule import rrulestr from dateutil.rrule import rrulestr
from fastapi import Request from fastapi import Request
from fastapi.security import HTTPAuthorizationCredentials from fastapi.security import HTTPAuthorizationCredentials
@@ -76,7 +77,14 @@ def _parse_rule(s: str, now: Optional[datetime] = None):
SECONDLY/MINUTELY/HOURLY rules use a fixed epoch DTSTART (2000-01-01 00:00) SECONDLY/MINUTELY/HOURLY rules use a fixed epoch DTSTART (2000-01-01 00:00)
so intervals snap to clock boundaries (e.g. every 5min = :00, :05, :10). so intervals snap to clock boundaries (e.g. every 5min = :00, :05, :10).
""" """
rrule_line = next((line for line in s.splitlines() if line.upper().startswith('RRULE:')), s) lines = s.splitlines()
rule_count = sum(1 for line in lines if line.upper().startswith('RRULE:'))
if 'EXRULE' in s.upper():
raise ValueError('EXRULE is not supported in recurrence rules')
if rule_count > 1:
raise ValueError('only one RRULE is supported per recurrence rule')
rrule_line = next((line for line in lines if line.upper().startswith('RRULE:')), s)
raw = rrule_line.split(':', 1)[1] if rrule_line.upper().startswith('RRULE:') else rrule_line raw = rrule_line.split(':', 1)[1] if rrule_line.upper().startswith('RRULE:') else rrule_line
parts = {k.upper(): v for k, v in (p.split('=', 1) for p in raw.split(';') if '=' in p)} parts = {k.upper(): v for k, v in (p.split('=', 1) for p in raw.split(';') if '=' in p)}
freq = parts.get('FREQ', '') freq = parts.get('FREQ', '')
@@ -84,19 +92,27 @@ def _parse_rule(s: str, now: Optional[datetime] = None):
if freq in ('SECONDLY', 'MINUTELY', 'HOURLY'): if freq in ('SECONDLY', 'MINUTELY', 'HOURLY'):
epoch = datetime(2000, 1, 1, 0, 0, 0) epoch = datetime(2000, 1, 1, 0, 0, 0)
anchor = now or datetime.now() anchor = now or datetime.now()
rule = '\n'.join(line for line in s.splitlines() if not line.upper().startswith('DTSTART')) or s rule = '\n'.join(line for line in lines if not line.upper().startswith('DTSTART')) or s
try: dtstart = next((line.rsplit(':', 1)[-1] for line in lines if line.upper().startswith('DTSTART')), None)
interval = int(parts.get('INTERVAL', '1')) interval = int(parts.get('INTERVAL', '1'))
if interval > 0: if interval < 1:
if freq == 'SECONDLY': raise ValueError('RRULE INTERVAL must be a positive integer')
step = timedelta(seconds=interval) if freq == 'SECONDLY':
elif freq == 'MINUTELY': step = timedelta(seconds=interval)
step = timedelta(minutes=interval) elif freq == 'MINUTELY':
else: step = timedelta(minutes=interval)
step = timedelta(hours=interval) else:
anchor = epoch + ((anchor - epoch) // step) * step step = timedelta(hours=interval)
except (TypeError, ValueError): if dtstart:
pass start = date_parser.parse(dtstart, ignoretz=True)
emitted = ((anchor - start) // step) if anchor > start else 0
if 'BYMINUTE' in parts:
emitted *= len(parts['BYMINUTE'].split(','))
if 'BYSECOND' in parts:
emitted *= len(parts['BYSECOND'].split(','))
if emitted <= 100_000:
return rrulestr(s, ignoretz=True)
anchor = epoch + ((anchor - epoch) // step) * step
return rrulestr(rule, dtstart=anchor, ignoretz=True) return rrulestr(rule, dtstart=anchor, ignoretz=True)
return rrulestr(s, ignoretz=True) return rrulestr(s, ignoretz=True)