KatKatKateryna
104 lines
4.1 KiB
Python
104 lines
4.1 KiB
Python
import threading
|
|
from specklepy_qt_ui.qt_ui.dockwidget_main import SpeckleQGISDialog as SpeckleQGISDialog_UI
|
|
import specklepy_qt_ui.qt_ui
|
|
|
|
from speckle.ui_widgets.widget_transforms import MappingSendDialogQGIS
|
|
|
|
from PyQt5 import QtWidgets, uic
|
|
import os
|
|
import inspect
|
|
from specklepy.logging.exceptions import (SpeckleException, GraphQLException)
|
|
from specklepy.logging import metrics
|
|
|
|
|
|
from PyQt5 import QtWidgets, uic
|
|
from PyQt5.QtWidgets import QCheckBox, QListWidgetItem, QHBoxLayout, QWidget
|
|
from PyQt5.QtCore import pyqtSignal
|
|
|
|
|
|
from specklepy_qt_ui.qt_ui.widget_transforms import MappingSendDialog
|
|
from specklepy_qt_ui.qt_ui.LogWidget import LogWidget
|
|
from specklepy_qt_ui.qt_ui.utils.logger import logToUser
|
|
from specklepy_qt_ui.qt_ui.DataStorage import DataStorage
|
|
|
|
FORM_CLASS, _ = uic.loadUiType(
|
|
os.path.join(os.path.dirname(specklepy_qt_ui.qt_ui.__file__), os.path.join("ui", "dockwidget_main.ui") )
|
|
)
|
|
|
|
class SpeckleQGISDialog(SpeckleQGISDialog_UI, FORM_CLASS):
|
|
|
|
def __init__(self, parent=None):
|
|
"""Constructor."""
|
|
super(SpeckleQGISDialog_UI, self).__init__(parent)
|
|
|
|
self.setupUi(self)
|
|
self.runAllSetup()
|
|
|
|
def createMappingDialog(self):
|
|
|
|
if self.mappingSendDialog is None:
|
|
self.mappingSendDialog = MappingSendDialogQGIS(None)
|
|
self.mappingSendDialog.dataStorage = self.dataStorage
|
|
|
|
self.mappingSendDialog.runSetup()
|
|
|
|
def completeStreamSection(self, plugin):
|
|
try:
|
|
self.streams_remove_button.clicked.connect( lambda: self.onStreamRemoveButtonClicked(plugin) )
|
|
self.streamList.currentIndexChanged.connect( lambda: self.onActiveStreamChanged(plugin) )
|
|
self.streamBranchDropdown.currentIndexChanged.connect( lambda: self.populateActiveCommitDropdown(plugin) )
|
|
return
|
|
except Exception as e:
|
|
logToUser(e, level = 2, func = inspect.stack()[0][3], plugin=self)
|
|
return
|
|
|
|
def onStreamRemoveButtonClicked(self, plugin):
|
|
try:
|
|
from speckle.utils.project_vars import set_project_streams
|
|
if not self: return
|
|
index = self.streamList.currentIndex()
|
|
if len(plugin.current_streams) > 0: plugin.current_streams.pop(index)
|
|
plugin.active_stream = None
|
|
self.streamBranchDropdown.clear()
|
|
self.commitDropdown.clear()
|
|
#self.streamIdField.setText("")
|
|
|
|
set_project_streams(plugin)
|
|
self.populateProjectStreams(plugin)
|
|
except Exception as e:
|
|
logToUser(e, level = 2, func = inspect.stack()[0][3], plugin=self)
|
|
return
|
|
|
|
def populateProjectStreams(self, plugin):
|
|
try:
|
|
from speckle.utils.project_vars import set_project_streams
|
|
if not self: return
|
|
self.streamList.clear()
|
|
for stream in plugin.current_streams:
|
|
self.streamList.addItems(
|
|
[f"Stream not accessible - {stream[0].stream_id}" if stream[1] is None or isinstance(stream[1], SpeckleException) else f"{stream[1].name}, {stream[1].id} | {stream[0].stream_url.split('/streams')[0].split('/projects')[0]}"]
|
|
)
|
|
if len(plugin.current_streams)==0: self.streamList.addItems([""])
|
|
self.streamList.addItems(["Create New Stream"])
|
|
set_project_streams(plugin)
|
|
index = self.streamList.currentIndex()
|
|
if index == -1: self.streams_remove_button.setEnabled(False)
|
|
else: self.streams_remove_button.setEnabled(True)
|
|
|
|
if len(plugin.current_streams)>0: plugin.active_stream = plugin.current_streams[0]
|
|
except Exception as e:
|
|
logToUser(e, level = 2, func = inspect.stack()[0][3], plugin=self)
|
|
return
|
|
|
|
def cancelOperations(self):
|
|
#print("____cancelOperations______")
|
|
for t in threading.enumerate():
|
|
#print(t.name)
|
|
if 'speckle_' in t.name:
|
|
#print(f"thread to kill: {t}")
|
|
t.kill()
|
|
t.join()
|
|
# not printed if same thread
|
|
#print("Remaining threads: ")
|
|
#print(threading.enumerate())
|
|
|