luxick/DrunkSoulsStatTool
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Stat info for current episode.

Browse Source
This commit is contained in:
luxick
2018-02-27 20:54:10 +01:00
parent 1e4e319ecc
commit 5631718969
7 changed files with 397 additions and 129 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
dsst/dsst_gtk3/gtk_ui.py
View File

@@ -1,3 +1,5 @@
from collections import Counter
import gi
gi.require_version('Gtk', '3.0')
from gi.repository import Gtk
@@ -53,13 +55,15 @@ class GtkUi:
if season_id is None or season_id == -1:
return
# Rebuild episodes store
ep_id = self.get_selected_episode_id()
selection = self.ui.get_object('episodes_tree_view').get_selection()
# selection.handler_block_by_func(self.handlers.on_selected_episode_changed)
store = self.ui.get_object('episodes_store')
store.clear()
selection.handler_block_by_func(self.handlers.on_selected_episode_changed)
model, selected_paths = selection.get_selected_rows()
model.clear()
for episode in sql_func.get_episodes_for_season(season_id):
store.append([episode.id, episode.number, str(episode.date)])
model.append([episode.id, episode.number, str(episode.date)])
if selected_paths:
selection.select_path(selected_paths[0])
selection.handler_unblock_by_func(self.handlers.on_selected_episode_changed)
# Load player stats for season
player_stats = {}
@@ -72,7 +76,10 @@ class GtkUi:
for name, stats in player_stats.items():
store.append([name, stats[0], stats[1]])
# Load enemy stats for season
enemy_stats = {enemy.name: [0, 0, enemy.id] for enemy in sql.Season.get(sql.Season.id == season_id).enemies}
season = sql.Season.get(sql.Season.id == season_id)
enemy_stats = {
enemy.name: [False, len(sql.Death.select().where(sql.Death.enemy == enemy)), enemy.id]
for enemy in season.enemies}
store = self.ui.get_object('enemy_season_store')
store.clear()
for name, stats in enemy_stats.items():
@@ -84,10 +91,31 @@ class GtkUi:
episode_id = self.get_selected_episode_id()
if not episode_id:
return
episode = sql.Episode.get(sql.Episode.id == episode_id)
store = self.ui.get_object('episode_players_store')
store.clear()
for player in sql.Episode.get(sql.Episode.id == self.get_selected_episode_id()).players:
for player in episode.players:
store.append([player.id, player.name, player.hex_id])
# Reload death store for notebook view
store = self.ui.get_object('episode_deaths_store')
store.clear()
for death in episode.deaths:
penalties = [x.drink.name for x in death.penalties]
penalties = [f'{number}x {drink}' for drink, number in Counter(penalties).items()]
penalty_string = ', '.join(penalties)
store.append([death.id, death.player.name, death.enemy.name, penalty_string])
# Stat grid
self.ui.get_object('ep_death_count_label').set_text(str(len(episode.deaths)))
drink_count = sum(len(death.penalties) for death in episode.deaths)
self.ui.get_object('ep_drinks_label').set_text(str(drink_count))
cl_booze = sum(len(death.penalties) * death.penalties[0].size for death in episode.deaths)
self.ui.get_object('ep_booze_label').set_text(str(cl_booze) + "cl")
enemy_list = [death.enemy.name for death in episode.deaths]
sorted_list = Counter(enemy_list).most_common(1)
if sorted_list:
enemy_name, deaths = sorted_list[0]
self.ui.get_object('ep_enemy_name_label').set_text(f'{enemy_name} ({deaths} Deaths)')
def get_selected_season_id(self) -> int:
"""Read ID of the selected season from the UI