main.py

#!/usr/bin/env python3
import os
import sys
import traceback
import tcod

import color
import console
from event_handlers.base_event_handler import BaseEventHandler
from event_handlers.event_handler import EventHandler
import exceptions

import setup_game
import global_vars


def save_game(handler: BaseEventHandler, filename: str) -> None:
    """If the current event handler has an active Engine then save it."""
    if isinstance(handler, EventHandler):
        handler.engine.save_as(filename)
        print("Game saved.")


def main() -> None:
    """Main startup function."""
    screen_width = 80
    screen_height = 50

    # This tries to open the file bundled in the executable or the current directory
    try:
        tileset = tcod.tileset.load_tilesheet(
            os.path.join(os.path.dirname(__file__), "assets/Alloy_curses_12x12.png"),
            16,
            16,
            tcod.tileset.CHARMAP_CP437,
        )
    except NameError:
        tileset = tcod.tileset.load_tilesheet(
            os.path.join(os.path.dirname(sys.argv[0]), "assets/Alloy_curses_12x12.png"),
            16,
            16,
            tcod.tileset.CHARMAP_CP437,
        )

    handler: BaseEventHandler = setup_game.MainMenu()

    # Check if '-debug' is present in sys.argv
    debug_mode = "-debug" in sys.argv

    global_vars.DEBUG_MODE = debug_mode

    title = "Castle of the Eternal Night"
    if debug_mode:
        title += " - DEBUG MODE"

    with tcod.context.new_terminal(
        screen_width,
        screen_height,
        tileset=tileset,
        title=title,
        vsync=True,
    ) as context:
        console.set_root_console(
            tcod.console.Console(screen_width, screen_height, order="F")
        )
        root_console = console.get_root_console()
        console.set_context(context)
        try:
            while True:
                root_console.clear()
                handler.on_render(console=root_console)
                console.get_context().present(root_console)

                try:
                    for event in tcod.event.wait():
                        context.convert_event(event)
                        handler = handler.handle_events(event)
                except Exception:  # Handle exceptions in game.
                    traceback.print_exc()  # Print error to stderr.
                    # Then print the error to the message log.
                    if isinstance(handler, EventHandler):
                        handler.engine.message_log.add_message(
                            traceback.format_exc(), color.error
                        )
        except exceptions.QuitWithoutSaving:
            raise
        except SystemExit:  # Save and quit.
            save_game(handler, "savegame.sav")
            raise
        except BaseException:  # Save on any other unexpected exception.
            save_game(handler, "savegame.sav")
            raise


if __name__ == "__main__":
    main()