feat: replace datetime.now(UTC).replace(tzinfo=None) with naive_utc_now() helper

- Add naive_utc_now import to 33+ files across the codebase
- Replace datetime patterns in controllers, services, models, and core modules
- Improve code consistency and maintainability
- Enable easier testing through centralized datetime helper

Fixes #22598
pull/22654/head
aryan 10 months ago
parent 31c8fafe4e
commit 7c04cba63c

@ -7,6 +7,7 @@ from sqlalchemy import select
from sqlalchemy.orm import Session
from core.app.entities.app_invoke_entities import AdvancedChatAppGenerateEntity, WorkflowAppGenerateEntity
from libs.datetime_utils import naive_utc_now
from core.app.entities.queue_entities import (
QueueAgentLogEvent,
QueueIterationCompletedEvent,

@ -12,6 +12,8 @@ from typing import Any, Optional
from pydantic import BaseModel, Field
from libs.datetime_utils import naive_utc_now
class WorkflowType(StrEnum):
"""

@ -6,6 +6,7 @@ from typing import Optional
from pydantic import BaseModel, Field
from core.workflow.entities.node_entities import NodeRunResult
from libs.datetime_utils import naive_utc_now
from core.workflow.entities.workflow_node_execution import WorkflowNodeExecutionStatus

@ -12,6 +12,7 @@ from typing import Any, Optional, cast
from flask import Flask, current_app
from configs import dify_config
from libs.datetime_utils import naive_utc_now
from core.app.apps.exc import GenerateTaskStoppedError
from core.app.entities.app_invoke_entities import InvokeFrom
from core.workflow.entities.node_entities import AgentNodeStrategyInit, NodeRunResult

@ -11,6 +11,7 @@ from typing import TYPE_CHECKING, Any, Optional, cast
from flask import Flask, current_app
from configs import dify_config
from libs.datetime_utils import naive_utc_now
from core.variables import ArrayVariable, IntegerVariable, NoneVariable
from core.variables.segments import ArrayAnySegment, ArraySegment
from core.workflow.entities.node_entities import (

@ -11,6 +11,7 @@ from core.variables import (
Segment,
SegmentType,
)
from libs.datetime_utils import naive_utc_now
from core.workflow.entities.node_entities import NodeRunResult
from core.workflow.entities.workflow_node_execution import WorkflowNodeExecutionMetadataKey, WorkflowNodeExecutionStatus
from core.workflow.graph_engine.entities.event import (

@ -9,6 +9,7 @@ from core.workflow.graph_engine.entities.event import (
NodeRunStreamChunkEvent,
NodeRunSucceededEvent,
)
from libs.datetime_utils import naive_utc_now
from core.workflow.graph_engine.entities.graph import Graph
from core.workflow.graph_engine.entities.runtime_route_state import RouteNodeState
from core.workflow.nodes.answer.answer_stream_processor import AnswerStreamProcessor

@ -23,6 +23,7 @@ from core.workflow.repositories.workflow_execution_repository import WorkflowExe
from core.workflow.repositories.workflow_node_execution_repository import WorkflowNodeExecutionRepository
from core.workflow.system_variable import SystemVariable
from core.workflow.workflow_cycle_manager import CycleManagerWorkflowInfo, WorkflowCycleManager
from libs.datetime_utils import naive_utc_now
from models.enums import CreatorUserRole
from models.model import AppMode
from models.workflow import Workflow, WorkflowRun

Loading…
Cancel
Save