libs.registry.team

  1"""
  2libs/registry/team.py
  3"""
  4
  5import logging
  6
  7import libs.global_value as g
  8from libs.data import initialization, lookup, modify
  9from libs.utils import dbutil, formatter, textutil, validator
 10
 11
 12def create(argument: list) -> str:
 13    """チーム作成
 14
 15    Args:
 16        argument (list): 作成するチーム名
 17
 18    Returns:
 19        str: 処理結果
 20    """
 21
 22    ret = False
 23    msg = "使い方が間違っています。"
 24
 25    if len(argument) == 1:  # 新規追加
 26        team_name = textutil.str_conv(argument[0], "h2z")
 27        if len(g.team_list) > g.cfg.team.registration_limit:
 28            msg = "登録上限を超えています。"
 29        else:  # 登録処理
 30            ret, msg = validator.check_namepattern(team_name, "team")
 31            if ret:
 32                resultdb = dbutil.connection(g.cfg.setting.database_file)
 33                resultdb.execute(
 34                    "insert into team(name) values (?)",
 35                    (team_name,)
 36                )
 37                resultdb.commit()
 38                resultdb.close()
 39                g.team_list = lookup.db.get_team_list()
 40                msg = f"チーム「{team_name}」を登録しました。"
 41                logging.info("add new team: %s", team_name)
 42
 43    return msg
 44
 45
 46def delete(argument: list) -> str:
 47    """チーム削除
 48
 49    Args:
 50        argument (list): 削除するチーム名
 51
 52    Returns:
 53        str: 処理結果
 54    """
 55
 56    msg = "使い方が間違っています。"
 57
 58    if len(argument) == 1:  # 新規追加
 59        team_name = textutil.str_conv(argument[0], "h2z")
 60        if team_name not in [x["team"] for x in g.team_list]:  # 未登録チームチェック
 61            msg = f"チーム「{team_name}」は登録されていません。"
 62        else:
 63            msg = modify.db_backup()
 64            team_id = [x["id"] for x in g.team_list if x["team"] == team_name][0]
 65            resultdb = dbutil.connection(g.cfg.setting.database_file)
 66            resultdb.execute(
 67                "delete from team where id = ?",
 68                (team_id,)
 69            )
 70            resultdb.execute(
 71                "update member set team_id = null where team_id = ?",
 72                (team_id,)
 73            )
 74            resultdb.commit()
 75            resultdb.close()
 76            g.team_list = lookup.db.get_team_list()
 77            msg += f"\nチーム「{team_name}」を削除しました。"
 78            logging.info("team delete: %s", team_name)
 79
 80    return msg
 81
 82
 83def append(argument: list) -> str:
 84    """チーム所属
 85
 86    Args:
 87        argument (list): 登録情報
 88            - argument[0]: 所属させるチーム名
 89            - argument[1]: 所属するメンバー名
 90
 91    Returns:
 92        str: 処理結果
 93    """
 94
 95    msg = "使い方が間違っています。"
 96
 97    if len(argument) == 1:  # 新規作成
 98        msg = create(argument)
 99
100    if len(argument) == 2:  # チーム所属
101        g.params.update(unregistered_replace=False)
102
103        team_name = textutil.str_conv(argument[0], "h2z")
104        player_name = formatter.name_replace(argument[1])
105        registration_flg = True
106        team_id = None
107
108        if team_name not in [x["team"] for x in g.team_list]:  # 未登録チームチェック
109            msg = f"チーム「{team_name}」はまだ登録されていません。"
110            registration_flg = False
111        else:
112            team_id = [x["id"] for x in g.team_list if x["team"] == team_name][0]
113
114        if player_name not in g.member_list:  # 未登録プレイヤーチェック
115            msg = f"「{player_name}」はレギュラーメンバーではありません。"
116            registration_flg = False
117
118        # 登録上限を超えていないか？
119        # select count() from member where team_id=? group by team_id;
120        # rows = resultdb.execute("select count() from team where name=?", (team_name,))
121        # count = rows.fetchone()[0]
122        # if count > g.cfg.team.member_limit:
123        #    msg = f"登録上限を超えています。"
124        #    registration_flg = False
125
126        if registration_flg and team_id:  # 登録処理
127            resultdb = dbutil.connection(g.cfg.setting.database_file)
128            resultdb.execute(
129                "update member set team_id = ? where name = ?",
130                (team_id, player_name)
131            )
132            resultdb.commit()
133            resultdb.close()
134            g.team_list = lookup.db.get_team_list()
135            msg = f"チーム「{team_name}」に「{player_name}」を所属させました。"
136            logging.info("team participation: %s -> %s", team_name, player_name)
137
138    return msg
139
140
141def remove(argument: list) -> str:
142    """チームから除名
143
144    Args:
145        argument (list): 登録情報
146            - argument[0]: 対象チーム名
147            - argument[1]: チームから離脱するメンバー名
148
149    Returns:
150        str: 処理結果
151    """
152
153    msg = "使い方が間違っています。"
154
155    resultdb = dbutil.connection(g.cfg.setting.database_file)
156
157    if len(argument) == 1:
158        msg = delete(argument)
159
160    if len(argument) == 2:  # チーム名指
161        g.params.update(unregistered_replace=False)
162        team_name = textutil.str_conv(argument[0], "h2z")
163        player_name = formatter.name_replace(argument[1])
164
165        registration_flg = True
166        team_id = None
167
168        if team_name not in [x["team"] for x in g.team_list]:  # 未登録チームチェック
169            msg = f"チーム「{team_name}」は登録されていません。"
170            registration_flg = False
171        else:
172            team_id = [x["id"] for x in g.team_list if x["team"] == team_name][0]
173
174        if player_name not in g.member_list:  # 未登録プレイヤーチェック
175            msg = f"「{player_name}」はレギュラーメンバーではありません。"
176            registration_flg = False
177
178        if registration_flg and team_id:  # 登録処理
179            resultdb = dbutil.connection(g.cfg.setting.database_file)
180            resultdb.execute(
181                "update member set team_id = null where name = ?",
182                (player_name,)
183            )
184            resultdb.commit()
185            resultdb.close()
186            g.team_list = lookup.db.get_team_list()
187            msg = f"チーム「{team_name}」から「{player_name}」を離脱させました。"
188            logging.info("team breakaway: %s -> %s", team_name, player_name)
189
190    return msg
191
192
193def clear() -> str:
194    """全チーム削除
195
196    Returns:
197        str: 処理結果
198    """
199
200    msg = modify.db_backup()
201
202    resultdb = dbutil.connection(g.cfg.setting.database_file)
203    resultdb.execute("update member set team_id = null;")
204    resultdb.execute("drop table team;")
205    resultdb.execute("delete from sqlite_sequence where name = 'team';")
206    resultdb.commit()
207    resultdb.close()
208
209    initialization.initialization_resultdb()
210    g.member_list = lookup.db.get_member_list()
211    g.team_list = lookup.db.get_team_list()
212
213    return msg