59 lines
2.0 KiB
Python
59 lines
2.0 KiB
Python
|
from sqlalchemy import select, func
|
||
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||
|
|
|
||
|
|
from ..db_models import AnimeTheme
|
||
|
|
from ..schemas import StorageStatsResponse
|
||
|
|
from ..config import downloader_settings
|
||
|
|
from ...db_models import Opening
|
||
|
|
|
||
|
|
|
||
|
|
class StorageTrackerService:
|
||
|
|
"""Service for tracking S3 storage usage from DB (without scanning S3)."""
|
||
|
|
|
||
|
|
def __init__(self, db: AsyncSession):
|
||
|
|
self.db = db
|
||
|
|
|
||
|
|
async def get_stats(self) -> StorageStatsResponse:
|
||
|
|
"""Calculate storage stats from database."""
|
||
|
|
|
||
|
|
# Sum file sizes from downloaded themes
|
||
|
|
result = await self.db.execute(
|
||
|
|
select(func.coalesce(func.sum(AnimeTheme.file_size_bytes), 0))
|
||
|
|
.where(AnimeTheme.file_size_bytes.isnot(None))
|
||
|
|
)
|
||
|
|
used_bytes = result.scalar() or 0
|
||
|
|
|
||
|
|
# Count openings in the main Opening table
|
||
|
|
result = await self.db.execute(
|
||
|
|
select(func.count(Opening.id))
|
||
|
|
)
|
||
|
|
openings_count = result.scalar() or 0
|
||
|
|
|
||
|
|
limit_bytes = downloader_settings.s3_storage_limit_bytes
|
||
|
|
available_bytes = max(0, limit_bytes - used_bytes)
|
||
|
|
used_percent = (used_bytes / limit_bytes * 100) if limit_bytes > 0 else 0
|
||
|
|
|
||
|
|
return StorageStatsResponse(
|
||
|
|
used_bytes=used_bytes,
|
||
|
|
limit_bytes=limit_bytes,
|
||
|
|
used_percent=round(used_percent, 2),
|
||
|
|
available_bytes=available_bytes,
|
||
|
|
can_download=used_bytes < limit_bytes,
|
||
|
|
openings_count=openings_count,
|
||
|
|
)
|
||
|
|
|
||
|
|
async def get_estimated_queue_size(self) -> int:
|
||
|
|
"""Get estimated size of pending downloads in queue."""
|
||
|
|
from ..db_models import DownloadTask, DownloadStatus
|
||
|
|
|
||
|
|
result = await self.db.execute(
|
||
|
|
select(func.coalesce(func.sum(DownloadTask.estimated_size_bytes), 0))
|
||
|
|
.where(DownloadTask.status.in_([
|
||
|
|
DownloadStatus.QUEUED,
|
||
|
|
DownloadStatus.DOWNLOADING,
|
||
|
|
DownloadStatus.CONVERTING,
|
||
|
|
DownloadStatus.UPLOADING,
|
||
|
|
]))
|
||
|
|
)
|
||
|
|
return result.scalar() or 0
|