mirror of
https://github.com/rommapp/romm.git
synced 2026-01-30 01:52:02 +08:00
43 lines
1.5 KiB
Python
43 lines
1.5 KiB
Python
from decorators.database import begin_session
|
|
from models.assets import Save, Screenshot, State
|
|
from models.rom import Rom
|
|
from sqlalchemy import distinct, func, select
|
|
from sqlalchemy.orm import Session
|
|
|
|
from .base_handler import DBBaseHandler
|
|
|
|
|
|
class DBStatsHandler(DBBaseHandler):
|
|
@begin_session
|
|
def get_platforms_count(self, session: Session = None) -> int:
|
|
"""Get the number of platforms with any roms."""
|
|
return (
|
|
session.scalar(
|
|
select(func.count(distinct(Rom.platform_id))).select_from(Rom)
|
|
)
|
|
or 0
|
|
)
|
|
|
|
@begin_session
|
|
def get_roms_count(self, session: Session = None) -> int:
|
|
return session.scalar(select(func.count()).select_from(Rom)) or 0
|
|
|
|
@begin_session
|
|
def get_saves_count(self, session: Session = None) -> int:
|
|
return session.scalar(select(func.count()).select_from(Save)) or 0
|
|
|
|
@begin_session
|
|
def get_states_count(self, session: Session = None) -> int:
|
|
return session.scalar(select(func.count()).select_from(State)) or 0
|
|
|
|
@begin_session
|
|
def get_screenshots_count(self, session: Session = None) -> int:
|
|
return session.scalar(select(func.count()).select_from(Screenshot)) or 0
|
|
|
|
@begin_session
|
|
def get_total_filesize(self, session: Session = None) -> int:
|
|
"""Get the total filesize of all roms in the database, in bytes."""
|
|
return (
|
|
session.scalar(select(func.sum(Rom.file_size_bytes)).select_from(Rom)) or 0
|
|
)
|