"""Shared imports for AI PMO cadence service."""

from __future__ import annotations

from datetime import datetime, timedelta, timezone
from typing import Any, Optional

from sqlalchemy import and_, func, or_, select
from sqlalchemy.ext.asyncio import AsyncSession

from ...models import (
    Budget,
    Notification,
    PortfolioSnapshot,
    Project,
    ProjectDomain,
    ProjectSignalLink,
    Signal,
    SignalSourceRegistry,
    Task,
    UserProject,
    UserWorkload,
    WorkItem,
)
from ...utils.timezone import now_cn_naive


def utc_now_naive() -> datetime:
    return now_cn_naive()


__all__ = [
    "Any",
    "AsyncSession",
    "Budget",
    "Notification",
    "Optional",
    "PortfolioSnapshot",
    "Project",
    "ProjectDomain",
    "ProjectSignalLink",
    "Signal",
    "SignalSourceRegistry",
    "Task",
    "UserProject",
    "UserWorkload",
    "WorkItem",
    "and_",
    "datetime",
    "func",
    "or_",
    "select",
    "timedelta",
    "timezone",
    "utc_now_naive",
]