main.py

try:
    import simplegui
except ImportError:
    import SimpleGUICS2Pygame.simpleguics2pygame as simplegui

import globals

from vector import Vector
from interactions import Interaction
from walls import Wall
from levels import Levels
from level1 import Level1
from level2 import Level2
from level3 import Level3
from level4 import Level4
from level5 import Level5
from menu import Menu

# constants
# add them in the global files so they can be used across multiple files
CANVAS_DIMS = globals.CANVAS_DIMS

menu = Menu()

level1 = Level1()
level2 = Level2()
level3 = Level3()
level4 = Level4()
level5 = Level5()


def draw(canvas):
    if menu.game_start and not menu.game_end:
        Levels.update()
        Levels.draw(canvas)
    menu.draw(canvas)
    menu.update()


def mouse_handler(pos):
    global level1
    global level2
    global level3
    global level4
    global level5
    global menu
    if not menu.game_start:
        if menu.BUTT_Pos[0] - menu.BUTT_CENTRE[0] <= pos[0] <= menu.BUTT_Pos[0] + menu.BUTT_CENTRE[0]:
            if menu.BUTT_Pos[1] - menu.BUTT_CENTRE[1] <= pos[1] <= menu.BUTT_Pos[1] + menu.BUTT_CENTRE[1]:
                level1.LoadLevel()
                menu.game_start = True
    if menu.game_end or menu.won:
        if menu.END_BUTT_Pos[0] - menu.BUTT_CENTRE[0] <= pos[0] <= menu.END_BUTT_Pos[0] + menu.BUTT_CENTRE[0]:
            if menu.END_BUTT_Pos[1] - menu.BUTT_CENTRE[1] <= pos[1] <= menu.END_BUTT_Pos[1] + menu.BUTT_CENTRE[1]:
                menu.game_music.pause()
                menu.menu_music.pause()
                menu = Menu()
                Levels.restart()
                level1 = Level1()
                level2 = Level2()
                level3 = Level3()
                level4 = Level4()
                level5 = Level5()
                frame.set_keydown_handler(Levels.kbd.keyDown)
                frame.set_keyup_handler(Levels.kbd.keyUp)


frame = simplegui.create_frame('Game', CANVAS_DIMS[0], CANVAS_DIMS[1])
frame.set_canvas_background('black')
frame.set_draw_handler(draw)
frame.set_keydown_handler(Levels.kbd.keyDown)
frame.set_keyup_handler(Levels.kbd.keyUp)
frame.set_mouseclick_handler(mouse_handler)
frame.start()