libs.configuration

  1"""
  2libs/configuration.py
  3"""
  4
  5import argparse
  6import logging
  7import os
  8import sys
  9from functools import partial
 10from typing import TYPE_CHECKING, cast
 11
 12import libs.commands.graph.entry
 13import libs.commands.ranking.entry
 14import libs.commands.report.entry
 15import libs.commands.results.entry
 16import libs.global_value as g
 17from cls.config import AppConfig
 18from integrations import factory
 19from libs.data import lookup
 20from libs.functions import compose
 21from libs.registry import member, team
 22from libs.types import StyleOptions
 23
 24if TYPE_CHECKING:
 25    from cls.config import SubCommand
 26    from integrations.protocols import MessageParserProtocol
 27
 28
 29def set_loglevel():
 30    """ログレベル追加"""
 31
 32    # DEBUG : 10
 33    # INFO : 20
 34    # WARNING : 30
 35    # ERROR : 40
 36    # CRITICAL : 50
 37
 38    # TRACE
 39    logging.TRACE = 5  # type: ignore
 40    logging.trace = partial(logging.log, logging.TRACE)  # type: ignore
 41    logging.addLevelName(logging.TRACE, "TRACE")  # type: ignore
 42
 43
 44def arg_parser() -> argparse.Namespace:
 45    """コマンドライン解析
 46
 47    Returns:
 48        argparse.Namespace: オブジェクト
 49    """
 50
 51    p = argparse.ArgumentParser(
 52        formatter_class=argparse.RawTextHelpFormatter,
 53        add_help=True,
 54    )
 55
 56    p.add_argument(
 57        "-c", "--config",
 58        default="config.ini",
 59        help="設定ファイル(default: %(default)s)",
 60    )
 61    p.add_argument(
 62        "--profile",
 63        help=argparse.SUPPRESS,
 64    )
 65    p.add_argument(
 66        "--service",
 67        choices=[
 68            "slack",
 69            "discord",
 70            "standard_io", "std",
 71            "web", "flask",
 72        ],
 73        default="slack",
 74        help="連携先サービス",
 75    )
 76
 77    logging_group = p.add_argument_group("logging options")
 78    logging_group.add_argument(
 79        "--debug",
 80        action="store_true",
 81        help="デバッグ情報表示",
 82    )
 83    logging_group.add_argument(
 84        "--verbose", "--trace",
 85        dest="verbose",
 86        action="store_true",
 87        help="詳細デバッグ情報表示",
 88    )
 89    logging_group.add_argument(
 90        "--moderate",
 91        action="store_true",
 92        help="ログレベルがエラー以下のもを非表示",
 93    )
 94    logging_group.add_argument(
 95        "--notime",
 96        action="store_true",
 97        help="ログフォーマットから日時を削除",
 98    )
 99
100    match os.path.basename(sys.argv[0]):
101        case "app.py":
102            service_stdio = p.add_argument_group("Only allowed when --service=standard_io")
103            service_stdio.add_argument(
104                "--text",
105                type=str,
106                help="input text strings",
107            )
108            service_web = p.add_argument_group("Only allowed when --service=web")
109            service_web.add_argument(
110                "--host",
111                type=str,
112                default="127.0.0.1",
113                help="listen  address(default: %(default)s)",
114            )
115            service_web.add_argument(
116                "--port",
117                type=int,
118                default=8000,
119                help="bind port(default: %(default)s)",
120            )
121        case "dbtools.py":  # dbtools専用オプション
122            required = p.add_argument_group("Required options(amutually exclusive)")
123            exclusive = required.add_mutually_exclusive_group()
124            exclusive.add_argument(
125                "--compar",
126                action="store_true",
127                help="データ突合",
128            )
129            exclusive.add_argument(
130                "--unification",
131                nargs="?",
132                const="rename.ini",
133                help="ファイルの内容に従って記録済みのメンバー名を修正する(default: %(const)s)",
134            )
135            exclusive.add_argument(
136                "--recalculation",
137                action="store_true",
138                help="ポイント再計算",
139            )
140            exclusive.add_argument(
141                "--export",
142                dest="export_data",
143                nargs="?",
144                const="export",
145                metavar="PREFIX",
146                help="メンバー設定情報をエクスポート(default prefix: %(const)s)",
147            )
148            exclusive.add_argument(
149                "--import",
150                nargs="?",
151                dest="import_data",
152                const="export",
153                metavar="PREFIX",
154                help="メンバー設定情報をインポート(default prefix: %(const)s)",
155            )
156            exclusive.add_argument(
157                "--vacuum",
158                action="store_true",
159                help="database vacuum",
160            )
161            exclusive.add_argument(
162                "--gen-test-data",
163                type=int,
164                dest="gen_test_data",
165                nargs="?",
166                const=1,
167                default=None,
168                metavar="count",
169                help="テスト用サンプルデータ生成(count=生成回数, default: %(const)s)",
170            )
171        case "test.py":  # 動作テスト用オプション
172            p.add_argument(
173                "-t", "--testcase",
174                dest="testcase",
175            )
176
177    return p.parse_args()
178
179
180def setup():
181    """設定ファイル読み込み"""
182
183    set_loglevel()
184
185    g.args = arg_parser()
186
187    # 連携サービス
188    match g.args.service:
189        case "slack":
190            g.selected_service = "slack"
191        case "discord":
192            g.selected_service = "discord"
193        case "standard_io" | "std":
194            g.selected_service = "standard_io"
195        case "web" | "flask":
196            g.selected_service = "web"
197        case _:
198            sys.exit()
199
200    if not hasattr(g.args, "testcase"):
201        g.args.testcase = False
202    else:
203        g.selected_service = "standard_io"
204
205    # ログフォーマット
206    if g.args.notime:
207        fmt = ""
208    else:
209        fmt = "[%(asctime)s]"
210    if g.args.debug or g.args.verbose:
211        fmt += "[%(levelname)s][%(name)s:%(module)s:%(funcName)s] %(message)s"
212    else:
213        fmt += "[%(levelname)s][%(module)s:%(funcName)s] %(message)s"
214
215    if g.args.debug:
216        if g.args.verbose:
217            logging.basicConfig(level=logging.TRACE, format=fmt)  # type: ignore
218            logging.info("DEBUG MODE(verbose)")
219        else:
220            logging.basicConfig(level=logging.DEBUG, format=fmt)
221            logging.info("DEBUG MODE")
222    else:
223        if g.args.moderate:
224            logging.basicConfig(level=logging.WARNING, format=fmt)
225        else:
226            logging.basicConfig(level=logging.INFO, format=fmt)
227
228    g.cfg = AppConfig(g.args.config)
229    g.adapter = factory.select_adapter(g.selected_service, g.cfg)
230    register()
231
232    # 設定内容のロギング
233    logging.info("conf: %s", os.path.join(g.cfg.config_dir, g.args.config))
234    logging.info("font: %s", g.cfg.setting.font_file)
235    logging.info("database: %s", g.cfg.setting.database_file)
236    logging.info("service: %s, graph_library: %s", g.selected_service, g.adapter.conf.plotting_backend)
237    logging.info(
238        "rule_version: %s, origin_point: %s, return_point: %s, time_adjust: %sh",
239        g.cfg.mahjong.rule_version, g.cfg.mahjong.origin_point, g.cfg.mahjong.return_point, g.cfg.setting.time_adjust
240    )
241
242
243def read_memberslist(log=True):
244    """メンバー/チームリスト読み込み
245
246    Args:
247        log (bool, optional): 読み込み時に内容をログに出力する. Defaults to True.
248    """
249
250    g.cfg.member.guest_name = lookup.db.get_guest()
251    g.member_list = lookup.db.get_member_list()
252    g.team_list = lookup.db.get_team_list()
253
254    if log:
255        logging.info("guest_name: %s", g.cfg.member.guest_name)
256        logging.info("member_list: %s", sorted(set(g.member_list.values())))
257        logging.info("team_list: %s", [x["team"] for x in g.team_list])
258
259
260def register():
261    """ディスパッチテーブル登録"""
262
263    def dispatch_help(m: "MessageParserProtocol"):
264        m.set_data("ヘルプ", compose.msg_help.event_message(), StyleOptions())
265        m.post.ts = m.data.event_ts
266
267    def dispatch_download(m: "MessageParserProtocol"):
268        m.set_data("成績記録DB", g.cfg.setting.database_file, StyleOptions())
269
270    def dispatch_members_list(m: "MessageParserProtocol"):
271        m.set_data("登録済みメンバー", lookup.textdata.get_members_list(), StyleOptions(codeblock=True))
272        m.post.ts = m.data.event_ts
273
274    def dispatch_team_list(m: "MessageParserProtocol"):
275        m.set_data("登録済みチーム", lookup.textdata.get_team_list(), StyleOptions(codeblock=True))
276        m.post.ts = m.data.event_ts
277
278    def dispatch_member_append(m: "MessageParserProtocol"):
279        m.set_data("メンバー追加", member.append(m.argument), StyleOptions(key_title=False))
280
281    def dispatch_member_remove(m: "MessageParserProtocol"):
282        m.set_data("メンバー削除", member.remove(m.argument), StyleOptions(key_title=False))
283
284    def dispatch_team_create(m: "MessageParserProtocol"):
285        m.set_data("チーム作成", team.create(m.argument), StyleOptions(key_title=False))
286
287    def dispatch_team_delete(m: "MessageParserProtocol"):
288        m.set_data("チーム削除", team.delete(m.argument), StyleOptions(key_title=False))
289
290    def dispatch_team_append(m: "MessageParserProtocol"):
291        m.set_data("チーム所属", team.append(m.argument), StyleOptions(key_title=False))
292
293    def dispatch_team_remove(m: "MessageParserProtocol"):
294        m.set_data("チーム脱退", team.remove(m.argument), StyleOptions(key_title=False))
295
296    def dispatch_team_clear(m: "MessageParserProtocol"):
297        m.set_data("全チーム削除", team.clear(), StyleOptions(key_title=False))
298
299    dispatch_table: dict = {
300        "results": libs.commands.results.entry.main,
301        "graph": libs.commands.graph.entry.main,
302        "ranking": libs.commands.ranking.entry.main,
303        "report": libs.commands.report.entry.main,
304        "member": dispatch_members_list,
305        "team_list": dispatch_team_list,
306        "download": dispatch_download,
307        "add": dispatch_member_append,
308        "delete": dispatch_member_remove,
309        "team_create": dispatch_team_create,
310        "team_del": dispatch_team_delete,
311        "team_add": dispatch_team_append,
312        "team_remove": dispatch_team_remove,
313        "team_clear": dispatch_team_clear,
314    }
315
316    # ヘルプ
317    g.keyword_dispatcher.update({g.cfg.setting.help: dispatch_help})
318
319    for command, ep in dispatch_table.items():
320        # 呼び出しキーワード登録
321        if hasattr(g.cfg, command):
322            sub_command = cast("SubCommand", getattr(g.cfg, command))
323            for alias in sub_command.commandword:
324                g.keyword_dispatcher.update({alias: ep})
325        # スラッシュコマンド登録
326        for alias in cast(list, getattr(g.cfg.alias, command)):
327            g.command_dispatcher.update({alias: ep})
328
329    # サービス別コマンド登録
330    g.command_dispatcher.update(g.adapter.conf.command_dispatcher)
331    g.keyword_dispatcher.update(g.adapter.conf.keyword_dispatcher)
332
333    logging.debug("keyword_dispatcher:\n%s", "\n".join([f"\t{k}: {v}" for k, v in g.keyword_dispatcher.items()]))
334    logging.debug("command_dispatcher:\n%s", "\n".join([f"\t{k}: {v}" for k, v in g.command_dispatcher.items()]))