main.py

import os
import re
import sys
import json
import numpy as np
import tkinter as tk
import tkinter.ttk as ttk
from tksheet import Sheet
import subprocess

import tkwidgets
import songdata
import bmstable
import version


# デフォルトの難易度表リスト
TABLE_LIST_DEFAULT = [
    { 'name': 'NEW GENERATION 通常難易度表', 'url': 'https://rattoto10.jounin.jp/table.html' },
    { 'name': 'NEW GENERATION 発狂難易度表', 'url': 'https://rattoto10.jounin.jp/table_insane.html' },
    { 'name': 'Satellite', 'url': 'https://stellabms.xyz/sl/table.html' },
    { 'name': 'LN難易度表', 'url': 'http://flowermaster.web.fc2.com/lrnanido/gla/LN.html' },
]

DEFAULT_TABLE_INDEX = 0


class MainWindow(tk.Tk):
    FONT_UI = ('MS UI Gothic', 9, 'normal')
    FONT_UI_UNDERLINE = ('MS UI Gothic', 9, 'underline')
    FONT_UI_TITLE = ('MS UI Gothic', 12, 'bold', 'underline')
    HEADERS = ['Level', 'Title', 'Artist', 'Found', 'index']

    def __init__(self, table_list):
        tk.Tk.__init__(self)

        title = f'bms-table-view {version.VERSION}'
        self.title(title)

        self.geometry('800x700')
        self.grid_columnconfigure(0, weight=1)
        self.grid_rowconfigure(2, weight=1)
        
        self.table_frame = tk.Frame(self)
        self.table_frame.grid_columnconfigure(0, weight=1)

        self.table_combobox = ttk.Combobox(self.table_frame,
                                           state='readonly',
                                           font=self.FONT_UI,
                                           values=[t.get('name') for t in table_list])
        self.table_combobox.current(DEFAULT_TABLE_INDEX)
        self.table_combobox.bind('<<ComboboxSelected>>', self._on_table_combobox_selected)

        self.sheet = Sheet(self.table_frame,
                           headers=self.HEADERS,
                           font=self.FONT_UI,
                           header_font=self.FONT_UI,
                           index_font=self.FONT_UI,
                           outline_thickness=1,
                           outline_color='gray')
        self.sheet.enable_bindings()
        self.sheet.extra_bindings([('all_select_events', self._sheet_select_event)])
        self.sheet.readonly_columns(list(range(len(self.HEADERS))))
        self.sheet.readonly_header(list(range(len(self.HEADERS))))
        self.sheet.hide_columns(columns=[4])
        self.sheet.align_columns(columns=[3], align='center')

        self.check_only_notfound = tkwidgets.CheckBox(parent=self.table_frame,
                                            text='未所持のみ表示',
                                            command=self._on_check_only_notfound)

        self.info_frame = tk.Frame(self, bd=1, relief=tk.SOLID)

        self.label_title = tkwidgets.ClickableLabel(parent=self.info_frame, font=self.FONT_UI_TITLE)

        self.label_url = tkwidgets.ClickableLabel(parent=self.info_frame, font=self.FONT_UI)

        self.label_urldiff = tkwidgets.ClickableLabel(parent=self.info_frame, font=self.FONT_UI)

        self.label_urlpack = tkwidgets.ClickableLabel(parent=self.info_frame, font=self.FONT_UI)

        self.search_frame = tk.Frame(self)
        self.search_frame.grid_columnconfigure(0, weight=1)
        self.search_frame.grid_rowconfigure(2, weight=1)

        self.textbox_search = tkwidgets.TextBox(parent=self.search_frame, font=self.FONT_UI)
        self.textbox_search.get().bind('<Return>', self._search_songs)

        self.search_option_frame = tk.Frame(self.search_frame)

        self.check_use_re = tkwidgets.CheckBox(parent=self.search_option_frame,
                                               text='正規表現',
                                               command=None)
        self.check_use_re.set_value(False)

        self.check_case_sensitive = tkwidgets.CheckBox(parent=self.search_option_frame,
                                                       text='大文字と小文字を区別',
                                                       command=None)
        self.check_case_sensitive.set_value(False)

        self.treeview_frame = tk.Frame(self.search_frame)
        self.treeview_frame.grid_columnconfigure(0, weight=1)
        self.treeview_frame.grid_rowconfigure(0, weight=1)

        self.treeview = ttk.Treeview(master=self.treeview_frame, columns=('artist', 'diff'), height=5)
        self.treeview.column('#0', width=400, stretch=0)
        self.treeview.column('artist', width=100, stretch=1)
        self.treeview.column('diff', width=100, stretch=1)
        self.treeview.heading('#0', text='Path/Title')
        self.treeview.heading('artist', text='Artist')
        self.treeview.heading('diff', text='Filename')
        self.treeview.bind('<3>', self._on_treeview_rclick)

        self.vscrollbar = ttk.Scrollbar(self.treeview_frame, orient="vertical", command=self.treeview.yview)
        self.treeview.configure(yscrollcommand=self.vscrollbar.set)

        # レイアウト

        # フレーム
        self.table_frame.grid(row=0, column=0, sticky='ew', padx=4, pady=2)
        self.info_frame.grid(row=1, column=0, sticky='ew', padx=4, pady=2)
        self.search_frame.grid(row=2, column=0, sticky=tk.NSEW, padx=4, pady=2)

        # フレーム内 (table_frame)
        self.table_combobox.grid(row=0, column=0, sticky='ew', padx=4, pady=2)
        self.sheet.grid(row=1, column=0, sticky='ew', padx=4, pady=2)
        self.check_only_notfound.get().grid(row=2, column=0, sticky='w')

        # フレーム内 (info_frame)
        self.label_title.get().grid(row=1, column=0, sticky='w', padx=4, pady=2)
        self.label_url.get().grid(row=2, column=0, sticky='w', padx=4, pady=2)
        self.label_urldiff.get().grid(row=3, column=0, sticky='w', padx=4, pady=2)
        self.label_urlpack.get().grid(row=4, column=0, sticky='w', padx=4, pady=2)

        # フレーム内 (search_frame)
        self.textbox_search.get().grid(row=0, column=0, sticky='ew', padx=4, pady=2, ipadx=2, ipady=2)
        self.search_option_frame.grid(row=1, column=0, sticky=tk.NSEW)
        self.treeview_frame.grid(row=2, column=0, sticky=tk.NSEW)
        # フレーム内 (search_option_frame)
        self.check_use_re.get().pack(side=tk.LEFT)
        self.check_case_sensitive.get().pack(side=tk.LEFT)
        # フレーム内 (treeview_frame)
        self.treeview.grid(row=0, column=0, sticky=tk.NSEW, padx=4, pady=2)
        self.vscrollbar.grid(row=0, column=1, sticky=tk.NSEW)

        # ttkウィジェットのスタイル（フォント）指定
        style = ttk.Style()
        style.configure("Treeview.Heading", font=self.FONT_UI)
        style.configure("Treeview", font=self.FONT_UI)
        style.configure("TCheckbutton", font=self.FONT_UI)

    def set_songdata(self, df_songdata):
        self.df_songdata = df_songdata

    def set_table(self, table):
        self.table = table
        self._make_table_for_view()

        # シートに曲リストを表示
        self._update_sheet(show_only_notfound=True)

    def _on_table_combobox_selected(self, event):
        index = self.table_combobox.current()
        self.table.load(index)
        self._make_table_for_view()

        # シートに曲リストを表示
        self._update_sheet(show_only_notfound=self.check_only_notfound.get_value())

    def _make_table_for_view(self):
        """シート表示用の譜面リスト（df_table_view）を作成する。

        - 重複の削除（'index'(難易度表上の並び順)による）
        - 難易度表ヘッダに 'level_order' の指定がある場合は並び替え
        """
        df = self.table.get_table().drop_duplicates(subset='index')

        if 'level_order' in self.table.get_header().keys():
            level_order = {str(x): i for i, x in enumerate(self.table.get_header()['level_order'])}
            df = df.sort_values('level', key=lambda col: col.map(level_order), kind='stable')

        self.df_table_view = df

    def _update_sheet(self, show_only_notfound=False):
        self._clear_sheet()

        df = self.df_table_view
        for level, title, artist, found, index in zip(df['level'], df['title'], df['artist'], df['found'], df['index']):
            found_str = '' if found else '未所持'
            self.sheet.insert_row(values=(f'{self.table.get_header()["symbol"]}{level}', title, artist, found_str, index), idx='end')

        not_found_rows = []
        i_line_notfound = 0
        for found in df['found']:
            if not found:
                not_found_rows += [i_line_notfound]
            i_line_notfound += 1

        self.sheet.dehighlight_all()
        self.sheet.highlight_rows(not_found_rows, fg='blue')

        if show_only_notfound:
            self.sheet.display_rows(not_found_rows, all_rows_displayed=False)

        if len(not_found_rows) > 0:
            self.sheet.select_row(0)
            self.sheet.see(row=0)

    def _clear_sheet(self):
        self.sheet.display_rows(None, all_rows_displayed=True)
        self.sheet.set_sheet_data([])
        self.sheet.column_width(column=0, width=50)
        self.sheet.column_width(column=1, width=400)
        self.sheet.column_width(column=2, width=180)
        self.sheet.column_width(column=3, width=80)

    def _on_check_only_notfound(self):
        self._update_sheet(show_only_notfound=self.check_only_notfound.get_value())

    def _sheet_select_event(self, event=None):
        if event[0] == 'select_cell':
            # セルがクリックされた → 行全体を選択する
            row_idx = event[1]
            self.sheet.select_row(row_idx)

        elif event[0] == 'select_row':
            # 行が選択された
            row_idx = event[1]
            # 表示されている行の番号から元データの行番号を得る（非表示の行がある場合を考慮）
            disp_row_idx = self.sheet.displayed_row_to_data(row_idx)

            # 曲データを取得
            row = self.df_table_view.iloc[disp_row_idx].replace(np.nan, '')
            title = row['title']
            md5 = row['md5']
            ir_url = f'http://www.dream-pro.info/~lavalse/LR2IR/search.cgi?mode=ranking&bmsmd5={md5}' if md5 else ''
            url = row['url'] if 'url' in row.keys() and row['url'] else ''
            url_diff = row['url_diff'] if 'url_diff' in row.keys() and row['url_diff'] else ''
            name_diff = f'- {row["name_diff"]}' if 'name_diff' in row.keys() and row['name_diff'] else ''

            if 'url_pack' in self.df_table_view.columns:
                url_pack = row['url_pack'] if 'url_pack' in row.keys() else ''
                name_pack = f'- {row["name_pack"]}' if 'name_pack' in row.keys() and row['name_pack'] else ''
            else:
                url_pack, name_pack = '', ''

            # 曲データをUIに反映
            self.label_title.set_text(title)
            self.label_title.set_click_event(ir_url)

            self.label_url.set_text(url)
            self.label_url.set_click_event(url)

            self.label_urldiff.set_text(f'{url_diff} {name_diff}')
            self.label_urldiff.set_click_event(url_diff)

            if url_pack:
                self.label_urlpack.set_text(f'{url_pack} {name_pack}')
                self.label_urlpack.set_click_event(url_pack)
            else:
                self.label_urlpack.set_text('')

            self.textbox_search.set_text(re.sub(r'\s*\[.[^\[]+\]$', '', title))

    def _search_songs(self, event):
        # テキストボックスの内容で曲を検索
        search_word = self.textbox_search.get_text()
        use_re = self.check_use_re.get_value()
        case_sensitive = self.check_case_sensitive.get_value()
        df_result = self.df_songdata[self.df_songdata['title_inc_sub'].str.contains(search_word, case=case_sensitive, regex=use_re)]
        
        # 検索結果：譜面格納フォルダのパスと、含まれる差分の一覧
        dirlist = {}
        for title, artist, path in zip(df_result['title_inc_sub'], df_result['artist'], df_result['path']):
            path_dir = os.path.dirname(path)
            path_base = os.path.basename(path)
            diff_data = { 'title':title, 'artist':artist, 'diff':path_base }

            if path_dir in dirlist.keys():
                dirlist[path_dir].append(diff_data)
            else:
                dirlist[path_dir] = [diff_data]
        
        # ツリービューに譜面格納フォルダのパス・差分一覧を表示
        self.treeview.delete(*self.treeview.get_children())
        for path, diff_list in dirlist.items():
            iid = self.treeview.insert(parent='', index='end', text=path, open=True)
            for diff in diff_list:
                self.treeview.insert(parent=iid, index='end', text=diff['title'], values=[diff['artist'], diff['diff']])

    def _on_treeview_rclick(self, event):
        iid = self.treeview.identify_row(y=event.y)
        if self.treeview.parent(iid) != '':
            iid = self.treeview.parent(iid)

        dir_path = self.treeview.item(iid)['text']
        if os.path.isdir(dir_path):
            subprocess.Popen(['explorer', dir_path], shell=True)


def main():
    # スクリプトのあるフォルダに移動
    os.chdir(os.path.dirname(sys.argv[0]))

    # songdata.db のパスを読み込み
    config = json.load(open('config.json', 'r', encoding='utf-8'))

    # songdata.db 読み込み
    df_songdata = songdata.read_songdata(config['SONGDATA_DB_PATH'])

    # 難易度表リスト読み込み
    # 設定ファイルに記述がなければデフォルトのリストを使用
    table_list = []
    if 'TABLE_LIST' in config.keys():
        table_list = config['TABLE_LIST']

    if len(table_list) == 0:
        table_list = TABLE_LIST_DEFAULT

    # 難易度表読み込み
    # キャッシュがなければダウンロードする
    table = bmstable.BmsTable(table_list, df_songdata)
    table.load(DEFAULT_TABLE_INDEX)

    # GUI表示
    main_window = MainWindow(table_list)
    main_window.set_songdata(df_songdata)
    main_window.set_table(table)
    main_window.mainloop()


if __name__ == '__main__':
    main()