<reponame>dwgoon/nezzle<gh_stars>1-10
import os
import importlib
import traceback

from qtpy.QtCore import Qt
from qtpy.QtCore import QObject
from qtpy.QtCore import Slot
from qtpy.QtCore import Signal
from qtpy.QtCore import QThread

from qtpy.QtWidgets import QWidget
from qtpy.QtWidgets import QMessageBox
from qtpy.QtWidgets import QDialog, QFileDialog
from qtpy.QtWidgets import QProgressBar

#import nezzle.utils
from nezzle.utils import reload_modules
from nezzle.utils import extract_name_and_ext
from nezzle.dialogs.progressdialog import ProgressDialog


class TaskThread(QThread):
    task_finished = Signal()
    def __init__(self, exc, nav, net):
        super().__init__()
        self._exc = exc
        self._nav = nav
        self._net = net

    def run(self):
        self._exc(self._nav, self._net)
        self.task_finished.emit()


class Worker(QObject):
    finished = Signal(object)

    def set_job(self, func, net):
        self._func_execute = func
        self._net = net

    @Slot()
    def execute(self):
        nets = self._func_execute(self._net)
        self.finished.emit(nets)


class CodeManager(QObject):
    def __init__(self, mainWindow):

        super().__init__(mainWindow)

        self.mw = mainWindow
        self._open_button = self.mw.ui_openCodeButton
        self._run_button = self.mw.ui_runButton
        self._file_path_edit = self.mw.ui_codeFilePathEdit

        self._open_button.released.connect(self.on_open_button_released)
        self._run_button.released.connect(self.on_run_button_released)
        self._run_button.setShortcut("Ctrl+R")

        self._worker = Worker()
        self._thread = QThread()
        self._worker.moveToThread(self._thread)
        self._worker.finished.connect(self.on_thread_finished)
        self._thread.started.connect(self._worker.execute)


    @property
    def file_path_edit(self):
        return self._file_path_edit

    def show_exception_message(self, err):
        print(err)
        msg = QMessageBox()
        msg.setIcon(QMessageBox.Critical)
        msg.setWindowTitle("An error has occurred")
        msg.setText(traceback.format_exc())
        msg.exec()


    @Slot()
    def on_open_button_released(self):
        try:
            dialog = QFileDialog(self.mw)
            dialog.setWindowTitle("Open a Python code")
            dialog.setAcceptMode(QFileDialog.AcceptOpen)
            dialog.setNameFilters([self.tr("Python module files (*.py)")])
            dialog.setFileMode(QFileDialog.ExistingFile)
            if dialog.exec() == QDialog.Accepted:
                fpath = dialog.selectedFiles()[0]
                fpath = fpath.strip()
                if not fpath.endswith('.py'):
                    raise RuntimeError("An inappropriate file "
                                       "for Python code: %s"%(fpath))
                # end of if
                self._file_path_edit.setText(fpath)
            # end of if
        except Exception as err:
            self.show_exception_message(err)

    @Slot()
    def on_run_button_released(self):
        try:
            self.process_code(self.file_path_edit.text())
        except Exception as err:
            self.show_exception_message(err)


    @Slot(object)
    def on_thread_finished(self, nets):
        self._thread.quit()
        if nets:
            for net in nets:
                self.mw.nt_manager.append_item(net.copy())

    def process_code(self, fpath):
        fname, fext = extract_name_and_ext(fpath)

        if not fname:
            raise RuntimeError("The path for the Python module is not valid.")

        # [REF] https://docs.python.org/3/library/importlib.html#importing-a-source-file-directly
        spec = importlib.util.spec_from_file_location(fname, fpath)
        mod = importlib.util.module_from_spec(spec)
        spec.loader.exec_module(mod)

        if self.mw.nt_manager.current_item:
            net = self.mw.nt_manager.current_item.data()
        else:
            net = None

        if hasattr(mod, 'create'):
            self._worker.set_job(mod.create, net.copy())
            self._thread.start()

        if hasattr(mod, 'update'):
            mod.update(self.mw.nt_manager, net)




import os
import importlib
import traceback

from qtpy.QtCore import Qt
from qtpy.QtCore import QObject
from qtpy.QtCore import Slot
from qtpy.QtCore import Signal
from qtpy.QtCore import QThread

from qtpy.QtWidgets import QWidget
from qtpy.QtWidgets import QMessageBox
from qtpy.QtWidgets import QDialog, QFileDialog
from qtpy.QtWidgets import QProgressBar

#import nezzle.utils
from nezzle.utils import reload_modules
from nezzle.utils import extract_name_and_ext
from nezzle.dialogs.progressdialog import ProgressDialog


class TaskThread(QThread):
    task_finished = Signal()
    def __init__(self, exc, nav, net):
        super().__init__()
        self._exc = exc
        self._nav = nav
        self._net = net

    def run(self):
        self._exc(self._nav, self._net)
        self.task_finished.emit()


class Worker(QObject):
    finished = Signal(object)

    def set_job(self, func, net):
        self._func_execute = func
        self._net = net

    @Slot()
    def execute(self):
        nets = self._func_execute(self._net)
        self.finished.emit(nets)


class CodeManager(QObject):
    def __init__(self, mainWindow):

        super().__init__(mainWindow)

        self.mw = mainWindow
        self._open_button = self.mw.ui_openCodeButton
        self._run_button = self.mw.ui_runButton
        self._file_path_edit = self.mw.ui_codeFilePathEdit

        self._open_button.released.connect(self.on_open_button_released)
        self._run_button.released.connect(self.on_run_button_released)
        self._run_button.setShortcut("Ctrl+R")

        self._worker = Worker()
        self._thread = QThread()
        self._worker.moveToThread(self._thread)
        self._worker.finished.connect(self.on_thread_finished)
        self._thread.started.connect(self._worker.execute)


    @property
    def file_path_edit(self):
        return self._file_path_edit

    def show_exception_message(self, err):
        print(err)
        msg = QMessageBox()
        msg.setIcon(QMessageBox.Critical)
        msg.setWindowTitle("An error has occurred")
        msg.setText(traceback.format_exc())
        msg.exec()


    @Slot()
    def on_open_button_released(self):
        try:
            dialog = QFileDialog(self.mw)
            dialog.setWindowTitle("Open a Python code")
            dialog.setAcceptMode(QFileDialog.AcceptOpen)
            dialog.setNameFilters([self.tr("Python module files (*.py)")])
            dialog.setFileMode(QFileDialog.ExistingFile)
            if dialog.exec() == QDialog.Accepted:
                fpath = dialog.selectedFiles()[0]
                fpath = fpath.strip()
                if not fpath.endswith('.py'):
                    raise RuntimeError("An inappropriate file "
                                       "for Python code: %s"%(fpath))
                # end of if
                self._file_path_edit.setText(fpath)
            # end of if
        except Exception as err:
            self.show_exception_message(err)

    @Slot()
    def on_run_button_released(self):
        try:
            self.process_code(self.file_path_edit.text())
        except Exception as err:
            self.show_exception_message(err)


    @Slot(object)
    def on_thread_finished(self, nets):
        self._thread.quit()
        if nets:
            for net in nets:
                self.mw.nt_manager.append_item(net.copy())

    def process_code(self, fpath):
        fname, fext = extract_name_and_ext(fpath)

        if not fname:
            raise RuntimeError("The path for the Python module is not valid.")

        # [REF] https://docs.python.org/3/library/importlib.html#importing-a-source-file-directly
        spec = importlib.util.spec_from_file_location(fname, fpath)
        mod = importlib.util.module_from_spec(spec)
        spec.loader.exec_module(mod)

        if self.mw.nt_manager.current_item:
            net = self.mw.nt_manager.current_item.data()
        else:
            net = None

        if hasattr(mod, 'create'):
            self._worker.set_job(mod.create, net.copy())
            self._thread.start()

        if hasattr(mod, 'update'):
            mod.update(self.mw.nt_manager, net)
#import nezzle.utils # end of if # end of if # [REF] https://docs.python.org/3/library/importlib.html#importing-a-source-file-directly