quick fix
This commit is contained in:
@@ -3,7 +3,7 @@
|
||||
import json
|
||||
import logging
|
||||
import re
|
||||
from datetime import datetime, timedelta
|
||||
from datetime import datetime, timedelta, timezone
|
||||
|
||||
from sqlalchemy import select
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
@@ -71,7 +71,7 @@ Examples:
|
||||
if result and result.get("has_event"):
|
||||
days_until = result.get("days_until", 1) or 1
|
||||
# Schedule follow-up for 1 day after the event
|
||||
trigger_at = datetime.utcnow() + timedelta(days=days_until + 1)
|
||||
trigger_at = datetime.now(timezone.utc) + timedelta(days=days_until + 1)
|
||||
|
||||
event = ScheduledEvent(
|
||||
user_id=user.id,
|
||||
@@ -159,7 +159,7 @@ Examples:
|
||||
break
|
||||
|
||||
# Create the event
|
||||
trigger_at = datetime.utcnow() + timedelta(days=days_until + 1)
|
||||
trigger_at = datetime.now(timezone.utc) + timedelta(days=days_until + 1)
|
||||
|
||||
event = ScheduledEvent(
|
||||
user_id=user.id,
|
||||
@@ -300,7 +300,7 @@ Examples:
|
||||
|
||||
def _next_birthday(self, birthday: datetime) -> datetime:
|
||||
"""Calculate the next occurrence of a birthday."""
|
||||
today = datetime.utcnow().date()
|
||||
today = datetime.now(timezone.utc).date()
|
||||
this_year = birthday.replace(year=today.year)
|
||||
|
||||
if this_year.date() < today:
|
||||
@@ -322,7 +322,7 @@ Examples:
|
||||
|
||||
async def get_pending_events(self, before: datetime | None = None) -> list[ScheduledEvent]:
|
||||
"""Get events that should be triggered."""
|
||||
cutoff = before or datetime.utcnow()
|
||||
cutoff = before or datetime.now(timezone.utc)
|
||||
stmt = (
|
||||
select(ScheduledEvent)
|
||||
.where(
|
||||
@@ -390,7 +390,7 @@ Examples:
|
||||
async def mark_event_triggered(self, event: ScheduledEvent) -> None:
|
||||
"""Mark an event as triggered and handle recurrence."""
|
||||
event.status = "triggered"
|
||||
event.triggered_at = datetime.utcnow()
|
||||
event.triggered_at = datetime.now(timezone.utc)
|
||||
|
||||
# Handle recurring events
|
||||
if event.is_recurring and event.recurrence_rule:
|
||||
|
||||
Reference in New Issue
Block a user