Stat info for current episode.

dsst/dsst_gtk3/handlers/death_handlers.py

@@ -1,3 +1,4 @@
+from gi.repository import Gtk
 from dsst_gtk3 import dialogs, gtk_ui
 
 
@@ -9,8 +10,9 @@ class DeathHandlers:
     def do_add_death(self, *_):
         ep_id = self.app.get_selected_episode_id()
         result = dialogs.show_edit_death_dialog(self.app.ui, ep_id)
-        if result:
+        if result == Gtk.ResponseType.OK:
             self.app.reload_for_season()
+            self.app.reload_for_episode()
 
     def on_penalty_drink_changed(self, widget, path, text):
-        self.app.ui.get_object('player_penalties_store')[path][2] = text
+        self.app.ui.get_object('player_penalties_store')[path][2] = text

dsst/dsst_gtk3/handlers/dialog_handlers.py

@@ -23,7 +23,7 @@ class DialogHandlers:
         if entry.get_text():
             store = self.app.ui.get_object('enemy_season_store')
             enemy = sql.Enemy.create(name=entry.get_text(), season=self.app.get_selected_season_id())
-            store.append([enemy.name, False, 0])
+            store.append([enemy.name, False, 0, enemy.id])
             entry.set_text('')
 
     def do_manage_drinks(self, *_):

dsst/dsst_gtk3/handlers/handlers.py

@@ -1,11 +1,11 @@
 import gi
-import sql_func
 gi.require_version('Gtk', '3.0')
 from gi.repository import Gtk
 from dsst_gtk3.handlers.season_handlers import SeasonHandlers
 from dsst_gtk3.handlers.base_data_handlers import BaseDataHandlers
 from dsst_gtk3.handlers.dialog_handlers import DialogHandlers
 from dsst_gtk3.handlers.death_handlers import DeathHandlers
+from dsst_sql import sql_func
 
 
 class Handlers(SeasonHandlers, BaseDataHandlers, DialogHandlers, DeathHandlers):

dsst/dsst_gtk3/handlers/season_handlers.py

@@ -11,7 +11,7 @@ class SeasonHandlers:
         name = dialogs.enter_string_dialog(self.app.ui, 'Name for the new Season')
         if name:
             sql.Season.create(game_name=name, number=1)
-            self.app.reload_seasons()
+            self.app.reload_base_data()
 
     def do_season_selected(self, *_):
         self.app.reload_for_season()