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

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)
        # Убрано "if p.strip()", чтобы сохранять пустые поля, если статус не указан
        parts = [p.strip() for p in body.split('|')]

        teams = []
        chunk_size = num_teams + 1

        for i in range(num_teams):
            chunk = parts[i * chunk_size : (i + 1) * chunk_size]
            if len(chunk) < chunk_size: 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. Парсинг матчей на вылет ---
    # Регулярка для И1, И2 и т.д.
    knockout_pattern = re.compile(r'\{\{И\d+\s*\|?(.*?)\}\}', 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

        for i in range(0, len(parts), 5):
            chunk = parts[i:i+5]
            if len(chunk) < 5: continue

            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)
                base_scores = base_scores.strip()
                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+):(\d+)', ext)
                    if pen_match:
                        p1, p2 = pen_match.groups()

            # Разбиваем по запятой для двойных противостояний
            matches_list = [s.strip() for s in base_scores.split(',')]

            if len(matches_list) == 2:
                # --- ЛОГИКА ДЛЯ ДВУХ МАТЧЕЙ ---
                m1 = matches_list[0]
                m2 = matches_list[1]

                g1_1, g2_1 = "nil", "nil"
                if ':' in m1:
                    g1_1, g2_1 = [x.strip() for x in m1.split(':')]

                g1_2, g2_2 = "nil", "nil"
                if ':' in m2:
                    g1_2, g2_2 = [x.strip() for x in m2.split(':')]

                lua_output.append(
                    f'                {{"{t1}", "{t2}", {g1_1}, {g2_1}, {g1_2}, {g2_2}, {mod}, {p1}, {p2}, "{st1}", "{st2}"}},'
                )
            else:
                # --- ЛОГИКА ДЛЯ ОДНОГО МАТЧА ---
                m1 = matches_list[0]
                g1, g2 = "nil", "nil"
                if ':' in m1:
                    g1, g2 = [x.strip() for x in m1.split(':')]

                lua_output.append(
                    f'                {{"{t1}", "{t2}", {g1}, {g2}, {mod}, {p1}, {p2}, "{st1}", "{st2}", 1}},'
                )

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

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


# --- ЗАПУСК ---
if __name__ == "__main__":
    # Тестовые данные из вашего примера
    data = """
    {{И1|G|КИР|4:3, 2:0|АЛЯ|R}}
    
    {{И2|G|КИР|4:3|АЛЯ|R}}
    
    {{И3||АЛЯ|3:5|СЕН|}}
    """
    
    result = convert_to_lua(data)
    print(result)