mirror of
https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot.git
synced 2026-02-26 22:31:44 +00:00
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
67 lines
2.0 KiB
Python
67 lines
2.0 KiB
Python
from __future__ import annotations
|
|
|
|
import asyncio
|
|
import uuid
|
|
from dataclasses import dataclass, field
|
|
from datetime import UTC, datetime
|
|
|
|
from app.services.backup_service import backup_service
|
|
|
|
|
|
@dataclass(slots=True)
|
|
class BackupTaskState:
|
|
task_id: str
|
|
status: str = 'queued'
|
|
message: str | None = None
|
|
file_path: str | None = None
|
|
created_by: int | None = None
|
|
created_at: datetime = field(default_factory=lambda: datetime.now(UTC))
|
|
updated_at: datetime = field(default_factory=lambda: datetime.now(UTC))
|
|
|
|
|
|
class BackupTaskManager:
|
|
def __init__(self) -> None:
|
|
self._tasks: dict[str, BackupTaskState] = {}
|
|
self._lock = asyncio.Lock()
|
|
|
|
async def enqueue(self, *, created_by: int | None) -> BackupTaskState:
|
|
task_id = uuid.uuid4().hex
|
|
state = BackupTaskState(task_id=task_id, created_by=created_by)
|
|
|
|
async with self._lock:
|
|
self._tasks[task_id] = state
|
|
|
|
asyncio.create_task(self._run_task(state))
|
|
return state
|
|
|
|
async def _run_task(self, state: BackupTaskState) -> None:
|
|
state.status = 'running'
|
|
state.updated_at = datetime.now(UTC)
|
|
|
|
try:
|
|
success, message, file_path = await backup_service.create_backup(created_by=state.created_by)
|
|
state.message = message
|
|
state.file_path = file_path
|
|
state.status = 'completed' if success else 'failed'
|
|
except Exception as exc:
|
|
state.status = 'failed'
|
|
state.message = f'Unexpected error: {exc}'
|
|
finally:
|
|
state.updated_at = datetime.now(UTC)
|
|
|
|
async def get(self, task_id: str) -> BackupTaskState | None:
|
|
async with self._lock:
|
|
return self._tasks.get(task_id)
|
|
|
|
async def list(self, *, active_only: bool = False) -> list[BackupTaskState]:
|
|
async with self._lock:
|
|
states = list(self._tasks.values())
|
|
|
|
if active_only:
|
|
return [state for state in states if state.status in {'queued', 'running'}]
|
|
|
|
return states
|
|
|
|
|
|
backup_task_manager = BackupTaskManager()
|