editing item

gui/modules/edit/__init__.py

@@ -0,0 +1 @@
+from .on_edit import *

gui/modules/edit/on_edit.py

@@ -0,0 +1,64 @@
+from gui.gui import Ui_MainWindow
+from data import common
+from gui.modules.core import items_list
+from modules.database import Database
+from modules.database.model import Item
+
+
+def fill_type_box(ui: Ui_MainWindow):
+    if ui.edit_box_item_class.currentText() == "Vehicle":
+        ui.edit_box_item_type.clear()
+        ui.edit_box_item_type.addItems(common.vehicle_type)
+    elif ui.edit_box_item_class.currentText() == "Property":
+        ui.edit_box_item_type.clear()
+        ui.edit_box_item_type.addItems(common.property_type)
+    else:
+        ui.edit_box_item_type.clear()
+        ui.edit_box_item_type.addItem("Other")
+
+
+def on_edit_item_click(ui: Ui_MainWindow):
+    item_to_edit_name = items_list.selected_item(ui)
+
+    if not item_to_edit_name:
+        return
+
+    item_to_edit = Database.get().items[item_to_edit_name]
+
+    ui.edit_old_item_name.setText(item_to_edit_name)
+
+    ui.edit_box_item_name.setText(item_to_edit.item_name)
+    ui.edit_box_price.setValue(item_to_edit.price)
+    ui.edit_box_item_class.addItems(common.items_class)
+    ui.edit_box_item_class.setCurrentText(item_to_edit.item_class)
+    ui.edit_box_item_type.addItem("Other")
+    ui.edit_box_item_type.addItem(item_to_edit.item_type)
+    ui.edit_box_item_type.setCurrentText(item_to_edit.item_type)
+    ui.edit_box_shop.addItems(common.shop_sites)
+    ui.edit_box_shop.setCurrentText(item_to_edit.shop)
+    ui.edit_box_image.setText(item_to_edit.image)
+
+    ui.content.setCurrentWidget(ui.edit_item_page)
+
+
+def on_save_edit(ui: Ui_MainWindow):
+    if {"", 0} & {ui.edit_box_item_name.text(), ui.edit_box_price.value()}:
+        return
+
+    db = Database.get()
+    db.items[ui.edit_box_item_name.text()] = Item(
+        ui.edit_box_item_name.text(),
+        ui.edit_box_item_class.currentText(),
+        ui.edit_box_item_type.currentText(),
+        ui.edit_box_shop.currentText(),
+        ui.edit_box_price.value(),
+        ui.edit_box_image.text() if ui.edit_box_image.text() != "" else None
+    )
+
+    if ui.edit_box_item_name.text() != ui.edit_old_item_name.text():
+        db.items.pop(ui.edit_old_item_name.text())
+
+    Database.write(db)
+
+    ui.content.setCurrentWidget(ui.main_page)
+    items_list.refill_list(ui)