creator.py

import sys
from pprint import pprint

from PyQt5.QtWidgets import QApplication

from gui import CreatorGUI
from Node import Node
from story import Story
from utils import VarCommand
from VarStore import VarStore


def open_creator():
    app = QApplication(sys.argv)
    ex = CreatorGUI()
    sys.exit(app.exec_())


def create_test_story() -> Story:

    store = VarStore()
    story = Story(store)

    # Create nodes
    n1 = story.create_node('n1')
    n2 = story.create_node('n2')
    n3 = story.create_node('n3')
    n4 = story.create_node('n4')
    end1 = story.create_node('end1')
    dead1 = story.create_node('dead-end')

    n1.set_text('Currently at n1')
    n2.set_text('Currently at n2')
    n3.set_text('Currently at n3')
    n4.set_text('Currently at n4')
    end1.set_text('Ending #1')
    dead1.set_text('Dead-end reached')

    block3 = VarCommand('test', '=', 5)
    goto_4_1 = [VarCommand('test', '+', 7), VarCommand('test2', '+', 'abd')]
    req4 = VarCommand('test', '=', 5)

    n3.add_change(block3)
    n4.add_requirement(req4)

    n1.add_outgoing(n2, 'Goto 2', None)
    n1.add_outgoing(n3, 'Goto 3', None)
    n2.add_outgoing(n3, 'Goto 3', None)
    n3.add_outgoing(n4, 'Goto 4', None)
    n3.add_outgoing(dead1, 'Goto dead-end', None)
    n4.add_outgoing(end1, 'Goto ending 1', None)
    n4.add_outgoing(n1, 'Goto 1', goto_4_1)

    story.change_beginning(n1)
    story.add_ending(end1)

    return story


def save_story(story: Story, path: str = 'story.sgf'):
    raise NotImplementedError
    # TODO: Write encoder for Story


def main():
    open_creator()


if __name__ == '__main__':
    main()