Zaktualizuj 'main.py'

main.py

@@ -4,6 +4,7 @@ import secrets
 from src.graphics import *
 from src.rabbit import Rabbit
 from src.waiter import *
+from src.decisionTree import *
 
 if __name__ == "__main__":
     # SETUP
@@ -33,6 +34,15 @@ if __name__ == "__main__":
                     pygame.quit()
                     sys.exit()
                     break
+                    
+                if event.key == pygame.K_s:
+                    tree.TasksList('check', [10,3])
+                    tree.TasksList('eat', [5,12])
+                    tree.TasksList('order', [12,4])
+                    tree.TasksList('goToBar', [10,10])
+                    tree.TasksList('check', [3,4])
+                    tree.TasksList('eat', [10,5])
+                    tree.print()
 
                 if event.key == pygame.K_r:
                     temp = False