Pick teams randomly and instantly on spacebar

teampicker.py

@@ -0,0 +1,183 @@
+#!/bin/python3
+#
+# gamepicker.py - pick games for Novemeber 2020 sub crown
+#
+# Two files provided:
+#  - subs.txt -- 8 subscribers
+#  - teams.txt -- 26 teams
+#
+# Pseudocode:
+#   Split screen in 1:3
+#   - LHS = subscriber+team space
+#   - RHS = teams space
+#   Set BG image
+#   Init seed
+#   Lay out teams in order (3 cols)
+#   On keypress, show subscriber name & show team random picker (via highlight)
+#   Once picked, add subscriber + team to LHS
+
+import sys
+import random
+import pygame
+
+pygame.init()
+random.seed("BERTIE BEEF BAGGIO")
+
+SIZE = WIDTH, HEIGHT = 1920, 1080
+
+BGIMG = pygame.image.load("sub_crown_bg_2.png")
+
+FONT = pygame.font.SysFont("Fetamont", 50)
+SUBFONTHEIGHT = 35
+SUBFONT = pygame.font.SysFont("Fetamont", SUBFONTHEIGHT)
+OFFWHITE = (224, 224, 224)
+HIGHLIGHT = (255, 209, 0)
+BLUE = (20, 25, 153)
+PURPLE = (121, 35, 158)
+
+screen = pygame.display.set_mode(SIZE)
+
+# Set up team text positioning references
+#
+# 26 teams so 3 cols, 9 rows
+
+COLSTART = int(WIDTH/3)
+COLWIDTH = int((WIDTH-COLSTART)/3)
+
+ROWSTART = int(HEIGHT/12)
+ROWHEIGHT = int((HEIGHT-ROWSTART)/10)
+
+TEAMSFILE = "teams.txt"
+
+TEAMNUMCOLS = 3
+TEAMNUMROWS = 9
+
+TEAMCOLS = []
+TEAMROWS = []
+
+for i in range(TEAMNUMCOLS):
+    print(COLSTART + i*COLWIDTH)
+    TEAMCOLS.append(COLSTART + i*COLWIDTH)
+
+for i in range(TEAMNUMROWS):
+    TEAMROWS.append(ROWSTART + i*ROWHEIGHT)
+    print(ROWSTART + i*ROWHEIGHT)
+
+SUBSFILE = "subs.txt"
+SUBS = []
+
+SUBNUMROWS = 8
+SUBROWHEIGHT = int((HEIGHT-ROWSTART)/9)
+SUBROWS = []
+
+for i in range(SUBNUMROWS):
+    SUBROWS.append(ROWSTART + i*SUBROWHEIGHT)
+
+TEAMS = []  # List of Team
+
+
+class Team(object):
+    """Teams for iteration"""
+
+    def __init__(self, team, x, y):
+        self.team = team
+        self.x = x
+        self.y = y
+        self.colour = OFFWHITE
+
+    def highlight(self):
+        self.colour = HIGHLIGHT
+
+    def unhighlight(self):
+        self.colour = OFFWHITE
+
+    def draw(self):
+        draw_text(self.team, FONT, self.colour, self.y, self.x)
+
+
+class Sub(object):
+    """Like Team"""
+
+    def __init__(self, sub):
+        self.sub = sub
+        self.team = ""
+        self.x = 20
+        self.y = 0
+
+    def draw(self):
+        draw_text(self.sub, SUBFONT, BLUE, self.x, self.y)
+        if self.team != "":
+            draw_text(self.team, SUBFONT, PURPLE,
+                      self.x, int(self.y+SUBFONTHEIGHT))
+
+
+def draw_text(text, font, text_col, x, y):
+    img = font.render(text, True, text_col)
+    screen.blit(img, (x, y))
+
+
+def setup_team_board():
+    """Grab teams from text file and set their positions"""
+    with open(TEAMSFILE, "r") as fh:
+        for row in range(TEAMNUMROWS):
+            for col in range(TEAMNUMCOLS):
+                name = fh.readline().strip()
+                if name == "":
+                    continue
+                print("Adding {} at {},{}".format(name,
+                                                  TEAMCOLS[col],
+                                                  TEAMROWS[row]))
+                TEAMS.append(Team(name,
+                                  TEAMROWS[row], TEAMCOLS[col]))
+
+
+def setup_subs():
+    """Check positioning, not actually used"""
+    for i in range(len(SUBS)):
+        # print("drawing {} at {},{}".format(SUBS[i], 20, SUBROWS[i]))
+        SUBS[i].y = SUBROWS[i]
+
+
+# Prep
+
+for line in open(SUBSFILE, "r"):
+    sub = line.strip()
+    SUBS.append(Sub(sub))
+print(SUBS)
+
+random.shuffle(SUBS)
+setup_team_board()
+random.shuffle(TEAMS)
+setup_subs()
+
+while True:
+    for event in pygame.event.get():
+        if event.type == pygame.QUIT:
+            sys.exit()
+        elif event.type == pygame.KEYDOWN:
+            if event.key == 113:  # Q
+                pygame.quit()
+                sys.exit()
+            elif event.key == 115:
+                TEAMS[5].highlight()
+            elif event.key == 32:  # Spacebar
+                for sub in SUBS:
+                    if sub.team == "":
+                        sub.team = TEAMS.pop().team
+            elif event.key == 49:
+                team = TEAMS.pop()
+                SUBS[0].team = team.team
+            elif event.key == 50:
+                SUBS[1].team = TEAMS[random.randrange(len(TEAMS))].team
+
+            print(event.key)
+
+    screen.blit(BGIMG, (0, 0))
+
+    for team in TEAMS:
+        team.draw()
+
+    for sub in SUBS:
+        sub.draw()
+
+    pygame.display.update()