main.py

import os
import sys
import json

from PyQt5 import QtGui, QtCore, QtWidgets
from PyQt5.QtWidgets import *

from utils.multiview import MultiView

from views.mainmenu import MainMenu
from views.townwizard import TownWizard
from views.sampleview import SampleView
from views.InfoDisplay import InfoDisplay
from views.helpscreen import HelpMenu


class Main(QMainWindow):
    def __init__(self):
        super().__init__()

        # hit the bars
        self.mb = self.menuBar()
        # self.mb.setNativeMenuBar(False)
        self.sb = self.statusBar()

        # houses custom view widgets
        # fancy stack widget
        self.mv = MultiView(self.mb, self.sb)
        self.mv.add_view('wizard', TownWizard(self.mv))
        self.mv.add_view('mainmenu', MainMenu(self.mv))
        self.mv.add_view('display', InfoDisplay(self.mv))
        self.mv.add_view('help', HelpMenu(self.mv))
        self.mv.set_view('mainmenu')

        # create other widgets
        self.widgets = {}
        self.widgets['main'] = QWidget()
        self.widgets['home'] = QPushButton('Home')
        self.widgets['wizard'] = QPushButton('Wizard')
        self.widgets['display'] = QPushButton('Display Town')
        self.widgets['help'] = QPushButton('Help')
        
        # create a base layout
        self.layout = QGridLayout()

        # set the layout
        self.widgets['main'].setLayout(self.layout)

        # add to the layout
        self.layout.addWidget(self.widgets['home'], 0, 0)
        self.layout.addWidget(self.widgets['wizard'], 0, 1)
        self.layout.addWidget(self.widgets['help'], 0, 3)
        self.layout.addWidget(self.widgets['display'], 0, 2)
        self.layout.addWidget(self.mv.get_widget(), 1, 0, 1, 4)

        # Add actions to buttons
        self.widgets['home'].clicked.connect(self.load_main_menu)
        self.widgets['wizard'].clicked.connect(self.load_wizard)
        self.widgets['display'].clicked.connect(self.load_display)
        self.widgets['help'].clicked.connect(self.load_help)

        # set center
        self.setCentralWidget(self.widgets['main'])

        # show
        self.show()

    def load_main_menu(self):
        self.mv.set_view('mainmenu')

    def load_wizard(self):
        self.mv.set_view('wizard')

    def load_display(self):
        self.mv.set_view('display')

    def load_help(self):
        self.mv.set_view('help')


if __name__ == '__main__':
    # QT IT UP
    app = QApplication(sys.argv)

    # initalize classes
    win = Main()

    # execute, clean up, and exit
    sys.exit(app.exec_())