Create and modify enemies function.
This commit is contained in:
@@ -1,6 +1,7 @@
|
||||
import datetime
|
||||
|
||||
from dsst_gtk3 import dialogs, util, gtk_ui
|
||||
from dsst_gtk3 import dialogs, util, gtk_ui, reload
|
||||
from common import models
|
||||
from gi.repository import Gtk
|
||||
|
||||
|
||||
@@ -29,10 +30,28 @@ class DialogHandlers:
|
||||
def do_add_enemy(self, entry):
|
||||
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, enemy.id])
|
||||
enemy = models.Enemy()
|
||||
enemy.name = entry.get_text()
|
||||
enemy.season = self.app.get_selected_season_id()
|
||||
enemy.boss = not self.app.ui.get_object('enemy_optional_ckeck').get_active()
|
||||
self.app.ui.get_object('enemy_optional_ckeck').set_active(False)
|
||||
entry.set_text('')
|
||||
|
||||
self.app.update_enemy(enemy)
|
||||
|
||||
def on_enemy_name_edited(self, _, index, value):
|
||||
row = self.app.ui.get_object('enemy_season_store')[index]
|
||||
enemy = [enemy for enemy in self.app.enemies.data if enemy.id == row[4]][0]
|
||||
enemy.name = value
|
||||
self.app.update_enemy(enemy)
|
||||
|
||||
def on_enemy_optional_edited(self, renderer, index):
|
||||
new_optional_value = not renderer.get_active()
|
||||
row = self.app.ui.get_object('enemy_season_store')[index]
|
||||
enemy = [enemy for enemy in self.app.enemies.data if enemy.id == row[4]][0]
|
||||
enemy.boss = new_optional_value
|
||||
self.app.update_enemy(enemy)
|
||||
|
||||
def do_show_date_picker(self, entry: 'Gtk.Entry', *_):
|
||||
dialog = self.app.ui.get_object('date_picker_dialog')
|
||||
result = dialog.run()
|
||||
|
||||
Reference in New Issue
Block a user