Uruguay2022-2811.svg

Source:

from copy import deepcopy

import re

print('strict digraph D {')
print('  rankdir = LR;')
def slugify(text):
    text = text.replace('🇵🇹', 'pt')
    text = text.replace('🇬🇭', 'gh')
    text = text.replace('🇺🇾', 'uy')
    text = text.replace('🇰🇷', 'kr')
    return re.sub(r'[\W_]+', '_', text.lower())

PORTUGAL = 'Portugal'
GHANA = 'Ghana'
URUGUAY = 'Uruguay'
KOREA = 'Corea del Sur'
abbr = {PORTUGAL: "🇵🇹", GHANA: "🇬🇭", URUGUAY: "🇺🇾", KOREA: "🇰🇷"}
initial = {
    PORTUGAL: {'points': 3, 'gd': ["+1"]},
    GHANA: {'points': 3, 'gd': ["-1"]},
    URUGUAY: {'points': 1, 'gd': []},
    KOREA: {'points': 1, 'gd': []},
}
counter = 0

for r1 in [
    # {'points': {KOREA: 3}, 'winner': KOREA},
    {'points': {GHANA: 3}, 'winner': GHANA},
    # {'points': {KOREA: 1, GHANA: 1}},
]:
    r1_teams = [KOREA, GHANA]
    for r2 in [
        {'points': {PORTUGAL: 3}, 'winner': PORTUGAL},
        {'points': {URUGUAY: 3}, 'winner': URUGUAY},
        {'points': {PORTUGAL: 1, URUGUAY: 1}},
    ]:
        r2_teams = [PORTUGAL, URUGUAY]
        for r3 in [
            {'points': {URUGUAY: 3}, 'winner': URUGUAY},
            {'points': {GHANA: 3}, 'winner': GHANA},
            {'points': {URUGUAY: 1, GHANA: 1}},
        ]:
            r3_teams = [URUGUAY, GHANA]
            for r4 in [
                {'points': {PORTUGAL: 3}, 'winner': PORTUGAL},
                {'points': {KOREA: 3}, 'winner': KOREA},
                {'points': {PORTUGAL: 1, KOREA: 1}},
            ]:
                r4_teams = [PORTUGAL, KOREA]
                points = deepcopy(initial)
                history = []
                for (r, teams) in ((r2, r2_teams), (r3, r3_teams), (r4, r4_teams)):
                    for (t, p) in r['points'].items():
                        points[t]['points'] += p
                    if 'winner' in r:
                        loser = [t for t in teams if t != r['winner']][0]
                        points[r['winner']]['gd'].append(f'+g_{abbr[r["winner"]]}_{abbr[loser]}')
                        points[loser]['gd'].append(f'-g_{abbr[r["winner"]]}_{abbr[loser]}')
                        history.append(f'{abbr[r["winner"]]} > {abbr[loser]}')
                    else:
                        history.append(f'{abbr[teams[0]]} = {abbr[teams[1]]}')
                more = len([1 for (t, val) in points.items() if t != URUGUAY and val['points'] > points[URUGUAY]['points']])
                equal = len([1 for (t, val) in points.items() if t != URUGUAY and val['points'] == points[URUGUAY]['points']])
                if more >= 2: bgcolor = 'red'
                elif more + equal >= 2: bgcolor = 'yellow'
                else: bgcolor = 'green'
                for i in range(len(history)):
                    breadcrumb_prev = slugify('_'.join((slugify(history[j]) for j in range(i))))
                    breadcrumb = slugify('_'.join((slugify(history[j]) for j in range(i+1))))
                    if breadcrumb_prev == "":
                        print(f'  {breadcrumb}[label="{history[i]}"];')
                    else:
                        print(f'  {breadcrumb_prev} -> {breadcrumb};')
                        if i == len(history) - 1:
                            print(f'  {breadcrumb}[label="{history[i]}"];')
                            print(f'  {breadcrumb} -> {breadcrumb}_table;')
                            print(f'  {breadcrumb}_table[shape=none label=<')
                            print(f'    <table bgcolor="{bgcolor}">')
                            for (t, val) in sorted(points.items(), key=lambda v: -v[1]['points']):
                                gd = "".join(val['gd'])
                                if gd == "":
                                    gd = "0"
                                print(f'<tr><td>{abbr[t]}</td><td>{val["points"]}</td></tr>')
                            print(f'    </table>')
                            print(f'  >]')
                        else:
                            print(f'  {breadcrumb}[label="{history[i]}"];')
print('}')