Участник:Гиперболоид инженера Мошонкина/скрипт

import re

def convert_to_lua(wiki_text):
    lua_output = ["return {", '    ["Tournament_Data"] = {']

    # --- 1. Парсинг групп (БЕЗ ИЗМЕНЕНИЙ) ---
    group_pattern = re.compile(
        r'(?:;\s*Группа\s+([^\n]+)\s*)?\{\{Группа\s+(\d+)-А(\d+)\s*\|?(.*?)\}\}', 
        re.IGNORECASE | re.DOTALL
    )

    group_counter = 1
    for match in group_pattern.finditer(wiki_text):
        group_name = match.group(1).strip() if match.group(1) else f"Unnamed_{group_counter}"
        group_counter += 1

        num_teams = int(match.group(2))
        legs_indicator = int(match.group(3))
        leg_val = 1 if legs_indicator == 2 else 0

        body = match.group(4)
        parts = [p.strip() for p in body.split('|') if p.strip()]

        teams = []
        chunk_size = num_teams + 1

        for i in range(num_teams):
            chunk = parts[i * chunk_size : (i + 1) * chunk_size]
            if not chunk: continue
            status = chunk[0].replace('LY', 'L')
            name = chunk[1]
            scores = chunk[2:]
            teams.append({"name": name, "status": status, "scores": scores})

        lua_output.append(f'\n        ["Group_{group_name}"] = {{')
        lua_output.append('            type = "group",')
        lua_output.append('            standings = {')
        for t in teams:
            lua_output.append(f'                {{"{t["name"]}", "{t["status"]}"}},')
        lua_output.append('            },')
        lua_output.append('            matches = {')

        for i, team in enumerate(teams):
            score_idx = 0
            for j, opp in enumerate(teams):
                if i == j: continue 
                if score_idx < len(team["scores"]):
                    score_str = team["scores"][score_idx]
                    score_idx += 1
                    if ':' in score_str:
                        g1, g2 = score_str.split(':')
                        lua_output.append(f'                {{"{team["name"]}", "{opp["name"]}", {g1.strip()}, {g2.strip()}, {leg_val}}},')

        lua_output.append('            }')
        lua_output.append('        },')

    # --- 2. Парсинг матчей на вылет (ОБНОВЛЕНО) ---
    # Регулярка теперь читает `{{И` и любые символы до первого знака `|`
    knockout_pattern = re.compile(r'\{\{И[^\|\}]*\|(.*?)\}\}', re.IGNORECASE | re.DOTALL)
    ko_counter = 1

    for match in knockout_pattern.finditer(wiki_text):
        body = match.group(1)
        # Убрали `if p.strip()`, чтобы сохранять пустые значения (например, отсутствие цвета/статуса)
        parts = [p.strip() for p in body.split('|')]

        lua_output.append(f'\n        ["Knockout_Stage_{ko_counter}"] = {{')
        lua_output.append('            type = "knockout",')
        lua_output.append('            matches = {')
        ko_counter += 1

        # Гарантируем, что берем только полные блоки по 5 элементов
        for i in range(0, len(parts) - 4, 5):
            chunk = parts[i:i+5]

            st1 = chunk[0].replace('LY', 'L')
            t1 = chunk[1]
            score_str = chunk[2]
            t2 = chunk[3]
            st2 = chunk[4].replace('LY', 'L')

            mod = "nil"
            p1, p2 = "nil", "nil"
            base_scores = score_str

            # Парсинг доп. времени и пенальти
            if '(' in score_str:
                base_scores, ext = score_str.split('(', 1)
                ext = ext.replace(')', '').strip().lower()
                
                if 'et' in ext or 'дв' in ext:
                    mod = '"aet"'
                elif 'пен' in ext or 'pen' in ext:
                    mod = '"pen"'
                    pen_match = re.search(r'(\d+)\s*:\s*(\d+)', ext)
                    if pen_match:
                        p1, p2 = pen_match.groups()

            # Парсинг счета (разделяем матчи запятой, если их два)
            matches_scores = [pair.strip() for pair in base_scores.split(',')]
            g1_1, g2_1 = "nil", "nil" # Первый матч
            g1_2, g2_2 = "nil", "nil" # Второй матч (если есть)

            if len(matches_scores) > 0 and ':' in matches_scores[0]:
                s1, s2 = matches_scores[0].split(':')
                g1_1, g2_1 = s1.strip(), s2.strip()
            
            if len(matches_scores) > 1 and ':' in matches_scores[1]:
                s1, s2 = matches_scores[1].split(':')
                g1_2, g2_2 = s1.strip(), s2.strip()

            # Собираем новую строку: Команда1, Команда2, Голы1_1, Голы2_1, Голы1_2, Голы2_2, Модификатор, Пен1, Пен2, Статус1, Статус2
            lua_output.append(
                f'                {{"{t1}", "{t2}", {g1_1}, {g2_1}, {g1_2}, {g2_2}, {mod}, {p1}, {p2}, "{st1}", "{st2}"}},'
            )

        lua_output.append('            }')
        lua_output.append('        },')

    lua_output.append("    }")
    lua_output.append("}")
    return "\n".join(lua_output)


# --- ЗАПУСК ---
if __name__ == "__main__":
    # Тестовые данные (двойной матч + матч с пропущенными цветами)
    data = """
    {{Ичто-то там
    |G|КИР|4:3, 2:0|АЛЯ|R
    }}

    {{И1||АЛЯ|3:5|СЕН|}}
    """
    
    result = convert_to_lua(data)
    print(result)