import sys

from PyQt5 import QtWidgets
from PySide2.QtWidgets import *
from PySide2 import QtGui
from PySide2 import QtCore
from PySide2.QtGui import QIcon
# from PySide2.QtWidgets import QMainWindow, QMdiArea, QTabWidget, QToolBar, QAction, QTextEdit, QMenu



from QTreeViewExample.src.model.Node import Node
from QTreeViewExample.src.view.HierarchyTreeModel import HierarchyTreeModel
from QTreeViewExample.src.view.HierarchyTreeView import HierarchyTreeView


class MainWindow():
    def __init__(self):

        app = QtWidgets.QApplication(sys.argv)

        window = QtWidgets.QWidget()

        # windo2 = QtWidgets.QMainWindow.

        window.setWindowTitle("Frka44")



        window.setGeometry(40,40,500,500)

        window.show()

        sys.exit(app.exec_())


class TreeView():
    def __init__(self):
        app = QApplication(sys.argv)

        rootNode = Node("Workspace")
        project1 = Node("Project1")
        project2 = Node("Project2")
        package11 = Node("Package11")

        rootNode.addChild(project1)
        rootNode.addChild(project2)
        project1.addChild(package11)

        treeModel = HierarchyTreeModel(rootNode)

        tree = HierarchyTreeView()
        tree.setModel(treeModel)
        tree.show()

        sys.exit(app.exec_())



class Example4(QtWidgets.QMainWindow):

    def __init__(self):
        super(Example4, self).__init__()
        self.initUI()

    def initUI(self):
        textEdit = QtWidgets.QTextEdit()
        # self.setCentralWidget(textEdit)

        exitAction = QtWidgets.QAction('Exit', self)
        exitAction.setShortcut('Ctrl+Q')
        exitAction.setStatusTip('Exit application')
        exitAction.triggered.connect(self.close)

        newFile = QtWidgets.QAction('New File', self)
        newFile.setShortcut('Ctrl+N')
        newFile.setStatusTip('Create New File')
        newFile.triggered.connect(self.close)

        openFile = QtWidgets.QAction('Open File', self)
        openFile.setShortcut('Ctrl+O')
        openFile.setStatusTip('Open New File')
        openFile.triggered.connect(self.close)




        self.statusBar()

        rootNode = Node("Workspace")
        project1 = Node("Project1")
        # project2 = Node("Project2")
        # package11 = Node("Package11")
        #
        rootNode.addChild(project1)
        # rootNode.addChild(project2)
        # project1.addChild(package11)

        tre = QtWidgets.QWidget()
        layout = QtWidgets.QVBoxLayout(tre)
        # tre.setFeatures(QtWidgets.QDockWidget.DockWidgetMovable)
        # tre.setAllowedAreas(QtCore.Qt.LeftDockWidgetArea)
        treeModel = HierarchyTreeModel(rootNode)
        tree = TreeView()
        tree.setModel(treeModel)
        tre.setWidget(tree)
        # self.addDockWidget(QtCore.Qt.LeftDockWidgetArea, tre)
        layout.addWidget(tre)
        tre.show()
        self.setCentralWidget(layout)

        #STA JE QTDOCKWIDGETAREA JOS SA "="


        menubar = self.menuBar()
        fileMenu = menubar.addMenu('&File')
        fileMenu.addAction(newFile)
        fileMenu.addAction(openFile)
        fileMenu.addAction(exitAction)


        toolbar = self.addToolBar('Exit')
        toolbar.addAction(exitAction)

        self.setGeometry(300, 300, 350, 250)
        self.setWindowTitle('Main window')
        self.show()