137 lines
5.2 KiB
Python
137 lines
5.2 KiB
Python
import os
|
|
import gi
|
|
gi.require_version('Gtk', '3.0')
|
|
from gi.repository import Gtk
|
|
from dsst_gtk3.handlers import handlers
|
|
from dsst_gtk3 import util, reload, client
|
|
from common import models
|
|
|
|
class GtkUi:
|
|
""" The main UI class """
|
|
def __init__(self, config: dict):
|
|
# Load Glade UI files
|
|
self.ui = Gtk.Builder()
|
|
glade_resources = [
|
|
['dsst_gtk3', 'resources', 'glade', 'window.glade'],
|
|
['dsst_gtk3', 'resources', 'glade', 'dialogs.glade']
|
|
]
|
|
for path in glade_resources:
|
|
self.ui.add_from_string(util.load_ui_resource_string(path))
|
|
# Set the status bar logo
|
|
dd_logo = ['dsst_gtk3', 'resources', 'images', 'dd.png']
|
|
logo_pixbuf = util.load_image_resource(dd_logo, 60, 13)
|
|
logo = self.ui.get_object('status_bar_logo').set_from_pixbuf(logo_pixbuf)
|
|
# Connect signal handlers to UI
|
|
self.handlers = handlers.Handlers(self)
|
|
self.ui.connect_signals(self.handlers)
|
|
# Show all widgets
|
|
self.ui.get_object('main_window').show_all()
|
|
# Connect to data server
|
|
config = config['servers'][0]
|
|
self.data_client = client.Access(config)
|
|
# Create local data caches
|
|
self.players = util.Cache()
|
|
self.drinks = util.Cache()
|
|
self.seasons = util.Cache()
|
|
self.episodes = util.Cache()
|
|
self.enemies = util.Cache()
|
|
self.season_stats = util.Cache()
|
|
# Create meta data cache
|
|
self.meta = {'connection': '{}:{}'.format(config.get('host'), config.get('port'))}
|
|
# Load base data and seasons
|
|
self.load_server_meta()
|
|
self.full_reload()
|
|
self.update_status_bar_meta()
|
|
|
|
def load_server_meta(self):
|
|
self.meta['database'] = self.data_client.send_request('load_db_meta')
|
|
|
|
def full_reload(self):
|
|
with util.network_operation(self):
|
|
self.players.data = self.data_client.send_request('load_players')
|
|
self.drinks.data = self.data_client.send_request('load_drinks')
|
|
self.seasons.data = self.data_client.send_request('load_seasons')
|
|
season_id = self.get_selected_season_id()
|
|
if season_id:
|
|
self.episodes.data = self.data_client.send_request('load_episodes', season_id)
|
|
self.season_stats.data = self.data_client.send_request('load_season_stats', season_id)
|
|
cur_season = [s for s in self.seasons.data if s.id == season_id][0]
|
|
self.enemies.data = cur_season.enemies
|
|
reload.rebuild_view_data(self)
|
|
|
|
def reload(self):
|
|
pass
|
|
|
|
def update_enemy(self, enemy: 'models.Enemy'):
|
|
with util.network_operation(self):
|
|
self.data_client.send_request('update_enemy', enemy)
|
|
self.full_reload()
|
|
|
|
def update_player(self, player: 'models.Player'):
|
|
with util.network_operation(self):
|
|
self.data_client.send_request('update_player', player)
|
|
self.full_reload()
|
|
|
|
def update_drink(self, drink: 'models.Drink'):
|
|
with util.network_operation(self):
|
|
self.data_client.send_request('update_drink', drink)
|
|
self.full_reload()
|
|
|
|
def save_death(self, death: 'models.Death'):
|
|
with util.network_operation(self):
|
|
self.data_client.send_request('save_death', death)
|
|
self.full_reload()
|
|
|
|
def save_victory(self, victory: 'models.Victory'):
|
|
with util.network_operation(self):
|
|
self.data_client.send_request('save_victory', victory)
|
|
self.full_reload()
|
|
|
|
def update_season(self, season: 'models.Season'):
|
|
with util.network_operation(self):
|
|
self.data_client.send_request('update_season', season)
|
|
self.seasons.valid = False
|
|
|
|
def update_episode(self, episode: 'models.Episode'):
|
|
with util.network_operation(self):
|
|
self.data_client.send_request('update_episode', episode)
|
|
self.episodes.valid = False
|
|
self.season_stats.valid = False
|
|
|
|
def update_status_bar_meta(self):
|
|
self.ui.get_object('connection_label').set_text(self.meta.get('connection'))
|
|
self.ui.get_object('db_label').set_text(self.meta.get('database') or '')
|
|
|
|
def get_selected_season_id(self) -> int:
|
|
"""Read ID of the selected season from the UI
|
|
:return: ID of the selected season
|
|
"""
|
|
season_id = util.get_combo_value(self.ui.get_object('season_combo_box'), 0)
|
|
return season_id if season_id != -1 else None
|
|
|
|
def get_selected_episode_id(self):
|
|
"""Parse ID of the selected episode from the UI
|
|
:return: ID of the selected episode
|
|
"""
|
|
(model, tree_iter) = self.ui.get_object('episodes_tree_view').get_selection().get_selected()
|
|
return model.get_value(tree_iter, 0) if tree_iter else None
|
|
|
|
@staticmethod
|
|
def get_by_id(cache: 'util.Cache', object_id: int):
|
|
try:
|
|
return [x for x in cache.data if x.id == object_id][0]
|
|
except KeyError:
|
|
return None
|
|
|
|
|
|
def main():
|
|
if not os.path.isfile(util.CONFIG_PATH):
|
|
util.save_config(util.DEFAULT_CONFIG, util.CONFIG_PATH)
|
|
config = util.load_config(util.CONFIG_PATH)
|
|
GtkUi(config)
|
|
Gtk.main()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|