libs.utils.textutil

  1"""
  2libs/utils/textutil.py
  3"""
  4
  5import os
  6import unicodedata
  7from math import ceil, floor
  8from typing import TYPE_CHECKING, Literal
  9
 10import libs.global_value as g
 11
 12if TYPE_CHECKING:
 13    from pathlib import Path
 14
 15
 16def len_count(text: str) -> int:
 17    """文字数をカウント(全角文字は2)
 18
 19    Args:
 20        text (str): 判定文字列
 21
 22    Returns:
 23        int: 文字数
 24    """
 25
 26    count = 0
 27    for c in text:
 28        if unicodedata.east_asian_width(c) in "FWA":
 29            count += 2
 30        else:
 31            count += 1
 32
 33    return count
 34
 35
 36def str_conv(text: str, kind: Literal["h2z", "z2h", "h2k", "k2h"]) -> str:
 37    """文字列変換
 38
 39    Args:
 40        text (str): 変換対象文字列
 41        kind (str): 変換種類
 42        - h2z: 半角文字を全角文字に変換(数字のみ)
 43        - z2h: 全角文字を半角文字に変換(数字のみ)
 44        - h2k: ひらがなをカタカナに変換
 45        - k2h: カタカナをひらがなに変換
 46
 47    Returns:
 48        str: 変換後の文字列
 49    """
 50
 51    zen = "".join(chr(0xff10 + i) for i in range(10))
 52    han = "".join(chr(0x30 + i) for i in range(10))
 53    hira = "".join(chr(0x3041 + i) for i in range(86))
 54    kana = "".join(chr(0x30a1 + i) for i in range(86))
 55
 56    match kind:
 57        case "h2z":  # 半角文字を全角文字に変換(数字のみ)
 58            trans_table = str.maketrans(han, zen)
 59        case "z2h":  # 全角文字を半角文字に変換(数字のみ)
 60            trans_table = str.maketrans(zen, han)
 61        case "h2k":  # ひらがなをカタカナに変換
 62            trans_table = str.maketrans(hira, kana)
 63        case "k2h":  # カタカナをひらがなに変換
 64            trans_table = str.maketrans(kana, hira)
 65        case _:
 66            return text
 67
 68    return text.translate(trans_table)
 69
 70
 71def count_padding(data):
 72    """プレイヤー名一覧の中の最も長い名前の文字数を返す
 73
 74    Args:
 75        data (list, dict): 対象プレイヤー名の一覧
 76
 77    Returns:
 78        int: 文字数
 79    """
 80
 81    name_list = []
 82
 83    if isinstance(data, list):
 84        name_list = data
 85
 86    if isinstance(data, dict):
 87        for i in data.keys():
 88            for name in [data[i][x]["name"] for x in ("東家", "南家", "西家", "北家")]:
 89                if name not in name_list:
 90                    name_list.append(name)
 91
 92    if name_list:
 93        return max(len_count(x) for x in name_list)
 94    return 0
 95
 96
 97def save_file_path(filename: str, delete: bool = False) -> "Path":
 98    """保存ファイルのフルパスを取得
 99
100    Args:
101        filename (str): デフォルトファイル名
102        delete (bool, optional): 生成済みファイルを削除. Defaults to False.
103
104    Returns:
105        Path: 保存ファイルパス
106    """
107
108    _, file_ext = os.path.splitext(filename)
109    file_name = f"{g.params["filename"]}{file_ext}" if g.params.get("filename") else f"{filename}"
110    file_path = g.cfg.setting.work_dir / file_name
111
112    if file_path.exists() and delete:
113        os.remove(file_path)
114
115    return file_path
116
117
118def split_balanced(data: list, target_size: int, tolerance: float = 0.15) -> list:
119    """リストデータを指定個数で分割
120
121    Args:
122        data (list): 対象データ
123        target_size (int): 分割サイズ
124        tolerance (float, optional): 個数誤差. Defaults to 0.15.
125
126    Returns:
127        list: 分割したリスト
128    """
129
130    # 分割サイズに0が指定されている場合は何もしない
131    if not target_size:
132        return data
133
134    n = len(data)
135    if n == 0:
136        return []
137
138    min_size = int(target_size * (1 - tolerance))
139    max_size = int(target_size * (1 + tolerance))
140
141    # 最小ブロック数の候補を計算
142    min_blocks = ceil(n / max_size)
143    max_blocks = floor(n / min_size)
144
145    # 許容範囲内でブロック数を決める（なるべく少ない）
146    for num_blocks in range(min_blocks, max_blocks + 1):
147        size = n / num_blocks
148        if min_size <= size <= max_size:
149            break
150    else:
151        # 条件を満たすブロック数がない場合は単純均等割り
152        num_blocks = ceil(n / target_size)
153
154    # 実際の分割処理
155    base_size = n // num_blocks
156    remainder = n % num_blocks
157
158    result: list = []
159    start = 0
160    for i in range(num_blocks):
161        end = start + base_size + (1 if i < remainder else 0)
162        result.append(data[start:end])
163        start = end
164
165    return result
166
167
168def split_text_blocks(text: str, limit: int = 2000) -> list[str]:
169    """指定文字数でテキストを行単位で分割してリストにする
170
171    Args:
172        text (str): 対象文字列
173        limit (int, optional): 分割文字数. Defaults to 2000.
174
175    Returns:
176        list[str]: 分割リスト
177    """
178
179    blocks = []
180    current_data = ""
181    buffer_data = ""
182    in_code = False
183    min_gap_after_code_start = 10
184    lines_count = 0
185
186    for _, line in enumerate(text.splitlines(keepends=True)):
187        stripped = line.strip()
188        buffer_data += line
189
190        # --- コードブロック開始／終了検出 ---
191        if stripped.startswith("```"):
192            in_code = not in_code
193            if not in_code:
194                current_data += buffer_data
195                buffer_data = ""
196            continue
197
198        lines_count += 1 if in_code else 0
199
200        # --- 文字数チェック ---
201        if len(current_data + buffer_data) > limit:
202            if lines_count > min_gap_after_code_start:
203                if in_code:
204                    blocks.append(current_data + buffer_data + "```\n")
205                    buffer_data = "```\n"
206                else:
207                    blocks.append(current_data + buffer_data)
208                    buffer_data = ""
209            else:
210                blocks.append(current_data)  # 先頭の改行は削除されてしまう
211            current_data = ""
212
213    return blocks