all the buttons work!!!

Author: GiovanH

gui.py

@@ -3,6 +3,7 @@
 import os
 import webbrowser
 from pathlib import Path
+from tkinter.simpledialog import askstring
 from typing import Any, Callable, Literal, Optional, TypeAlias, Union
 
 from roa import RoaCategoriesFile, RoaCategory, RoaEntry, RoaOrderFile
@@ -16,11 +17,13 @@
 
 Direction: TypeAlias = Union[Literal[1], Literal[-1]]
 
+
 @dataclass
 class ListboxItem():
     label: str
     value: Any
 
+
 class Counter():
     def __init__(self, value=0):
         self.value = value
@@ -30,34 +33,35 @@ def inc(self):
         self.value += 1
         return last_val
 
+
 def sort_name(entry: RoaEntry):
     try:
         return entry.name.upper()
     except:
         return 'ERROR'
 
+
 class ItemListFrame(tk.Frame):
     def __init__(
         self,
         parent,
         selectmode=tk.SINGLE
     ) -> None:
-        super().__init__(parent, bg="yellow")
-
+        super().__init__(parent)
         self.items: list[ListboxItem] = []
 
         myscroll = tk.Scrollbar(self)
-        myscroll.pack(side = tk.RIGHT, fill = tk.Y)
+        myscroll.pack(side=tk.RIGHT, fill=tk.Y)
 
         self.listbox = tk.Listbox(
             self,
             relief=tk.GROOVE,
             selectmode=selectmode,
             exportselection=False,
-            yscrollcommand = myscroll.set
+            yscrollcommand=myscroll.set
         )
 
-        myscroll.config(command = self.listbox.yview)
+        myscroll.config(command=self.listbox.yview)
 
         self.listbox.pack(side=tk.TOP, fill=tk.Y, expand=1)
 
@@ -87,23 +91,23 @@ def move_selected_items(self, sel_indexes, direction: Direction) -> list[Listbox
         return self.items
 
 
-class CharacterManager(tk.Tk):
+class CharacterManager(tk.Tk):  # noqa: PLR0904
     def __init__(
         self,
         order_roa: RoaOrderFile,
         categories_roa: RoaCategoriesFile
     ) -> None:
         super().__init__()
+        self.title("Re-ROAder")
 
         self.text_status: tk.StringVar = tk.StringVar(value="Status")
         self.initwindow()
 
         self.order_roa: RoaOrderFile = order_roa
         self.categories_roa: RoaCategoriesFile = categories_roa
 
-        self.nested_state: dict[str, list[RoaEntry]] = roa_zip_chars(order_roa, categories_roa)
-
-        self.load_roa_state()
+        self.load_state_from_roa()
+        self.load_gui_from_state()
 
         self.mainloop()
 
@@ -112,12 +116,20 @@ def log(self, line) -> None:
         print(line)
         self.text_status.set(line)
 
+    def load_state_from_roa(self):
+        self.nested_state: dict[str, list[RoaEntry]] = roa_zip_chars(order_roa, categories_roa)
+
     # Widget management
 
     def widget_buttons_middle(self) -> tk.Frame:
         frame = tk.Frame(self)
         y = Counter()
 
+        btn_export = ttk.Button(
+            frame, text="Reload discarding changes",
+            command=self.load_state_from_roa)
+        btn_export.grid(row=y.inc(), sticky=tk.EW)
+
         btn_export = ttk.Button(
             frame, text="Export to ROA",
             command=self.save_state_to_roas)
@@ -142,13 +154,14 @@ def widget_buttons_cats(self) -> tk.Frame:
 
         frame_updown.grid(row=y.inc(), sticky=tk.EW)
 
-        btn_add = ttk.Button(frame, text="TODO Add")
+        btn_add = ttk.Button(frame, text="Add", command=self.add_category)
         btn_add.grid(row=y.inc(), sticky=tk.EW)
 
-        btn_del = ttk.Button(frame, text="TODO Delete")
+        btn_del = ttk.Button(frame, text="Delete", command=self.delete_category)
         btn_del.grid(row=y.inc(), sticky=tk.EW)
 
-        btn_rename = ttk.Button(frame, text="TODO Rename")
+        btn_rename = ttk.Button(frame, text="Rename",
+                                command=self.interactive_rename_category)
         btn_rename.grid(row=y.inc(), sticky=tk.EW)
 
         return frame
@@ -184,15 +197,21 @@ def widget_buttons_chars(self) -> tk.Frame:
         )
         btn_char_info.grid(row=y.inc(), sticky=tk.EW)
 
-        btn_char_movecat = ttk.Button(
-            frame, text="TODO Move to...",
-            # command=
-        )
-        btn_char_movecat.grid(row=y.inc(), sticky=tk.EW)
+        # btn_char_movecat = ttk.Button(
+        #     frame, text="TODO Move to...",
+        #     # command=
+        # )
+        # btn_char_movecat.grid(row=y.inc(), sticky=tk.EW)
+
+        self.combo_cats = ttk.Combobox(frame)
+        self.combo_cats.bind("<<ComboboxSelected>>", self.move_chars_to_combobox_cat)
+        self.combo_cats.set("Move to category...")
+        self.combo_cats.grid(row=y.inc(), sticky=tk.EW)
+
         return frame
 
     def initwindow(self) -> None:
-        self.geometry("860x600")
+        self.geometry("500x600")
 
         self.grid_rowconfigure(0, weight=1)
         self.grid_rowconfigure(1, weight=0)
@@ -226,12 +245,15 @@ def gen_listitems_categories(self) -> list[ListboxItem]:
             for category, chars in self.nested_state.items()
         ]
 
-    def load_roa_state(self):
+    def load_gui_from_state(self):
         category_items: list[ListboxItem] = self.gen_listitems_categories()
         self.list_cats.set_items(category_items)
         self.log(f"Loaded {len(category_items)} categories")
 
         self.list_cats.listbox.selection_set(0)
+
+        self.combo_cats.configure(values=[c.label for c in category_items])
+
         self.open_selected_category()
 
     def save_state_to_roas(self, event=None) -> None:
@@ -242,8 +264,10 @@ def save_state_to_roas(self, event=None) -> None:
             if len(char_list) < 1:
                 continue
             new_cat = RoaCategory(len(characters), label.encode('utf-8'))
+            print(new_cat)
             categories_roa.categories.append(new_cat)
             for char in char_list:
+                print(char)
                 try:
                     characters.append(char)
                 except KeyError:
@@ -280,6 +304,11 @@ def open_category(self, category: str) -> None:
         self.list_chars.set_items(group_items)
         self.log(f"Loaded {len(group_items)} chars from group {category!r}")
 
+        # Update UI in case category was opened programatically
+        self.list_cats.listbox.select_clear(0, tk.END)
+        cat_index = [i.value for i in self.list_cats.items].index(category)
+        self.list_cats.listbox.selection_set(cat_index)
+
     def fac_sort_chars_by(self, key_fn) -> Callable[..., None]:
         def do_sort(event=None):
             category = self.get_selected_category()
@@ -323,6 +352,59 @@ def open_info(self, event=None):
         url = f"steam://openurl/https://steamcommunity.com/sharedfiles/filedetails/?id={char.id}"
         webbrowser.open(url, autoraise=True)
 
+    def move_char_to_category(self, src_cat: str, dest_cat: str, char: RoaEntry):
+        self.log(f"Moving {char} from {src_cat} to {dest_cat}")
+        self.nested_state[src_cat].remove(char)
+        self.nested_state[dest_cat].append(char)
+
+        self.load_gui_from_state()
+
+    def interactive_rename_category(self):
+        cur_cat = self.get_selected_category()
+        new_name = askstring(title=None, prompt=f"New name for {cur_cat!r}")
+        if new_name:
+            self.rename_category(cur_cat, new_name)
+            self.open_category(new_name)
+
+    def rename_category(self, cat: str, new_name: str):
+        tups = list(self.nested_state.items())
+        for i, t in enumerate(tups):
+            label, charlist = t
+            if label == cat:
+                tups[i] = (new_name, charlist)
+                break
+            else:
+                self.log(f"Couldn't find {cat} in {tups}")
+        self.nested_state = OrderedDict(tups)
+
+        self.load_gui_from_state()
+
+    def delete_category(self):
+        cat_name = self.get_selected_category()
+        if cat_name and len(self.nested_state[cat_name]) == 0:
+            self.nested_state.pop(cat_name)
+            self.load_gui_from_state()
+        else:
+            self.log("Can only remove empty categories")
+
+    def add_category(self):
+        new_name = askstring(title=None, prompt=f"Name for new category")
+        if new_name and new_name not in self.nested_state.keys():
+            self.nested_state[new_name] = []
+            self.load_gui_from_state()
+            self.open_category(new_name)
+
+    def move_chars_to_combobox_cat(self, event=None):
+        self.log(event)
+        src_cat = self.get_selected_category()
+        dest_cat_label = self.combo_cats.get()
+        dest_cat = {c.label: c.value for c in self.gen_listitems_categories()}[dest_cat_label]
+        for sel_index in self.list_chars.listbox.curselection():
+            char = self.list_chars.items[sel_index].value
+            self.move_char_to_category(src_cat, dest_cat, char)
+
+        self.open_category(src_cat)
+        self.combo_cats.set("Move to category...")
 
 if __name__ == '__main__':
     order_roa = RoaOrderFile(ROA_DIR / 'order.roa')

yaml_sync.py

@@ -14,7 +14,7 @@
 
 def sort_name(entry: RoaEntry):
     try:
-        return entry.name.upper()
+        return entry.name.upper() # type: ignore
     except:
         return 'ERROR'