libs.commands.registry.team

  1"""
  2libs/registry/team.py
  3"""
  4
  5import logging
  6from typing import TYPE_CHECKING, TypedDict, cast
  7
  8import libs.global_value as g
  9from libs.bootstrap import initialization
 10from libs.bootstrap.section import BaseSection
 11from libs.domain import modify
 12from libs.types import CommandType
 13from libs.utils import dbutil, formatter, textutil, validator
 14
 15if TYPE_CHECKING:
 16    from configparser import ConfigParser, SectionProxy
 17
 18    from libs.bootstrap.app_config import AppConfig
 19
 20
 21class TeamDataDict(TypedDict):
 22    """チーム情報格納辞書"""
 23
 24    id: int
 25    """チームID"""
 26    team: str
 27    """チーム名"""
 28    members: list[str]
 29    """所属メンバーリスト"""
 30
 31
 32class TeamSection(BaseSection):
 33    """teamセクション処理"""
 34
 35    default_commandword: str
 36    """コマンドワードデフォルト値"""
 37    commandword: list[str]
 38    """呼び出しキーワード"""
 39    command_suffix: list[str]
 40    """コマンド接尾辞(登録キーワード+接尾辞を呼び出しキーワードとして扱う)"""
 41
 42    section: str
 43    main_parser: "ConfigParser"
 44
 45    info: list[TeamDataDict]
 46    """チーム情報(キャッシュデータ)"""
 47    registration_limit: int
 48    """登録チーム上限数"""
 49    character_limit: int
 50    """チーム名に使用できる文字数"""
 51    member_limit: int
 52    """チームに所属できるメンバー上限"""
 53    friendly_fire: bool
 54    """チームメイトが同卓しているゲームを集計対象に含めるか"""
 55
 56    def __init__(self, outer: "AppConfig") -> None:
 57        self.default_commandword = "チーム一覧"
 58        self.section = str(CommandType.TEAM_LIST)
 59        self.main_parser = outer.main_parser
 60        self._reset()
 61
 62    def _reset(self) -> None:
 63        self.info = []
 64        self.commandword = []
 65        self.command_suffix = []
 66        self.registration_limit = int(255)
 67        self.character_limit = int(16)
 68        self.member_limit = int(16)
 69        self.friendly_fire = bool(True)
 70
 71    def config_load(self, section_proxy: "SectionProxy") -> None:
 72        """
 73        設定値取り込み
 74
 75        Args:
 76            section_proxy (SectionProxy): 読み込み先(パーサー + セクション名)
 77
 78        """
 79        self._reset()
 80        self.initialization(section_proxy)
 81
 82        # 呼び出しキーワード取り込み
 83        self.commandword = self.getlist("commandword", fallback=self.default_commandword)
 84
 85        logging.trace("%s: %s", self.section, self)  # type: ignore
 86
 87    def member(self, team: str) -> list[str]:
 88        """
 89        指定チームの所属メンバーをリストで返す
 90
 91        Args:
 92            team (str): 対象チーム名
 93
 94        Returns:
 95            list[str]: 所属メンバーリスト
 96
 97        """
 98        for x in self.info:
 99            if x.get("team") == team:
100                return x.get("members")
101        return []
102
103    def which(self, name: str) -> str | None:
104        """
105        指定メンバーの所属チームを返す
106
107        Args:
108            name (str): 対象メンバー名
109
110        Returns:
111            Union[str, None]:
112            - str: 所属しているチーム名
113            - None: 未所属
114
115        """
116        for team in self.lists:
117            if name in self.member(team):
118                return team
119
120        return None
121
122    @property
123    def lists(self) -> list[str]:
124        """
125        チーム名一覧をリストで返す
126
127        Returns:
128            list[str]: チーム名一覧
129
130        """
131        return [x.get("team") for x in self.info]
132
133    @property
134    def get_info(self) -> list[TeamDataDict]:
135        """
136        全チーム情報取得
137
138        Returns:
139            list[TeamDataDict]: チーム情報
140
141        """
142        ret = g.params.read_data("TEAM_INFO").to_dict(orient="records")
143        for row in ret:
144            row.update(members=str(row["members"]).split(","))
145
146        return cast(list[TeamDataDict], ret)
147
148
149def create(argument: list[str]) -> str:
150    """
151    チーム作成
152
153    Args:
154        argument (list[str]): 作成するチーム名
155
156    Returns:
157        str: 処理結果
158
159    """
160    ret = False
161    msg = "使い方が間違っています。"
162
163    if len(argument) == 1:  # 新規追加
164        team_name = textutil.str_conv(argument[0], textutil.ConversionType.HtoZ)
165        if len(g.cfg.team.lists) > g.cfg.team.registration_limit:
166            msg = "登録上限を超えています。"
167        else:  # 登録処理
168            ret, msg = validator.check_namepattern(team_name, "team")
169            if ret:
170                resultdb = dbutil.connection(g.cfg.setting.database_file)
171                resultdb.execute(
172                    "insert into team(name) values (?)",
173                    (team_name,),
174                )
175                resultdb.commit()
176                resultdb.close()
177                g.cfg.team.info = g.cfg.team.get_info
178                msg = f"チーム「{team_name}」を登録しました。"
179                logging.info("add new team: %s", team_name)
180
181    return msg
182
183
184def delete(argument: list[str]) -> str:
185    """
186    チーム削除
187
188    Args:
189        argument (list[str]): 削除するチーム名
190
191    Returns:
192        str: 処理結果
193
194    """
195    msg = "使い方が間違っています。"
196
197    if len(argument) == 1:  # 新規追加
198        team_name = textutil.str_conv(argument[0], textutil.ConversionType.HtoZ)
199        if team_name not in g.cfg.team.lists:  # 未登録チームチェック
200            msg = f"チーム「{team_name}」は登録されていません。"
201        else:
202            msg = modify.db_backup()
203            team_id = [x["id"] for x in g.cfg.team.info if x["team"] == team_name][0]
204            resultdb = dbutil.connection(g.cfg.setting.database_file)
205            resultdb.execute("delete from team where id = ?", (team_id,))
206            resultdb.execute(
207                "update member set team_id = null where team_id = ?",
208                (team_id,),
209            )
210            resultdb.commit()
211            resultdb.close()
212            g.cfg.team.info = g.cfg.team.get_info
213            msg += f"\nチーム「{team_name}」を削除しました。"
214            logging.info("team delete: %s", team_name)
215
216    return msg
217
218
219def append(argument: list[str]) -> str:
220    """
221    チーム所属
222
223    Args:
224        argument (list[str]): 登録情報
225            - argument[0]: 所属させるチーム名
226            - argument[1]: 所属するメンバー名
227
228    Returns:
229        str: 処理結果
230
231    """
232    msg = "使い方が間違っています。"
233
234    if len(argument) == 1:  # 新規作成
235        msg = create(argument)
236
237    if len(argument) == 2:  # チーム所属
238        g.params.unregistered_replace = False
239
240        team_name = textutil.str_conv(argument[0], textutil.ConversionType.HtoZ)
241        player_name = formatter.name_replace(argument[1])
242        registration_flg = True
243        team_id = None
244
245        if team_name not in g.cfg.team.lists:  # 未登録チームチェック
246            msg = f"チーム「{team_name}」はまだ登録されていません。"
247            registration_flg = False
248        else:
249            team_id = [x["id"] for x in g.cfg.team.info if x["team"] == team_name][0]
250
251        if player_name not in g.cfg.member.lists:  # 未登録プレイヤーチェック
252            msg = f"「{player_name}」はレギュラーメンバーではありません。"
253            registration_flg = False
254
255        # 登録上限を超えていないか？
256        # select count() from member where team_id=? group by team_id;
257        # rows = resultdb.execute("select count() from team where name=?", (team_name,))
258        # count = rows.fetchone()[0]
259        # if count > g.cfg.team.member_limit:
260        #    msg = f"登録上限を超えています。"
261        #    registration_flg = False
262
263        if registration_flg and team_id:  # 登録処理
264            resultdb = dbutil.connection(g.cfg.setting.database_file)
265            resultdb.execute(
266                "update member set team_id = ? where name = ?",
267                (team_id, player_name),
268            )
269            resultdb.commit()
270            resultdb.close()
271            g.cfg.team.info = g.cfg.team.get_info
272            msg = f"チーム「{team_name}」に「{player_name}」を所属させました。"
273            logging.info("team participation: %s -> %s", team_name, player_name)
274
275    return msg
276
277
278def remove(argument: list[str]) -> str:
279    """
280    チームから除名
281
282    Args:
283        argument (list[str]): 登録情報
284            - argument[0]: 対象チーム名
285            - argument[1]: チームから離脱するメンバー名
286
287    Returns:
288        str: 処理結果
289
290    """
291    msg = "使い方が間違っています。"
292
293    resultdb = dbutil.connection(g.cfg.setting.database_file)
294
295    if len(argument) == 1:
296        msg = delete(argument)
297
298    if len(argument) == 2:  # チーム名指
299        g.params.unregistered_replace = False
300        team_name = textutil.str_conv(argument[0], textutil.ConversionType.HtoZ)
301        player_name = formatter.name_replace(argument[1])
302
303        registration_flg = True
304        team_id = None
305
306        if team_name not in g.cfg.team.lists:  # 未登録チームチェック
307            msg = f"チーム「{team_name}」は登録されていません。"
308            registration_flg = False
309        else:
310            team_id = [x["id"] for x in g.cfg.team.info if x["team"] == team_name][0]
311
312        if player_name not in g.cfg.member.lists:  # 未登録プレイヤーチェック
313            msg = f"「{player_name}」はレギュラーメンバーではありません。"
314            registration_flg = False
315
316        if registration_flg and team_id:  # 登録処理
317            resultdb = dbutil.connection(g.cfg.setting.database_file)
318            resultdb.execute(
319                "update member set team_id = null where name = ?",
320                (player_name,),
321            )
322            resultdb.commit()
323            resultdb.close()
324            g.cfg.team.info = g.cfg.team.get_info
325            msg = f"チーム「{team_name}」から「{player_name}」を離脱させました。"
326            logging.info("team breakaway: %s -> %s", team_name, player_name)
327
328    return msg
329
330
331def clear() -> str:
332    """
333    全チーム削除
334
335    Returns:
336        str: 処理結果
337
338    """
339    msg = modify.db_backup()
340
341    resultdb = dbutil.connection(g.cfg.setting.database_file)
342    resultdb.execute("update member set team_id = null;")
343    resultdb.execute("drop table team;")
344    resultdb.execute("delete from sqlite_sequence where name = 'team';")
345    resultdb.commit()
346    resultdb.close()
347
348    initialization.setup_resultdb(g.cfg.setting.database_file)
349    g.cfg.member.info = g.cfg.member.get_info
350    g.cfg.team.info = g.cfg.team.get_info
351
352    return msg