mirror of
https://github.com/rommapp/romm.git
synced 2026-02-19 07:50:57 +01:00
45 lines
1.4 KiB
Python
45 lines
1.4 KiB
Python
from decorators.database import begin_session
|
|
from handler.db_handler import DBHandler
|
|
from models.platform import Platform
|
|
from models.rom import Rom
|
|
from models.assets import Save, Screenshot, State
|
|
from sqlalchemy import func, select
|
|
from sqlalchemy.orm import Session
|
|
|
|
|
|
class DBStatsHandler(DBHandler):
|
|
@begin_session
|
|
def get_platforms_count(self, session: Session = None):
|
|
# Only count platforms with more then 0 roms
|
|
return session.scalar(
|
|
select(func.count())
|
|
.select_from(Platform)
|
|
.where(
|
|
select(func.count())
|
|
.select_from(Rom)
|
|
.filter_by(platform_id=Platform.id)
|
|
.as_scalar()
|
|
> 0
|
|
)
|
|
)
|
|
|
|
@begin_session
|
|
def get_roms_count(self, session: Session = None) -> int:
|
|
return session.scalar(select(func.count()).select_from(Rom))
|
|
|
|
@begin_session
|
|
def get_saves_count(self, session: Session = None) -> int:
|
|
return session.scalar(select(func.count()).select_from(Save))
|
|
|
|
@begin_session
|
|
def get_states_count(self, session: Session = None) -> int:
|
|
return session.scalar(select(func.count()).select_from(State))
|
|
|
|
@begin_session
|
|
def get_screenshots_count(self, session: Session = None) -> int:
|
|
return session.scalar(select(func.count()).select_from(Screenshot))
|
|
|
|
@begin_session
|
|
def get_total_filesize(self, session: Session = None) -> int:
|
|
return 0
|