added files

modules/example.py

@@ -0,0 +1,186 @@
+import os
+import re as regex
+import codecs
+import shutil
+import tkinter
+from tkinter import Tk, filedialog, messagebox, ttk
+
+path = os.getcwd()
+png_list = []
+png_count = 0
+kks_card_list = []
+kks_folder = "_KKS_card_"
+kks_folder2 = "_KKS_to_KK_"
+do_convert_default = True
+
+
+def get_list(folder_path):
+    new_list = []
+    for (_, _, files) in os.walk(folder_path):
+        for filename in files:
+            if regex.match(r".*(\.png)$", filename):
+                new_list.append(filename)
+        return new_list
+
+
+def check_png(card_path):
+    with codecs.open(card_path, "rb") as card:
+        data = card.read()
+        card_type = 0
+        if data.find(b"KoiKatuChara") != -1:
+            card_type = 1
+            if data.find(b"KoiKatuCharaSP") != -1:
+                card_type = 2
+            elif data.find(b"KoiKatuCharaSun") != -1:
+                card_type = 3
+        print(f"[{card_type}] {card_path}")
+    return card_type
+
+
+def do_convert_check_event():
+    print(f"convert = {do_convert.get()}")
+
+
+def convert_kk(card_name, card_path, destination_path):
+    with codecs.open(card_path, mode="rb") as card:
+        data = card.read()
+
+        replace_list = [
+            [b"\x15\xe3\x80\x90KoiKatuCharaSun", b"\x12\xe3\x80\x90KoiKatuChara"],
+            [b"Parameter\xa7version\xa50.0.6", b"Parameter\xa7version\xa50.0.5"],
+            [b"version\xa50.0.6\xa3sex", b"version\xa50.0.5\xa3sex"],
+        ]
+
+        for text in replace_list:
+            data = data.replace(text[0], text[1])
+
+        new_file_path = os.path.normpath(os.path.join(destination_path, f"KKS2KK_{card_name}"))
+        # print(f"new_file_path {new_file_path}")
+
+        with codecs.open(new_file_path, "wb") as new_card:
+            new_card.write(data)
+
+
+def c_get_list():
+    global path, png_list, png_count, png_count_t
+    b_sel['state'] = 'disabled'
+    path = filedialog.askdirectory(title="Select target path (folder contain cards)", mustexist=True)
+
+    if path:
+        print(f"path: {path}")
+        os.chdir(path)
+    else:
+        print("no path")
+        b_sel['state'] = 'normal'
+        return
+
+    png_list = get_list(path)
+    png_count = len(png_list)
+    png_count_t.set(f"png found: {png_count}")
+
+    if png_count > 0:
+        b_p['state'] = 'normal'
+    else:
+        b_p['state'] = 'disabled'
+
+    b_sel['state'] = 'normal'
+
+
+def process_png():
+    global path, png_list, kks_folder, kks_folder2, kks_card_list, window, png_count, png_count_t
+
+    count = len(png_list)
+    if count > 0:
+        bar = ttk.Progressbar(window, maximum=count, length=250)
+        bar.pack(pady=10)
+        bar_val = 0
+        print("0: unknown / 1: kk / 2: kksp / 3: kks")
+        for png in png_list:
+            if check_png(png) == 3:
+                kks_card_list.append(png)
+            bar_val = bar_val + 1
+            bar['value'] = bar_val
+            window.update()
+        bar.destroy()
+    else:
+        messagebox.showinfo("Done", f"no PNG found")
+        return
+
+    count = len(kks_card_list)
+    if count > 0:
+        print(kks_card_list)
+
+        target_folder = os.path.normpath(os.path.join(path, kks_folder))
+        target_folder2 = os.path.normpath(os.path.join(path, kks_folder2))
+        if not os.path.isdir(target_folder):
+            os.mkdir(kks_folder)
+
+        is_convert = do_convert.get()
+        if is_convert:
+            print(f"do convert is [{is_convert}]")
+            if not os.path.isdir(target_folder2):
+                os.mkdir(target_folder2)
+
+        for card in kks_card_list:
+            source = os.path.normpath(os.path.join(path, card))
+            target = os.path.normpath(os.path.join(target_folder, card))
+
+            # copy & convert before move
+            if is_convert:
+                convert_kk(card, source, target_folder2)
+
+            # move file
+            shutil.move(source, target)
+
+        if is_convert:
+            messagebox.showinfo("Done", f"[{count}] cards\nmove to [{kks_folder}] folder\nconvert and save to [{kks_folder2}] folder")
+        else:
+            messagebox.showinfo("Done", f"[{count}] cards\nmove to [{kks_folder}] folder")
+    else:
+        messagebox.showinfo("Done", f"no KKS card found")
+
+    # reset
+    png_list = []
+    kks_card_list = []
+    png_count = 0
+    png_count_t.set(f"png found: 0")
+    b_p['state'] = 'disabled'
+
+
+window = Tk()
+# window.withdraw()
+window.title("KKSCF")
+w = 300
+h = 350
+ltx = int((window.winfo_screenwidth() - w) / 2)
+lty = int((window.winfo_screenheight() - h) / 2)
+window.geometry(f'{w}x{h}+{ltx}+{lty}')
+
+tkinter.Label(window, text=" ", pady=5).pack()
+
+b_sel = tkinter.Button(window, text="Select folder contain cards", padx=10, pady=10, relief="raised", bd=3, command=c_get_list)
+b_sel.pack()
+
+tkinter.Label(window, text=" ", pady=5).pack()
+
+png_count_t = tkinter.StringVar()
+png_count_t.set(f"png found: {png_count}")
+tkinter.Label(window, textvariable=png_count_t, pady=10).pack()
+
+tkinter.Label(window, text=" ", pady=5).pack()
+
+do_convert = tkinter.BooleanVar()
+do_convert.set(do_convert_default)
+do_convert_check = tkinter.Checkbutton(window, text='copy & convert KKS to KK', variable=do_convert, command=do_convert_check_event)
+do_convert_check.pack()
+
+tkinter.Label(window, text=" ", pady=5).pack()
+
+b_p = tkinter.Button(window, text="Process", relief="raised", padx=10, pady=10, bd=3, command=process_png)
+b_p.pack()
+b_p['state'] = 'disabled'
+
+tkinter.Label(window, text=" ", pady=5).pack()
+
+window.mainloop()
+exit(0)