# SPDX-License-Identifier: WTFPL import os from gi.repository import GLib, Gtk, Gdk from .view import View from yms import gtkutils, combobox, treeview from yms.lookup import Dropdown from yms.model import storage class StorageView(View): def __init__(self, context): super().__init__(context, _("Storage"), "storage.ui") self.context = context self.storage = storage.Storage(context.db) # current record self.newrecord = False self.lst_storage = Gtk.ListStore(int, str, str, str) self.load_list() treeview.add_column(self.tree['storage'], _('ID'), 0) treeview.add_column(self.tree['storage'], _('Name'), 1) treeview.add_column(self.tree['storage'], _('Position'), 2) treeview.add_column(self.tree['storage'], _('Remarks'), 3) self.tree['storage'].set_model(self.lst_storage) self.dd_stype = self.context.lookup.dropdown(Dropdown.STORAGE_TYPE) combobox.load_from_dict(self.combo['type'], self.dd_stype) self.dd_unit = self.context.lookup.enum('storage.capaunit') combobox.load_from_dict(self.combo['unit'], self.dd_unit) def add_row(self): pass def refresh_row(self): pass def load_list(self): self.lst_storage.clear() sql = ( "SELECT sid, sname, CONCAT(x, ',', y, ',', z) AS position, remarks " "FROM storage " "WHERE vid=%s " "ORDER BY sname" ) self.db.cur.execute(sql, (self.context.vessel.vid,)) for row in self.db.cur.fetchall(): self.lst_storage.append(row) def clear_record(self): self.storage.clear() self.entry['sname'].set_text('') self.spin['x'].set_value(0) self.spin['y'].set_value(0) self.spin['z'].set_value(0) self.spin['wx'].set_value(0) self.spin['wy'].set_value(0) self.spin['wz'].set_value(0) self.spin['angle'].set_value(0) self.spin['capacity'].set_value(0) combobox.activate_row(self.combo['type'], -1) combobox.activate_row(self.combo['unit'], -1) self.text['remarks'].get_buffer().set_text('') def display_record(self): self.label['name'].set_text(self.storage.sname) self.label['type'].set_text(self.dd_stype[self.storage.stype]) self.label['position'].set_text(f"({self.storage.x}, {self.storage.y}, {self.storage.z})") self.label['dimension'].set_text(f"({self.storage.wx}, {self.storage.wy}, {self.storage.wz})") self.label['angle'].set_text(f"{self.storage.angle}°") self.label['capacity'].set_text(f"{self.storage.capacity} {self.storage.capaunit}") if self.storage.color: self.label['color'].set_markup(f'#{self.storage.color} ●') else: self.label['color'].set_text('') self.label['remarks'].set_text(self.storage.remarks or '') def edit_record(self): self.entry['sname'].set_text(self.storage.sname) self.spin['x'].set_value(self.storage.x) self.spin['y'].set_value(self.storage.y) self.spin['z'].set_value(self.storage.z) self.spin['wx'].set_value(self.storage.wx) self.spin['wy'].set_value(self.storage.wy) self.spin['wz'].set_value(self.storage.wz) self.spin['angle'].set_value(self.storage.angle) self.spin['capacity'].set_value(self.storage.capacity) combobox.activate_row(self.combo['type'], self.storage.stype) combobox.activate_row(self.combo['unit'], self.storage.capaunit) # flags self.text['remarks'].get_buffer().set_text(self.storage.remarks or '') # buttons on list page def on_but_new_clicked(self, widget): self.newrecord = True self.label['title_edit'].set_text("Add new Storage") self.clear_record() self.storage.vid = self.context.vessel.vid self.topwidget.set_visible_child(self.box['stor_edit']) def on_but_edit_clicked(self, widget): sid = treeview.get_value(self.tree['storage'], 0) if sid is None: return self.newrecord = False self.label['title_edit'].set_text("Edit Storage") self.storage.load(sid) self.edit_record() self.topwidget.set_visible_child(self.box['stor_edit']) def on_but_detail_clicked(self, widget): sid = treeview.get_value(self.tree['storage'], 0) if not sid: return self.storage.load(sid) self.display_record() self.topwidget.set_visible_child(self.box['stor_detail']) self.button['close'].grab_focus() def on_tvw_storage_row_activated(self, widget, path, column): sid = treeview.get_value(widget, 0) self.storage.load(sid) self.display_record() self.topwidget.set_visible_child(self.box['stor_detail']) self.button['close'].grab_focus() # buttons on detail page # edit2 uses edit callback def on_but_close_clicked(self, widget): self.topwidget.set_visible_child(self.box['storage']) def on_but_delete_clicked(self, widget): self.topwidget.set_visible_child(self.box['storage']) self.newrecord = False # buttons on edit page def on_but_save_clicked(self, widget): # check mandatory fields sname = self.entry['sname'].get_text().strip() if not sname: # TODO error marker return # get fields from ui form self.storage.sname = sname self.storage.stype = combobox.get_list_value(self.combo['type']) self.storage.x = self.spin['x'].get_value() self.storage.y = self.spin['y'].get_value() self.storage.z = self.spin['z'].get_value() self.storage.wx = self.spin['wx'].get_value() self.storage.wy = self.spin['wy'].get_value() self.storage.wz = self.spin['wz'].get_value() self.storage.angle = self.spin['angle'].get_value() self.storage.capacity = self.spin['capacity'].get_value() self.storage.capaunit = combobox.get_list_value(self.combo['unit']) self.storage.remarks = gtkutils.get_text_contents(self.text['remarks']) self.storage.save() if self.newrecord: self.load_list() self.newrecord = False self.topwidget.set_visible_child(self.box['storage']) def on_but_cancel_clicked(self, widget): self.topwidget.set_visible_child(self.box['storage']) self.newrecord = False