libs.commands.report.winner

  1"""
  2libs/commands/report/winner.py
  3"""
  4
  5from typing import TYPE_CHECKING, Any
  6
  7import matplotlib.pyplot as plt
  8import pandas as pd
  9
 10import libs.global_value as g
 11from libs.domain.datamodels import GameInfo
 12from libs.functions import message
 13from libs.functions.compose import text_item
 14from libs.types import StyleOptions
 15from libs.utils import formatter, graphutil, textutil
 16
 17if TYPE_CHECKING:
 18    from integrations.protocols import MessageParserProtocol
 19
 20
 21def plot(m: "MessageParserProtocol") -> None:
 22    """
 23    成績上位者を一覧化
 24
 25    Args:
 26        m (MessageParserProtocol): メッセージデータ
 27
 28    """
 29    # --- データ取得
 30    game_info = GameInfo()
 31    results_df = g.params.read_data("REPORT_WINNER")
 32    if len(results_df) == 0:
 33        m.set_headline(message.random_reply(m, "no_hits"), StyleOptions(title="成績上位"))
 34        m.status.result = False
 35        return
 36
 37    # 匿名化
 38    if g.params.anonymous:
 39        name_list: list[Any] = []
 40        for col in [f"name{x}" for x in range(1, 6)]:
 41            name_list.extend(results_df[col].unique().tolist())
 42        mapping_dict = formatter.anonymous_mapping(list(set(name_list)))
 43        for col in [f"name{x}" for x in range(1, 6)]:
 44            results_df[col] = results_df[col].replace(mapping_dict)
 45
 46    # --- 集計
 47    results: dict[str, Any] = {}
 48    for _, v in results_df.iterrows():
 49        results[v["collection"]] = {}
 50        results[v["collection"]]["集計月"] = v["collection"]
 51        for x in range(1, 6):
 52            if v.isna()[f"point{x}"]:
 53                results[v["collection"]][f"{x}位"] = "該当者なし"
 54            else:
 55                results[v["collection"]][f"{x}位"] = "{} ({}pt)".format(
 56                    v[f"name{x}"],
 57                    str("{:+}".format(v[f"point{x}"])).replace("-", "▲"),
 58                )
 59
 60    m.set_headline(message.header(game_info, m), StyleOptions(title="成績上位者"))
 61
 62    # --- グラフ設定
 63    match g.adapter.conf.plotting_backend:
 64        case "plotly":
 65            report_file_path = textutil.save_file_path("report.html")
 66        case _:
 67            graphutil.setup()
 68            report_file_path = textutil.save_file_path("report.png")
 69            plt.rcParams["font.size"] = 6
 70
 71            # 色彩設定
 72            match (plt.rcParams["text.color"], plt.rcParams["figure.facecolor"]):
 73                case text_color, bg_color if text_color == "black" and bg_color == "white":
 74                    line_color1 = "#ffffff"
 75                    line_color2 = "#dddddd"
 76                case text_color, bg_color if text_color == "white" and bg_color == "black":
 77                    line_color1 = "#000000"
 78                    line_color2 = "#111111"
 79                case _:
 80                    line_color1 = plt.rcParams["figure.facecolor"]
 81                    line_color2 = plt.rcParams["figure.facecolor"]
 82
 83            column_labels = list(results[list(results.keys())[0]].keys())
 84            column_color = ["#000080" for i in column_labels]
 85
 86            cell_param = []
 87            cell_color = []
 88            line_count = 0
 89            for _, val in results.items():
 90                line_count += 1
 91                cell_param.append([val[y] for y in column_labels])
 92                if int(line_count % 2):
 93                    cell_color.append([line_color1 for i in column_labels])
 94                else:
 95                    cell_color.append([line_color2 for i in column_labels])
 96
 97            fig = plt.figure(figsize=(6.5, (len(results) * 0.2) + 0.8), dpi=200, tight_layout=True)
 98            ax_dummy = fig.add_subplot(111)
 99            ax_dummy.axis("off")
100
101            plt.title("成績上位者", fontsize=12)
102
103            tb = plt.table(
104                colLabels=column_labels,
105                colColours=column_color,
106                cellText=cell_param,
107                cellColours=cell_color,
108                loc="center",
109            )
110
111            tb.auto_set_font_size(False)
112            for i in range(len(column_labels)):
113                tb[0, i].set_text_props(color="#FFFFFF", weight="bold")
114            for i in range(len(results.keys()) + 1):
115                for j in range(len(column_labels)):
116                    tb[i, j].set_text_props(ha="center")
117
118            # 追加テキスト
119            remark_text = "".join(text_item.remarks(True)) + text_item.search_word(True)
120            add_text = "{} {}".format(
121                f"[検索範囲：{text_item.search_range()}]",
122                f"[{remark_text}]" if remark_text else "",
123            )
124
125            fig.text(
126                0.01,
127                0.02,  # 表示位置(左下0,0 右下0,1)
128                add_text,
129                transform=fig.transFigure,
130                fontsize=6,
131            )
132
133            fig.savefig(report_file_path)
134
135    match g.adapter.interface_type:
136        case "slack" | "discord":
137            m.set_message(report_file_path, StyleOptions(title="成績上位者", use_comment=True, header_hidden=True))
138        case "web":
139            m.set_message(formatter.df_rename(results_df, StyleOptions()), StyleOptions(title="成績上位者"))
140        case _:
141            m.set_message(pd.DataFrame(results).T, StyleOptions())