main.py

import sys
import pygame
from space_game import game as space_game
from setup_menu import start_setup
import space_game.game_config as config
import flappy_bird.game as flappy
import pdf.show as pdf

def close_setup_window():
    pygame.display.quit()
    pygame.quit()

def main(mode):
    pygame.init()
    size = (config.WINDOW_WIDTH, config.WINDOW_HEIGHT)
    screen = pygame.display.set_mode(size)
    pygame.display.set_caption(config.WINDOW_CAPTION)

    start_setup(screen)

    if mode == 'space':
        space_game.start_game(screen)
    elif mode == 'flappy':
        close_setup_window()
        flappy.start_game('flappy_bird\\flappy_bird.exe')
    elif mode == 'reader':
        close_setup_window()
        pdf.start_pdf_reader('pdf\War_and_Peace.pdf')

if __name__ == '__main__':
    main(sys.argv[1])