Make buttons flash when keys pressed

PyGame/Calculator/calc.py

@@ -1,8 +1,10 @@
 from dataclasses import dataclass
+from datetime import datetime
 import os
 os.environ['PYGAME_HIDE_SUPPORT_PROMPT'] = "hide"
 import pygame
 import sys
+import time
 
 pygame.init()
 
@@ -17,6 +19,7 @@ num_float = False
 float_precision = 1
 yellow = (240, 240, 20)
 white = (255, 255, 255)
+black = (0, 0, 0)
 
 button_origin_x = 90
 button_origin_y = 280
@@ -29,6 +32,8 @@ class Button:
     label: str
     key: int
     is_number: bool = False
+    flashing: bool = False
+    unflash_on: float = 0.0
 
 keypad_map = {
     str(pygame.K_KP_0): pygame.K_0,
@@ -72,32 +77,54 @@ valid_keys_numbers = [
     pygame.K_KP_9
 ]
 
-buttons = [
+valid_keys_equals = [
-    Button(3, 0, "0", pygame.K_0, True),
+    pygame.K_EQUALS,
-    Button(2, 0, "1", pygame.K_1, True),
+    pygame.K_RETURN,
-    Button(2, 1, "2", pygame.K_2, True),
+    pygame.K_KP_ENTER
-    Button(2, 2, "3", pygame.K_3, True),
-    Button(1, 0, "4", pygame.K_4, True),
-    Button(1, 1, "5", pygame.K_5, True),
-    Button(1, 2, "6", pygame.K_6, True),
-    Button(0, 0, "7", pygame.K_7, True),
-    Button(0, 1, "8", pygame.K_8, True),
-    Button(0, 2, "9", pygame.K_9, True),
-    Button(3, 1, ".", pygame.K_PERIOD),
-    Button(3, 2, "+-", pygame.K_QUESTION),
-    Button(0, 3, "<-", pygame.K_BACKSPACE),
-    Button(1, 3, "AC", pygame.K_a),
-    Button(2, 3, "C", pygame.K_c),
-    Button(3, 3, "=", pygame.K_EQUALS),
-    Button(4, 3, "*", pygame.K_ASTERISK),
-    Button(4, 0, "+", pygame.K_PLUS),
-    Button(4, 1, "-", pygame.K_MINUS),
-    Button(4, 2, chr(247), pygame.K_BACKSPACE)
 ]
 
+buttons = {
+    str(pygame.K_0): Button(3, 1, "0", pygame.K_0, True),
+    str(pygame.K_1): Button(2, 0, "1", pygame.K_1, True),
+    str(pygame.K_2): Button(2, 1, "2", pygame.K_2, True),
+    str(pygame.K_3): Button(2, 2, "3", pygame.K_3, True),
+    str(pygame.K_4): Button(1, 0, "4", pygame.K_4, True),
+    str(pygame.K_5): Button(1, 1, "5", pygame.K_5, True),
+    str(pygame.K_6): Button(1, 2, "6", pygame.K_6, True),
+    str(pygame.K_7): Button(0, 0, "7", pygame.K_7, True),
+    str(pygame.K_8): Button(0, 1, "8", pygame.K_8, True),
+    str(pygame.K_9): Button(0, 2, "9", pygame.K_9, True),
+    str(pygame.K_PERIOD): Button(3, 2, ".", pygame.K_PERIOD),
+    str(pygame.K_QUESTION): Button(3, 0, "+-", pygame.K_QUESTION),
+    str(pygame.K_BACKSPACE): Button(0, 3, "<-", pygame.K_BACKSPACE),
+    str(pygame.K_a): Button(1, 3, "AC", pygame.K_a),
+    str(pygame.K_c): Button(2, 3, "C", pygame.K_c),
+    str(pygame.K_EQUALS): Button(3, 3, "=", pygame.K_EQUALS),
+    str(pygame.K_ASTERISK): Button(4, 3, "*", pygame.K_ASTERISK),
+    str(pygame.K_PLUS): Button(4, 0, "+", pygame.K_PLUS),
+    str(pygame.K_MINUS): Button(4, 1, "-", pygame.K_MINUS),
+    str(pygame.K_SLASH): Button(4, 2, chr(247), pygame.K_SLASH)
+}
+
+def flash_button(b):
+    if b.flashing == False:
+        center_x = button_origin_x + (b.column * button_spacing)
+        center_y = button_origin_y + (b.row * button_spacing)
+        center = (center_x, center_y)
+        pygame.draw.circle(screen, white, center, 50, 4)
+        b.flashing = True
+        b.unflash_on = time.time() + 0.1
+    else:
+        center_x = button_origin_x + (b.column * button_spacing)
+        center_y = button_origin_y + (b.row * button_spacing)
+        center = (center_x, center_y)
+        pygame.draw.circle(screen, black, center, 50, 4)
+        pygame.draw.circle(screen, yellow, center, 50, 2)
+        b.flashing = False
+
 def draw_buttons():
 
-    for b in buttons:
+    for key, b in buttons.items():
         center_x = button_origin_x + (b.column * button_spacing)
         center_y = button_origin_y + (b.row * button_spacing)
         center = (center_x, center_y)
@@ -130,12 +157,29 @@ while True:
         if event.type == pygame.QUIT or (event.type == pygame.KEYDOWN and event.key == pygame.K_ESCAPE):
             pygame.quit()
             sys.exit()
+        #elif event.type == pygame.KEYDOWN:
+        #    print(event.key)
         elif event.type == pygame.KEYDOWN and event.key == pygame.K_c:
             display_num_whole = clear_display
+            display_num_decimal = ""
+            num_float = False
             float_precision = 1
+            flash_button(buttons[str(event.key)])
+        elif event.type == pygame.KEYDOWN and event.key == pygame.K_a:
+            display_num_whole = clear_display
+            display_num_decimal = ""
+            num_float = False
+            float_precision = 1
+            flash_button(buttons[str(event.key)])
         elif event.type == pygame.KEYDOWN and (event.key == pygame.K_PERIOD or event.key == pygame.K_KP_PERIOD):
+            actual_key = pygame.K_PERIOD
+            flash_button(buttons[str(actual_key)])
             num_float = True
+        elif event.type == pygame.KEYDOWN and event.key in valid_keys_equals and not pygame.key.get_mods() & pygame.KMOD_SHIFT:
+            actual_key = pygame.K_EQUALS
+            flash_button(buttons[str(actual_key)])
         elif event.type == pygame.KEYDOWN and event.key == pygame.K_BACKSPACE:
+            flash_button(buttons[str(event.key)])
             if num_float == True:
                 if display_num_decimal == "":
                     num_float = False
@@ -150,18 +194,32 @@ while True:
                     display_num_whole = display_num_whole[:-1]
                 elif display_num_whole != "0":
                     display_num_whole = "0"
-        elif event.type == pygame.KEYDOWN and event.key in valid_keys_numbers and len(display_num_whole) <= 10:
+        elif event.type == pygame.KEYDOWN and ((event.key == pygame.K_EQUALS and pygame.key.get_mods() & pygame.KMOD_SHIFT) or event.key == pygame.K_KP_PLUS):
+            actual_key = pygame.K_PLUS
+            flash_button(buttons[str(actual_key)])
+        elif event.type == pygame.KEYDOWN and (event.key == pygame.K_MINUS or event.key == pygame.K_KP_MINUS):
+            actual_key = pygame.K_MINUS
+            flash_button(buttons[str(actual_key)])
+        elif event.type == pygame.KEYDOWN and ((event.key == pygame.K_8 and pygame.key.get_mods() & pygame.KMOD_SHIFT) or event.key == pygame.K_KP_MULTIPLY):
+            actual_key = pygame.K_ASTERISK
+            flash_button(buttons[str(actual_key)])
+        elif event.type == pygame.KEYDOWN and (event.key == pygame.K_SLASH or event.key == pygame.K_KP_DIVIDE):
+            actual_key = pygame.K_SLASH
+            flash_button(buttons[str(actual_key)])
+        elif event.type == pygame.KEYDOWN and event.key in valid_keys_numbers:
             if str(event.key) in keypad_map.keys():
                 actual_key = keypad_map[str(event.key)]
             else:
                 actual_key = event.key
-            if num_float == False:
+            flash_button(buttons[str(actual_key)])
-                if display_num_whole == clear_display and actual_key != pygame.K_0:
+            if len(display_num_whole) <= 10:
-                    display_num_whole = chr(actual_key)
+                if num_float == False:
+                    if display_num_whole == clear_display and actual_key != pygame.K_0:
+                        display_num_whole = chr(actual_key)
+                    elif display_num_whole != clear_display:
+                        display_num_whole += chr(actual_key)
                 else:
-                    display_num_whole += chr(actual_key)
+                    display_num_decimal += chr(actual_key)
-            else:
-                display_num_decimal += chr(actual_key)
     if display_num_whole != old_display_num_whole or display_num_decimal != old_display_num_decimal:
         print(f"Number: {display_num_whole}", end='')
         if num_float == True:
@@ -174,4 +232,7 @@ while True:
     text = sevenseg_font.render(display_num_whole + "." + display_num_decimal, True, white)
     screen.blit(text, (580 - text.get_width(), 120))
     pygame.draw.rect(screen, yellow, pygame.Rect(10, 100, 580, 110), 2)
+    for key, b in buttons.items():
+        if b.flashing == True and time.time() >= b.unflash_on:
+            flash_button(b)
     pygame.display.flip()