Fringg
eb18994b7d
- Migrate 660+ datetime.utcnow() across 153 files to datetime.now(UTC) - Migrate 30+ datetime.now() without UTC to datetime.now(UTC) - Convert all 170 DateTime columns to DateTime(timezone=True) - Add migrate_datetime_to_timestamptz() in universal_migration with SET LOCAL timezone='UTC' safety - Remove 70+ .replace(tzinfo=None) workarounds - Fix utcfromtimestamp → fromtimestamp(..., tz=UTC) - Fix fromtimestamp() without tz= (system_logs, backup_service, referral_diagnostics) - Fix fromisoformat/isoparse to ensure aware output (platega, yookassa, wata, miniapp, nalogo) - Fix strptime() to add .replace(tzinfo=UTC) (backup_service, referral_diagnostics) - Fix datetime.combine() to include tzinfo=UTC (remnawave_sync, traffic_monitoring) - Fix datetime.max/datetime.min sentinels with .replace(tzinfo=UTC) - Rename panel_datetime_to_naive_utc → panel_datetime_to_utc - Remove DTZ003 from ruff ignore list
106 lines
3.8 KiB
Python
106 lines
3.8 KiB
Python
from __future__ import annotations
|
|
|
|
import asyncio
|
|
import json
|
|
from collections.abc import Callable
|
|
from datetime import UTC, datetime
|
|
from typing import Any
|
|
|
|
import structlog
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.services.webhook_service import webhook_service
|
|
|
|
|
|
logger = structlog.get_logger(__name__)
|
|
|
|
|
|
class EventEmitter:
|
|
"""Event emitter для отслеживания и распространения событий системы."""
|
|
|
|
def __init__(self) -> None:
|
|
self._listeners: dict[str, list[Callable]] = {}
|
|
self._websocket_connections: set[Any] = set()
|
|
|
|
def on(self, event_type: str, callback: Callable) -> None:
|
|
"""Подписаться на событие."""
|
|
if event_type not in self._listeners:
|
|
self._listeners[event_type] = []
|
|
self._listeners[event_type].append(callback)
|
|
|
|
def off(self, event_type: str, callback: Callable) -> None:
|
|
"""Отписаться от события."""
|
|
if event_type in self._listeners:
|
|
try:
|
|
self._listeners[event_type].remove(callback)
|
|
except ValueError:
|
|
pass
|
|
|
|
def register_websocket(self, websocket: Any) -> None:
|
|
"""Зарегистрировать WebSocket подключение."""
|
|
self._websocket_connections.add(websocket)
|
|
logger.debug(
|
|
'WebSocket connection registered. Total', websocket_connections_count=len(self._websocket_connections)
|
|
)
|
|
|
|
def unregister_websocket(self, websocket: Any) -> None:
|
|
"""Отменить регистрацию WebSocket подключения."""
|
|
self._websocket_connections.discard(websocket)
|
|
logger.debug(
|
|
'WebSocket connection unregistered. Total', websocket_connections_count=len(self._websocket_connections)
|
|
)
|
|
|
|
async def emit(
|
|
self,
|
|
event_type: str,
|
|
payload: dict[str, Any],
|
|
db: AsyncSession | None = None,
|
|
) -> None:
|
|
"""Отправить событие всем подписчикам."""
|
|
event_data = {
|
|
'type': event_type,
|
|
'payload': payload,
|
|
'timestamp': str(datetime.now(UTC)),
|
|
}
|
|
|
|
# Вызываем локальные слушатели
|
|
if event_type in self._listeners:
|
|
for callback in self._listeners[event_type]:
|
|
try:
|
|
if asyncio.iscoroutinefunction(callback):
|
|
await callback(event_data)
|
|
else:
|
|
callback(event_data)
|
|
except Exception as error:
|
|
logger.exception('Error in event listener for', event_type=event_type, error=error)
|
|
|
|
# Отправляем через WebSocket
|
|
await self._broadcast_to_websockets(event_data)
|
|
|
|
# Отправляем webhooks
|
|
if db:
|
|
await webhook_service.send_webhook(db, event_type, payload)
|
|
|
|
async def _broadcast_to_websockets(self, event_data: dict[str, Any]) -> None:
|
|
"""Отправить событие всем подключенным WebSocket клиентам."""
|
|
if not self._websocket_connections:
|
|
return
|
|
|
|
disconnected = set()
|
|
message = json.dumps(event_data, default=str, ensure_ascii=False)
|
|
|
|
for ws in self._websocket_connections:
|
|
try:
|
|
await ws.send_text(message)
|
|
except Exception as error:
|
|
logger.warning('Failed to send WebSocket message', error=error)
|
|
disconnected.add(ws)
|
|
|
|
# Удаляем отключенные соединения
|
|
for ws in disconnected:
|
|
self.unregister_websocket(ws)
|
|
|
|
|
|
# Глобальный экземпляр event emitter
|
|
event_emitter = EventEmitter()
|