-
Notifications
You must be signed in to change notification settings - Fork 0
/
AfterSave.py
75 lines (61 loc) · 2.58 KB
/
AfterSave.py
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
import pygame
from buttonClass import Buttons
from coloursss import *
import os
class AfterSavePage:
def __init__(self, screen, font, small_font, post_login_page):
self.screen = screen
self.font = font
self.small_font = small_font
self.post_login_page = post_login_page
self.message = ''
self.back_button = Buttons(self.screen, GREY, 500, 400, 200, 60, text='Back')
self.save_device_button = Buttons(self.screen, GREY, 500, 200, 200, 60, text='Save on device')
self.save_online_button = Buttons(self.screen, GREY, 500, 300, 200, 60, text='Save online')
def draw(self):
self.screen.fill(WHITE)
self.save_device_button.draw_button()
self.save_online_button.draw_button()
self.back_button.draw_button()
if self.message:
message_surface = self.small_font.render(self.message, True, BLACK)
self.screen.blit(message_surface, (500, 500))
pygame.display.flip()
def handle_event(self, event):
if event.type == pygame.MOUSEBUTTONDOWN:
if self.back_button.isOver(pygame.mouse.get_pos()):
return 'back'
elif self.save_device_button.isOver(pygame.mouse.get_pos()):
return 'save_on_device'
elif self.save_online_button.isOver(pygame.mouse.get_pos()):
print("Save online button clicked")
return 'save_online'
return None
def save_maze_to_file(self):
filename = input("Enter a filename to save the maze: ") + ".txt"
if os.path.exists(filename):
print("Name taken. Choose another name for your maze file.")
else:
try:
with open(filename, 'w') as f:
for row in self.post_login_page.grid_state:
f.write(' '.join(str(cell) for cell in row) + '\n')
print(f"{filename} saved")
except:
print("Error")
def start(self):
running = True
clock = pygame.time.Clock()
while running:
for event in pygame.event.get():
if event.type == pygame.QUIT:
running = False
action = self.handle_event(event)
if action == 'back':
self.post_login_page.start()
return
elif action == 'save_on_device':
self.save_maze_to_file()
self.draw()
clock.tick(30)
pygame.quit()