deadball/art.py at master · lepot311/deadball · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
from termcolor import colored

from enums import Hand, Positions


field = r'''
                  ___________________
            _____/                   \_____
        ___/                               \___
     __/                                       \__
   _/                     CF                      \_
 _/                                                 \_
/                                                     \
\           LF       XXXXXXXXXXXXX       RF           /
  \                                                 /
    \                     R2                      /
      \              SS  .   .  2B              /
        \              .       .              /
          \      3B  .    PPP    .  1B      /
            \      R3      -      R1      /
              \      .           .      /
                \      .       .      /
                  \      .   .      /
                    \     BBB     /
                      \   CCC   /
                        \     /
                          \_/

'''

positions = {
    'BBB': (0, 'L'),
    'R1' : (1, 'R'),
    'R2' : (2, 'R'),
    'R3' : (3, 'R'),

    'PPP': (1, 'R'),
    'CCC': (2, 'R'),
    '1B' : (3, 'R'),
    '2B' : (4, 'R'),
    '3B' : (5, 'R'),
    'SS' : (6, 'R'),
    'LF' : (7, 'R'),
    'CF' : (8, 'R'),
    'RF' : (9, 'R'),
}

def color_field(field):
    # color field green
    field = field.replace('\\', colored('\\', 'light_green'))
    field = field.replace('/', colored('/', 'light_green'))
    field = field.replace('_', colored('_', 'light_green'))
    field = field.replace('.', colored('.', 'yellow'))
    return field

def print_field(game, field):
    field = color_field(field)
    assert 'PPP' in field

    # DRAW FIELDERS
    # draw pitcher
    player = game.inning.half.fielding.pitcher
    if player.hand == Hand.R:
        sub = f"{player.number:>3}"
    else:
        sub = f"{player.number:<3}"
    field = field.replace('PPP', colored(sub, game.inning.half.fielding.color))

    # draw fielders
    for player in game.inning.half.fielding.lineup:
        #print(player.name, player.pos.name, player.number)
        if player.pos.name == 'C':
            pos = 'CCC'
            if game.inning.half.batting.up_to_bat.hand == Hand.R:
                sub = f"{player.number:>3}"
            else:
                sub = f"{player.number:<3}"
        else:
            pos = player.pos.name
            sub = f"{player.number:>2}"
        field = field.replace(pos, colored(sub, game.inning.half.fielding.color))

    # DRAW BATTING TEAM
    # draw batter
    player = game.inning.half.batting.up_to_bat
    #print('batter:', player.name, player.number)
    if player.hand == Hand.R:
        sub = f"{player.number:<3}"
    else:
        sub = f"{player.number:>3}"
    field = field.replace('BBB', colored(sub, game.inning.half.batting.color))

    # draw runners
    #print('runners:')
    for base_number, player in enumerate(game.bases):
        base_number += 1
        pos = f"R{base_number}"
        if player:
            #print(player.name, player.number)
            sub = f"{player.number:>2}"
            field = field.replace(pos, colored(sub, game.inning.half.batting.color))
        else:
            if base_number == 3:
                sub = f"{'o':<2}"
            else:
                sub = f"{'o':>2}"
            field = field.replace(pos, colored(sub, 'white'))

    # draw result of AB
    ab_result = game.inning.half.at_bat.result or ''
    sub = f"{ab_result:^13}"
    field = field.replace('XXXXXXXXXXXXX', sub)

    print(field)


if __name__ == '__main__':
    field = color_field(field)

    for pos, info in positions.items():
        number, hand = info
        if len(pos) == 2:
            sub = f"{number:>2}"
        else:
            if pos == 'BBB':
                if hand == 'R':
                    sub = f"{number:<3}"
                else:
                    sub = f"{number:>3}"
            elif pos == 'CCC':
                batter_hand = positions['BBB'][1]
                if batter_hand == 'R':
                    sub = f"{number:>3}"
                else:
                    sub = f"{number:<3}"
            else:
                if hand == 'R':
                    sub = f"{number:>3}"
                else:
                    sub = f"{number:<3}"
        field = field.replace(pos, colored(sub, 'light_red'))

    print(field)