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

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)) # Читаем цифру после А (1 или 2)
        leg_val = 1 if legs_indicator == 2 else 0 # 1 - два круга (дома/выезд), 0 - один круг (нейтральное)

        body = match.group(4)
        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
                    
                    # ЛОГИКА ФИЛЬТРАЦИИ:
                    # Если 2 круга (leg_val == 1) - пишем все матчи
                    # Если 1 круг (leg_val == 0) - пишем только уникальные пары (где индекс i < j)
                    if leg_val == 1 or (leg_val == 0 and i < j):
                        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'\{\{И\d+\s*\|?(.*?)\}\}', re.IGNORECASE | re.DOTALL)
    ko_counter = 1

    for match in knockout_pattern.finditer(wiki_text):
        body = match.group(1)
        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__":
    # Тестовые данные с группами А2 (2 круга) и А1 (1 круг)
    data = """
    ; Группа 1
    {{Группа 3-А2|
    G|КИР|2:0|3:3|
    G|ЦАР|0:2|4:2|
    R|ДОМ|3:3|2:4}}
    
    ; Группа 2
    {{Группа 3-А1|
    G|КИР|2:0|3:3|
    G|ЦАР|0:2|4:2|
    R|ДОМ|3:3|2:4}}
    """
    
    result = convert_to_lua(data)
    print(result)