52 lines
1.9 KiB
Python
52 lines
1.9 KiB
Python
"""
|
|
This module contains shorthand functions for common queries to ease access from the UI
|
|
"""
|
|
from data_access.sql import *
|
|
|
|
|
|
def get_episodes_for_season(season_id: int) -> list:
|
|
"""Load list of episodes for a specific season
|
|
:param season_id: ID of a season
|
|
:return: List of sql.Episode or empty list
|
|
"""
|
|
try:
|
|
return list(Season.get(Season.id == season_id).episodes)
|
|
except Episode.DoesNotExist:
|
|
return []
|
|
|
|
|
|
def get_player_deaths_for_season(season_id: int, player_id: int) -> int:
|
|
"""Load all the aggregate count of all deaths for a player in a given season
|
|
:param season_id: ID of a season
|
|
:param player_id: ID of a player
|
|
:return: Number of deaths of the player in the season
|
|
"""
|
|
deaths = 0
|
|
for episode in list(Season.get(Season.id == season_id).episodes):
|
|
deaths = deaths + len([death for death in list(episode.deaths) if death.player.id == player_id])
|
|
return deaths
|
|
|
|
|
|
def get_player_victories_for_season(season_id: int, player_id: int) -> int:
|
|
"""Load all the aggregate count of all victories for a player in a given season
|
|
:param season_id: ID of a season
|
|
:param player_id: ID of a player
|
|
:return: Number of all victories of the player in the season
|
|
"""
|
|
victories = 0
|
|
for episode in list(Season.get(Season.id == season_id).episodes):
|
|
victories = victories + len([vic for vic in list(episode.victories) if vic.player.id == player_id])
|
|
return victories
|
|
|
|
|
|
def create_tables():
|
|
"""Create all database tables"""
|
|
models = [Season, Episode, Player, Drink, Enemy, Death, Victory, Penalty, Episode.players.get_through_model()]
|
|
for model in models:
|
|
model.create_table()
|
|
|
|
|
|
def drop_tables():
|
|
"""Drop all data in database"""
|
|
models = [Season, Episode, Player, Drink, Enemy, Death, Victory, Penalty, Episode.players.get_through_model()]
|
|
db.drop_tables(models) |