main.py

import logging
import pygame
from editor import NodeEditor
from toolbar import Toolbar
from button import Button
from actions import (AddNodeAction,
                     DeleteAllAction,
                     DumpGraphAction,
                     UndoAction,
                     SaveGraphAction,
                     LoadGraphAction)

if __name__ == "__main__":
    pygame.init()
    logging.basicConfig(level=logging.INFO)
    toolbar = Toolbar()
    toolbar.add_button(Button(action=AddNodeAction(), label="Add Node"))
    toolbar.add_button(Button(action=DeleteAllAction(), label="Clear All"))
    toolbar.add_button(Button(action=UndoAction(), label="Undo"))
    toolbar.add_button(Button(action=DumpGraphAction(), label="Print Graph Model"))
    toolbar.add_button(Button(action=SaveGraphAction(), label="Save"))
    toolbar.add_button(Button(action=LoadGraphAction(), label="Load"))
    editor = NodeEditor(toolbar)
    editor.run()