1#!/usr/bin/python3
 2import os, tkinter, Constants
 3from tkinter.filedialog import askopenfilename
 4from Functions import check_rom, show_warning, start_patching, apply_preset, set_progress
 5from pathlib import Path
 6from pygubu import Builder
 7
 8PROJECT_PATH = Path(__file__).parent
 9PROJECT_UI = PROJECT_PATH / "patcher.ui"
10
11class PatcherApp:
12    def __init__(self, master=None):
13        self.builder = builder = Builder()
14        builder.add_resource_path(PROJECT_PATH)
15        builder.add_from_file(PROJECT_UI)
16        # Main widget
17        self.mainwindow = builder.get_object("main_frame", master)
18        self.apply_button = builder.get_object("apply_button")
19        self.preset_combo = builder.get_object("preset_combo")
20        
21        self.browse_path = None
22        self.preset = None
23        self.progress = None
24        self.status_text = None
25        self.font = None
26        self.places = None
27        self.sprites = None
28        self.nes_palette = None
29        self.skip_m1 = None
30        builder.import_variables(
31            self,
32            [
33                "browse_path",
34                "preset",
35                "progress",
36                "status_text",
37                "font",
38                "places",
39                "sprites",
40                "nes_palette",
41                "skip_m1",
42            ],
43        )
44        
45        for preset in Constants.PRESETS.keys():
46            self.preset_combo['values'] = (*self.preset_combo['values'], preset)
47        self.preset.trace('w', self.on_change_preset)
48        self.preset.set(Constants.DEF_PRESET[0])
49        self.status_text.set(Constants.STATUS_START)
50        
51        baserom = None
52        for file in os.listdir('.'):
53            if file.lower().endswith('.gba'):
54                filename = os.path.join('.', file)
55                baserom = filename if check_rom(filename) else baserom
56        if baserom is not None:
57            self.browse_path.set(baserom)
58        
59        builder.connect_callbacks(self)
60    
61    def run(self):
62        self.mainwindow.mainloop()
63
64    def on_browse_button(self):
65        fn = askopenfilename(filetypes=[(Constants.VAR_FILEPICKER, '*.gba')])
66        self.browse_path.set(fn)
67    
68    def on_change_preset(self, *arg):
69        preset = self.preset.get()
70        if apply_preset(self, preset):
71            set_progress(self, 0, f'{Constants.STATUS_PRESET}: {preset}.')
72
73    def on_apply_button(self):
74        self.apply_button['state'] = 'disabled'
75        
76        baserom_temp = self.browse_path.get()
77        
78        if not check_rom(baserom_temp):
79            show_warning(Constants.WARNING_MD5_MISMATCH)
80            self.apply_button['state'] = 'normal'
81            return
82        self.baserom = baserom_temp
83        start_patching(self)
84        self.apply_button['state'] = 'enabled'
85        set_progress(self, 0, Constants.STATUS_START)
86
87def main():
88    if not os.path.isdir(Constants.PATH_TOOLS):
89        show_warning(Constants.WARNING_EXTRACT)
90        return
91    
92    root = tkinter.Tk()
93    root.title(Constants.VAR_WINDOW_TITLE)
94    root.resizable(False, False)
95    app = PatcherApp(root)
96    app.run()
97
98if __name__ == "__main__":
99    main()