class UI(QWidget):
update_signal = pyqtSignal(str)
show_qr_signal = pyqtSignal(bytes)
finish_signal = pyqtSignal()
close_qr_signal = pyqtSignal()
def __init__(self):
super().__init__()
self.logger = logging.getLogger(__name__)
formatter = logging.Formatter(
fmt="%(asctime)s-%(levelname)s-%(message)s",
datefmt="%Y-%m-%d %H:%M:%S")
filehandler = logging.FileHandler(filename="logs.log",
mode="w",
encoding="utf-8")
handler = QLogger(update_signal=self.update_signal)
handler.setLevel(logging.INFO)
filehandler.setLevel(logging.INFO)
self.logger.setLevel(logging.INFO)
if os.path.exists("config.json") == False:
self.gen_conf()
with open(file="config.json", mode="r",
encoding="utf-8") as conf_reader:
conf = json.loads(conf_reader.read())
debug = bool(conf["debug"])
if debug == True:
handler.setLevel(logging.DEBUG)
filehandler.setLevel(logging.DEBUG)
self.logger.setLevel(logging.DEBUG)
handler.setFormatter(formatter)
filehandler.setFormatter(formatter)
self.logger.addHandler(handler)
self.logger.addHandler(filehandler)
self.logger.debug("当前调试状态：%s" % debug)
self.resize(1024, 768)
self.setWindowOpacity(0.9)
self.setAttribute(Qt.WidgetAttribute.WA_TranslucentBackground)
self.setWindowFlag(Qt.WindowFlags.FramelessWindowHint)
self.setAutoFillBackground(True)
self.work = Work(show_qr_signal=self.show_qr_signal,
finish_signal=self.finish_signal,
close_qr_signal=self.close_qr_signal)
self.work_thread = QThread()
self.work.moveToThread(self.work_thread)
self.main_layout = QGridLayout()
self.setLayout(self.main_layout)
self.title = QLabel("ChinaUniOnlineGUI")
self.title.setStyleSheet(
"QLabel{border:none;border-radius:5px;background:transparent;color:#9AD3BC;font-size:60px;}"
)
self.title.setAlignment(Qt.Alignment.AlignCenter)
handler.widget.setStyleSheet(
"QPlainTextEdit{font-family:Microsoft YaHei;background:#F3EAC2;border:none;border-radius:5px;}QScrollBar:vertical,QScrollBar::handle:vertical{background:#F3EAC2;border:none;border-radius:8px;width:16px;}QScrollBar::handle:vertical:hover{background:#F5B461;}QScrollBar::add-page:vertical,QScrollBar::sub-page:vertical{background:#FFFDF9;border:none;border-radius:8px;width:16px;}QScrollBar::down-arrow:vertical,QScrollBar::up-arrow:vertical{background:#F5B461;border:none;border-radius:8px;width:16px;height:16px;}QScrollBar::sub-line:vertical,QScrollBar::add-line:vertical{background:transparent;border:none;}"
)
self.control = QVBoxLayout()
self.control_close = QPushButton()
self.control_close.setToolTip("关闭")
self.control_close.setStyleSheet(
"QPushButton{background:#FFE3ED;border-radius:5px;border:none;}QPushButton:hover{background:#EC524B;}"
)
self.contron_max = QPushButton()
if self.isMaximized() == False:
self.contron_max.setToolTip("最大化")
else:
self.contron_max.setToolTip("还原")
self.contron_max.setStyleSheet(
"QPushButton{background:#FFFDF9;border-radius:5px;border:none;}QPushButton:hover{background:#F5B461;}"
)
self.control_min = QPushButton()
self.control_min.setToolTip("最小化")
self.control_min.setStyleSheet(
"QPushButton{background:#BEEBE9;border-radius:5px;border:none;}QPushButton:hover{background:#F3EAC2;}"
)
self.start_button = QPushButton("开始(&S)")
self.start_button.setStyleSheet(
"QPushButton{background:#9BE3DE;border:none;border-radius:5px;font-size:20px;font-family:DengXian;}QPushButton:hover{background:#9AD3BC;}"
)
self.start_button.setToolTip("开始")
self.start_button.setFixedSize(120, 60)
self.start_button.setDefault(True)
setting_button = QPushButton("设置")
setting_button.setToolTip("设置")
setting_button.setFixedSize(60, 60)
setting_button.setStyleSheet(
"QPushButton{background:#9BE3DE;border:none;border-radius:5px;font-size:20px;font-family:DengXian;}QPushButton:hover{background:#9AD3BC;}"
)
setting_button.clicked.connect(self.setting_callback)
start = QHBoxLayout()
start.addWidget(self.start_button, 2)
start.addWidget(setting_button, 1)
self.control_close.clicked.connect(self.close)
self.control_min.clicked.connect(self.min_callback)
self.contron_max.clicked.connect(self.max_callback)
self.start_button.clicked.connect(self.start_callback)
self.work_thread.started.connect(self.work.start)
self.finish_signal.connect(self.finish_callback)
self.close_qr_signal.connect(self.close_qr)
self.control.addWidget(self.control_min)
self.control.addWidget(self.contron_max)
self.control.addWidget(self.control_close)
self.main_layout.addLayout(self.control, 0, 0)
self.main_layout.addWidget(self.title, 0, 1)
self.main_layout.addLayout(start, 0, 2)
self.main_layout.addWidget(handler.widget, 1, 1)
self.update_signal.connect(handler.widget.appendPlainText)
handler.widget.textChanged.connect(handler.scroll_widget_to_bottom)
self.show_qr_signal.connect(self.show_qr)
self.logger.debug("已初始化UI")
def min_callback(self):
if self.isMinimized() == False:
self.showMinimized()
def max_callback(self):
if self.isMaximized() == False:
self.showMaximized()
self.contron_max.setToolTip("还原")
else:
self.showNormal()
self.contron_max.setToolTip("最大化")
def start_callback(self):
self.start_time = time.time()
self.work_thread.start()
self.start_button.setEnabled(False)
self.start_button.setText("执行中...")
def finish_callback(self):
self.start_button.setEnabled(True)
self.start_button.setText("开始")
passed_time = time.time() - self.start_time
mins, secs = divmod(passed_time, 60)
hours, mins = divmod(mins, 60)
self.logger.info("执行完成，共计用时 {:0>2d}:{:0>2d}:{:0>2d}".format(
int(hours), int(mins), int(secs)))
def show_qr(self, qr: bytes):
title_label = QLabel("请使用微信扫描小程序码完成登陆")
title_label.setStyleSheet(
"QLabel{color:#ffe3ed;border:none;background-color:transparent;border-radius:5px;}"
)
title_label.setAlignment(Qt.Alignment.AlignCenter)
title_label.setFixedHeight(20)
qr_label = QLabel()
pixmap = QPixmap()
pixmap.loadFromData(qr)
qr_label.setPixmap(pixmap)
qr_label.setStyleSheet(
"QLabel{color:#ffe3ed;border:none;background-color:transparent;border-radius:5px;}"
)
layout_ = QVBoxLayout()
layout_.addWidget(title_label, 1)
layout_.addWidget(qr_label, 9)
self.qr_dialog = QWidget(self)
self.qr_dialog.setLayout(layout_)
self.main_layout.addWidget(self.qr_dialog, 1, 1,
Qt.Alignment.AlignCenter)
self.qr_dialog.show()
def close_qr(self):
self.qr_dialog.close()
def setting_callback(self):
setting = SettingWindow(parent=self)
setting.setStyleSheet(
"QDialog{border:none;border-radius:5px;background:#F3EAC2;}")
setting.show()
def gen_conf(self):
default_conf = {
"debug": False,
"hero": {
"title": "英雄篇",
"enabled": True,
"times": 1
},
"revival": {
"title": "复兴篇",
"enabled": True,
"times": 1
},
"creation": {
"title": "创新篇",
"enabled": True,
"times": 1
},
"belief": {
"title": "信念篇",
"enabled": True,
"times": 1
},
"limit_time": {
"title": "限时赛",
"enabled": True,
"times": 1
},
"rob": {
"title": "抢十赛",
"enabled": True,
"times": 1
}
}
with open(file="config.json", mode="w",
encoding="utf-8") as conf_writer:
conf_writer.write(
json.dumps(default_conf,
indent=4,
sort_keys=True,
ensure_ascii=False))
self.logger.info("已生成默认配置文件")
def mousePressEvent(self, event: QMouseEvent):
self.logger.debug("触发鼠标按压事件")
super().mousePressEvent(event)
self.setFocus()
self.m_flag = True
if event.button() == Qt.MouseButtons.LeftButton and self.isMaximized(
) == False and self.hasFocus() == True:
self.old_pos = event.globalPosition() #获取鼠标相对窗口的位置
self.logger.debug("已获取鼠标位置")
self.setCursor(QtGui.QCursor(
Qt.CursorShape.SizeAllCursor)) #更改鼠标图标
def mouseMoveEvent(self, event: QMouseEvent):
self.logger.debug("触发鼠标移动事件")
super().mouseMoveEvent(event)
if self.m_flag == True:
delta_x = int(event.globalPosition().x() - self.old_pos.x())
delta_y = int(event.globalPosition().y() - self.old_pos.y())
self.move(self.x() + delta_x, self.y() + delta_y) #更改窗口位置
self.logger.debug("已更改窗口位置")
self.old_pos = event.globalPosition()
def mouseReleaseEvent(self, event: QMouseEvent):
self.logger.debug("触发鼠标释放事件")
super().mouseReleaseEvent(event)
self.m_flag = False
self.setCursor(QtGui.QCursor(Qt.CursorShape.ArrowCursor))
def __init__(self, parent: QWidget):
super().__init__()
self.logger = logging.getLogger(__name__)
with open(file="config.json", mode="r",
encoding="utf-8") as conf_reader:
self.conf = json.loads(conf_reader.read())
self.logger.debug("初始化设置界面。设置内容：%s" % self.conf)
layout = QGridLayout()
self.setLayout(layout)
self.setModal(True)
self.setParent(parent)
self.resize(400, 300)
title = QLabel("设置")
title.setStyleSheet(
"QLabel{border:none;border-radius:5px;background:transparent;color:#9AD3BC;font-size:20px;}"
)
title.setAlignment(Qt.Alignment.AlignCenter)
layout.addWidget(title, 0, 1, Qt.Alignment.AlignCenter)
control_close = QPushButton()
control_close.setStyleSheet(
"QPushButton{background:#FFE3ED;border-radius:5px;border:none;}QPushButton:hover{background:#EC524B;}"
)
control_close.setToolTip("关闭")
control_close.setFixedHeight(20)
control_close.clicked.connect(self.close_callback)
layout.addWidget(control_close, 0, 0)
debug_check = QCheckBox("调试模式")
debug_check.setChecked(self.conf["debug"])
debug_check.setToolTip("单击切换开关状态")
debug_check.setStyleSheet(
"QCheckBox::indicator{width:10px;height:10px;border:none;border-radius:5px;background:#9BE3DE;}QCheckBox::indicator:unchecked{background:#BEEBE9;}QCheckBox::indicator:unchecked:hover{background:#9AD3BC;}QCheckBox::indicator:checked{background:#95E1D3;}QCheckBox::indicator:checked:hover{background:#98DED9;}"
)
self.content = QGridLayout()
(x, y) = self.show_setting(conf=self.conf,
layout=self.content) # 返回content的最后一个元素的x,y
proxy = QGroupBox()
proxy.setObjectName("proxy")
proxy_layout = QVBoxLayout()
proxy_label = QLabel("代理地址：")
proxy_label.setStyleSheet(
"QLabel{background:transparent;border:none;}")
proxy_input = EnhancedEdit()
proxy_input.setText(self.conf["proxy"])
proxy_input.setToolTip("格式为协议://IP:端口，留空保持直连")
proxy_input.setStyleSheet(
"QLineEdit{border:1px solid #F3EAC2;border-radius:5px;background:transparent;}QLineEdit:hover{border:1px solid #F5B461;}"
)
proxy_layout.addWidget(proxy_label)
proxy_layout.addWidget(proxy_input)
proxy.setLayout(proxy_layout)
proxy.setStyleSheet("QGroupBox{border-radius:5px;}")
proxy.setToolTip("代理设置")
if y + 1 >= 3:
y_ = 0
x_ = x + 1
else:
y_ = y + 1
x_ = x
self.content.addWidget(proxy, x_, y_)
self.content.addWidget(debug_check)
layout.addLayout(self.content, 1, 1)
class Ui_KeysPanel(object):
def setupUi(self, KeysWidget):
self.mainLayout = QGridLayout(KeysWidget)
self.buttons = []
self.exercises = []
if KeysWidget.classifyExercises is not None:
for ind in range(0, len(KeysWidget.classifyExercises.exercises)):
self.exercises.append(
KeysWidget.classifyExercises.exercises[ind])
self.createRow(
exercise=KeysWidget.classifyExercises.exercises[ind],
index=ind)
self.actions = QHBoxLayout()
self.supportedKeys = QPushButton('Supported keys')
self.supportedKeys.setIcon(QIcon.fromTheme("dialog-information"))
self.supportedKeys.clicked.connect(self.allKeyDialog)
self.saveProfile = QPushButton('Save keys')
self.actions.addWidget(self.supportedKeys)
self.actions.addWidget(self.saveProfile)
self.mainLayout.addLayout(self.actions, self.mainLayout.rowCount(), 0)
def createRow(self, exercise, index):
item = QHBoxLayout()
label = QLabel(exercise.name)
item.addWidget(label)
item.setAlignment(label, Qt.Alignment.AlignCenter)
button = QPushButton(exercise.assigned_key[0])
# button.installEventFilter(self)
button.setStyleSheet(""" QPushButton
{
border: 1px solid grey;
background-color: white;
}
""")
button.clicked.connect(partial(self.openDialog, button))
self.buttons.append(button)
item.addWidget(button)
self.mainLayout.addLayout(item, index, 0)
print(index, exercise.name, exercise.assigned_key)
def openDialog(self, button):
print(button)
index = self.buttons.index(button)
button.setStyleSheet(""" QPushButton
{
border: 1px solid #B8B8B8;
background-color: grey;
}
""")
dialog = ChangeKeyDialog(buttons=self.buttons,
exercises=self.exercises,
index=index)
dialog.setAttribute(QtCore.Qt.WidgetAttribute.WA_DeleteOnClose)
dialog.exec()
# if button is QPushButton:
def allKeyDialog(self):
dialog = AllKeysDialog()
dialog.exec()
class DataGUI(QWidget):
def __init__(self):
super().__init__()
self.experiment_file_name = None
self.experiment_file_directory = None
self.data_directory = None
self.max_rois = 50
self.roi_type = 'freehand'
self.roi_radius = None
self.existing_roi_set_paths = {}
self.current_roi_index = 0
self.current_z_slice = 0
self.current_channel = 1 # index
self.image_series_name = ''
self.series_number = None
self.roi_response = []
self.roi_mask = []
self.roi_path = []
self.roi_image = None
self.roi_path_list = []
self.blank_image = np.zeros((1, 1))
self.colors = [
mcolors.to_rgb(x) for x in list(mcolors.TABLEAU_COLORS)[:20]
]
self.initUI()
def initUI(self):
self.grid = QGridLayout(self)
self.file_control_grid = QGridLayout()
self.file_control_grid.setSpacing(3)
self.grid.addLayout(self.file_control_grid, 0, 0)
self.file_tree_grid = QGridLayout()
self.file_tree_grid.setSpacing(3)
self.grid.addLayout(self.file_tree_grid, 1, 0)
self.group_control_grid = QGridLayout()
self.group_control_grid.setSpacing(3)
self.grid.addLayout(self.group_control_grid, 0, 1)
self.attribute_grid = QGridLayout()
self.attribute_grid.setSpacing(3)
self.grid.addLayout(self.attribute_grid, 1, 1)
self.roi_control_grid = QGridLayout()
self.roi_control_grid.setSpacing(3)
self.grid.addLayout(self.roi_control_grid, 0, 2)
self.plot_grid = QGridLayout()
self.plot_grid.setSpacing(3)
self.grid.addLayout(self.plot_grid, 1, 2)
# # # # File control browser: # # # # # # # # (0,0)
loadButton = QPushButton("Load expt. file", self)
loadButton.clicked.connect(self.selectDataFile)
# Label with current expt file
self.currentExperimentLabel = QLabel('')
self.file_control_grid.addWidget(loadButton, 0, 0)
self.file_control_grid.addWidget(self.currentExperimentLabel, 1, 0)
directoryButton = QPushButton("Select data directory", self)
directoryButton.clicked.connect(self.selectDataDirectory)
self.file_control_grid.addWidget(directoryButton, 0, 1)
self.data_directory_display = QLabel('')
self.data_directory_display.setFont(QtGui.QFont('SansSerif', 8))
self.file_control_grid.addWidget(self.data_directory_display, 1, 1)
# Attach metadata to file
attachDatabutton = QPushButton("Attach metadata to file", self)
attachDatabutton.clicked.connect(self.attachData)
self.file_control_grid.addWidget(attachDatabutton, 2, 0, 1, 2)
# Select image data file
selectImageDataFileButton = QPushButton("Select image data file", self)
selectImageDataFileButton.clicked.connect(self.selectImageDataFile)
self.file_control_grid.addWidget(selectImageDataFileButton, 3, 0, 1, 2)
# # # # File tree: # # # # # # # # (1,0)
self.groupTree = QTreeWidget(self)
self.groupTree.setHeaderHidden(True)
self.groupTree.itemClicked.connect(self.onTreeItemClicked)
self.file_tree_grid.addWidget(self.groupTree, 3, 0, 2, 7)
# # # # Group control: # # # # # # # # (0, 1)
deleteGroupButton = QPushButton("Delete selected group", self)
deleteGroupButton.clicked.connect(self.deleteSelectedGroup)
self.group_control_grid.addWidget(deleteGroupButton, 0, 0, 1, 2)
# File name display
self.currentImageFileNameLabel = QLabel('')
self.group_control_grid.addWidget(self.currentImageFileNameLabel, 1, 0)
# Channel drop down
ch_label = QLabel('Channel:')
self.ChannelComboBox = QComboBox(self)
self.ChannelComboBox.addItem("1")
self.ChannelComboBox.addItem("0")
self.ChannelComboBox.activated.connect(self.selectChannel)
self.group_control_grid.addWidget(ch_label, 2, 0)
self.group_control_grid.addWidget(self.ChannelComboBox, 2, 1)
# # # # Attribute table: # # # # # # # # (1, 1)
self.tableAttributes = QTableWidget()
self.tableAttributes.setStyleSheet("")
self.tableAttributes.setColumnCount(2)
self.tableAttributes.setObjectName("tableAttributes")
self.tableAttributes.setRowCount(0)
item = QTableWidgetItem()
font = QtGui.QFont()
font.setPointSize(10)
item.setFont(font)
item.setBackground(QtGui.QColor(121, 121, 121))
brush = QtGui.QBrush(QtGui.QColor(91, 91, 91))
brush.setStyle(QtCore.Qt.BrushStyle.SolidPattern)
item.setForeground(brush)
self.tableAttributes.setHorizontalHeaderItem(0, item)
item = QTableWidgetItem()
item.setBackground(QtGui.QColor(123, 123, 123))
brush = QtGui.QBrush(QtGui.QColor(91, 91, 91))
brush.setStyle(QtCore.Qt.BrushStyle.SolidPattern)
item.setForeground(brush)
self.tableAttributes.setHorizontalHeaderItem(1, item)
self.tableAttributes.horizontalHeader().setCascadingSectionResizes(
True)
self.tableAttributes.horizontalHeader().setHighlightSections(False)
self.tableAttributes.horizontalHeader().setSortIndicatorShown(True)
self.tableAttributes.horizontalHeader().setStretchLastSection(True)
self.tableAttributes.verticalHeader().setVisible(False)
self.tableAttributes.verticalHeader().setHighlightSections(False)
item = self.tableAttributes.horizontalHeaderItem(0)
item.setText("Attribute")
item = self.tableAttributes.horizontalHeaderItem(1)
item.setText("Value")
self.tableAttributes.itemChanged.connect(self.update_attrs_to_file)
self.attribute_grid.addWidget(self.tableAttributes, 3, 0, 1, 8)
# # # # Roi control # # # # # # # # (0, 2)
# ROI type drop-down
self.RoiTypeComboBox = QComboBox(self)
self.RoiTypeComboBox.addItem("freehand")
radii = [1, 2, 3, 4, 6, 8]
for radius in radii:
self.RoiTypeComboBox.addItem("circle:" + str(radius))
self.RoiTypeComboBox.activated.connect(self.selectRoiType)
self.roi_control_grid.addWidget(self.RoiTypeComboBox, 0, 0)
# Clear all ROIs button
self.clearROIsButton = QPushButton("Clear ROIs", self)
self.clearROIsButton.clicked.connect(self.clearRois)
self.roi_control_grid.addWidget(self.clearROIsButton, 0, 2)
# Response display type dropdown
self.RoiResponseTypeComboBox = QComboBox(self)
self.RoiResponseTypeComboBox.addItem("RawTrace")
self.RoiResponseTypeComboBox.addItem("TrialAverage")
self.RoiResponseTypeComboBox.addItem("TrialResponses")
self.RoiResponseTypeComboBox.addItem("TrialAverageDFF")
self.roi_control_grid.addWidget(self.RoiResponseTypeComboBox, 2, 2)
# ROIset file name line edit box
self.defaultRoiSetName = "roi_set_name"
self.le_roiSetName = QLineEdit(self.defaultRoiSetName)
self.roi_control_grid.addWidget(self.le_roiSetName, 1, 1)
# Save ROIs button
self.saveROIsButton = QPushButton("Save ROIs", self)
self.saveROIsButton.clicked.connect(self.saveRois)
self.roi_control_grid.addWidget(self.saveROIsButton, 1, 0)
# Load ROI set combobox
self.loadROIsComboBox = QComboBox(self)
self.loadROIsComboBox.addItem("(load existing ROI set)")
self.loadROIsComboBox.activated.connect(self.selectedExistingRoiSet)
self.roi_control_grid.addWidget(self.loadROIsComboBox, 1, 2)
self.updateExistingRoiSetList()
# Delete current roi button
self.deleteROIButton = QPushButton("Delete ROI", self)
self.deleteROIButton.clicked.connect(self.deleteRoi)
self.roi_control_grid.addWidget(self.deleteROIButton, 2, 0)
# Current roi slider
self.roiSlider = QSlider(QtCore.Qt.Orientation.Horizontal, self)
self.roiSlider.setMinimum(0)
self.roiSlider.setMaximum(self.max_rois)
self.roiSlider.valueChanged.connect(self.sliderUpdated)
self.roi_control_grid.addWidget(self.roiSlider, 2, 1, 1, 1)
ctx = plt.rc_context({
'xtick.major.size': 1,
'axes.spines.top': False,
'axes.spines.right': False,
'xtick.labelsize': 'xx-small',
'ytick.labelsize': 'xx-small',
'xtick.major.size': 1.0,
'ytick.major.size': 1.0,
'xtick.major.pad': 1.0,
'ytick.major.pad': 1.0
})
with ctx:
self.responseFig = plt.figure(frameon=False, layout='constrained')
self.responsePlot = self.responseFig.add_subplot(111)
self.responseCanvas = FigureCanvas(self.responseFig)
self.responseCanvas.draw_idle()
self.plot_grid.addWidget(self.responseCanvas, 0, 0)
# # # # Image canvas # # # # # # # # (1, 2)
self.roi_fig = plt.figure()
self.roi_ax = self.roi_fig.add_subplot(111)
self.roi_canvas = FigureCanvas(self.roi_fig)
self.toolbar = NavigationToolbar(self.roi_canvas, self)
self.roi_ax.set_aspect('equal')
self.roi_ax.set_axis_off()
self.plot_grid.addWidget(self.toolbar, 1, 0)
self.plot_grid.addWidget(self.roi_canvas, 2, 0)
self.plot_grid.setRowStretch(0, 1)
self.plot_grid.setRowStretch(1, 3)
self.plot_grid.setRowStretch(2, 3)
# Current z slice slider
self.zSlider = QSlider(QtCore.Qt.Orientation.Horizontal, self)
self.zSlider.setMinimum(0)
self.zSlider.setMaximum(50)
self.zSlider.setValue(0)
self.zSlider.valueChanged.connect(self.zSliderUpdated)
self.plot_grid.addWidget(self.zSlider, 3, 0)
self.roi_fig.tight_layout()
self.setWindowTitle('Visanalysis')
self.setGeometry(200, 200, 1200, 600)
self.show()
def _populateTree(self, widget, dict):
widget.clear()
self.fill_item(widget.invisibleRootItem(), dict)
def fill_item(self, item, value):
item.setExpanded(True)
if type(value) is dict:
for key, val in sorted(value.items()):
child = QTreeWidgetItem()
child.setText(0, key)
item.addChild(child)
self.fill_item(child, val)
elif type(value) is list:
for val in value:
child = QTreeWidgetItem()
item.addChild(child)
if type(val) is dict:
child.setText(0, '[dict]')
self.fill_item(child, val)
elif type(val) is list:
child.setText(0, '[list]')
self.fill_item(child, val)
else:
child.setText(0, val)
child.setExpanded(True)
else:
child = QTreeWidgetItem()
child.setText(0, value)
item.addChild(child)
def onTreeItemClicked(self, item, column):
file_path = os.path.join(self.experiment_file_directory,
self.experiment_file_name + '.hdf5')
group_path = h5io.getPathFromTreeItem(
self.groupTree.selectedItems()[0])
self.clearRois()
self.series_number = None
if 'series_' in group_path:
self.series_number = int(
group_path.split('series_')[-1].split('/')[0])
if self.plugin.dataIsAttached(file_path, self.series_number):
self.plugin.updateImagingDataObject(
self.experiment_file_directory, self.experiment_file_name,
self.series_number)
# look for image_file_name or ask user to select it
if self.data_directory is not None:
image_file_name = h5io.readImageFileName(
file_path, self.series_number)
if image_file_name is None or image_file_name == '':
image_file_path, _ = QFileDialog.getOpenFileName(
self, "Select image file")
print('User selected image file at {}'.format(
image_file_path))
image_file_name = os.path.split(image_file_path)[-1]
self.data_directory = os.path.split(
image_file_path)[:-1][0]
h5io.attachImageFileName(file_path, self.series_number,
image_file_name)
print('Attached image_file_name {} to series {}'.format(
image_file_name, self.series_number))
print('Data directory is {}'.format(self.data_directory))
self.image_file_name = image_file_name
self.currentImageFileNameLabel.setText(self.image_file_name)
else: # clicked part of the tree upstream of any series
self.series_number = None
if item.parent() is not None:
if item.parent().text(
column) == 'rois': # selected existing roi group
roi_set_name = item.text(column)
# print('Selected roi set {} from series {}'.format(roi_set_name, self.series_number))
self.le_roiSetName.setText(roi_set_name)
roi_set_path = h5io.getPathFromTreeItem(
self.groupTree.selectedItems()[0])
self.loadRois(roi_set_path)
self.redrawRoiTraces()
if group_path != '':
attr_dict = h5io.getAttributesFromGroup(file_path, group_path)
editable_values = True # user can edit metadata
self.populate_attrs(attr_dict=attr_dict,
editable_values=editable_values)
# show roi image
if self.series_number is not None: # Clicked on node of the tree associated with a single series
if self.data_directory is not None: # user has selected a raw data directory
if self.plugin.dataIsAttached(file_path, self.series_number):
self.plugin.updateImageSeries(
data_directory=self.data_directory,
image_file_name=self.image_file_name,
series_number=self.series_number,
channel=self.current_channel)
self.roi_image = self.plugin.mean_brain
self.zSlider.setValue(0)
self.zSlider.setMaximum(self.roi_image.shape[2] - 1)
self.redrawRoiTraces()
else:
print('Attach metadata to file before drawing rois')
else:
print('Select a data directory before drawing rois')
# # # TEST # # #
memory_usage = psutil.Process(os.getpid()).memory_info().rss * 10**-9
print('Current Memory Usage: {:.2f}GB'.format(memory_usage))
sys.stdout.flush()
# # # TEST # # #
def updateExistingRoiSetList(self):
if self.experiment_file_name is not None:
file_path = os.path.join(self.experiment_file_directory,
self.experiment_file_name + '.hdf5')
self.existing_roi_set_paths = self.plugin.getRoiSetPaths(
file_path) # dictionary of name: full path
self.loadROIsComboBox.clear()
for r_path in self.existing_roi_set_paths:
self.loadROIsComboBox.addItem(r_path)
self.show()
def selectedExistingRoiSet(self):
file_path = os.path.join(self.experiment_file_directory,
self.experiment_file_name + '.hdf5')
roi_set_key = self.loadROIsComboBox.currentText()
roi_set_path = self.existing_roi_set_paths[roi_set_key]
_, _, self.roi_path, self.roi_mask = self.plugin.loadRoiSet(
file_path, roi_set_path)
if self.series_number is not None:
self.roi_response = []
for new_path in self.roi_path:
new_roi_resp = self.plugin.getRoiDataFromPath(
roi_path=new_path)
self.roi_response.append(new_roi_resp)
# update slider to show most recently drawn roi response
self.current_roi_index = len(self.roi_response) - 1
self.roiSlider.setValue(self.current_roi_index)
# Update figures
self.redrawRoiTraces()
def selectDataFile(self):
filePath, _ = QFileDialog.getOpenFileName(
self, "Open experiment (hdf5) file")
self.experiment_file_name = os.path.split(filePath)[1].split('.')[0]
self.experiment_file_directory = os.path.split(filePath)[0]
if self.experiment_file_name != '':
self.currentExperimentLabel.setText(self.experiment_file_name)
self.initializeDataAnalysis()
self.populateGroups()
self.updateExistingRoiSetList()
def selectDataDirectory(self):
filePath = str(
QFileDialog.getExistingDirectory(self, "Select data directory"))
self.data_directory = filePath
self.data_directory_display.setText('..' + self.data_directory[-24:])
def initializeDataAnalysis(self):
file_path = os.path.join(self.experiment_file_directory,
self.experiment_file_name + '.hdf5')
data_type = h5io.getDataType(file_path)
# Load plugin based on Rig name in hdf5 file
if data_type == 'Bruker':
from visanalysis.plugin import bruker
self.plugin = bruker.BrukerPlugin()
elif data_type == 'AODscope':
from visanalysis.plugin import aodscope
self.plugin = aodscope.AodScopePlugin()
else:
self.plugin = h5io.BasePlugin()
self.plugin.parent_gui = self
# # # TEST # # #
memory_usage = psutil.Process(os.getpid()).memory_info().rss * 10**-9
print('Current memory usage: {:.2f}GB'.format(memory_usage))
sys.stdout.flush()
# # # TEST # # #
def attachData(self):
if self.data_directory is not None:
file_path = os.path.join(self.experiment_file_directory,
self.experiment_file_name + '.hdf5')
self.plugin.attachData(self.experiment_file_name, file_path,
self.data_directory)
print('Data attached')
else:
print('Select a data directory before attaching new data')
def selectImageDataFile(self):
file_path = os.path.join(self.experiment_file_directory,
self.experiment_file_name + '.hdf5')
image_file_path, _ = QFileDialog.getOpenFileName(
self, "Select image file")
print('User selected image file at {}'.format(image_file_path))
self.image_file_name = os.path.split(image_file_path)[-1]
self.data_directory = os.path.split(image_file_path)[:-1][0]
h5io.attachImageFileName(file_path, self.series_number,
self.image_file_name)
print('Attached image_file_name {} to series {}'.format(
self.image_file_name, self.series_number))
print('Data directory is {}'.format(self.data_directory))
self.currentImageFileNameLabel.setText(self.image_file_name)
# show roi image
if self.series_number is not None:
if self.data_directory is not None: # user has selected a raw data directory
self.plugin.updateImageSeries(
data_directory=self.data_directory,
image_file_name=self.image_file_name,
series_number=self.series_number,
channel=self.current_channel)
self.roi_image = self.plugin.mean_brain
self.zSlider.setValue(0)
self.zSlider.setMaximum(self.roi_image.shape[2] - 1)
self.redrawRoiTraces()
else:
print('Select a data directory before drawing rois')
def deleteSelectedGroup(self):
file_path = os.path.join(self.experiment_file_directory,
self.experiment_file_name + '.hdf5')
group_path = h5io.getPathFromTreeItem(
self.groupTree.selectedItems()[0])
group_name = group_path.split('/')[-1]
buttonReply = QMessageBox.question(
self, 'Delete series',
"Are you sure you want to delete group {}?".format(group_name),
QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No,
QMessageBox.StandardButton.No)
if buttonReply == QMessageBox.StandardButton.Yes:
h5io.deleteGroup(file_path=file_path, group_path=group_path)
print('Deleted group {}'.format(group_name))
self.updateExistingRoiSetList()
self.populateGroups()
else:
print('Delete aborted')
def populateGroups(self):
file_path = os.path.join(self.experiment_file_directory,
self.experiment_file_name + '.hdf5')
self.group_dset_dict = h5io.getHierarchy(file_path)
self._populateTree(self.groupTree, self.group_dset_dict)
def populate_attrs(self, attr_dict=None, editable_values=False):
"""Populate attribute for currently selected group."""
self.tableAttributes.blockSignals(
True) # block udpate signals for auto-filled forms
self.tableAttributes.setRowCount(0)
self.tableAttributes.setColumnCount(2)
self.tableAttributes.setSortingEnabled(False)
if attr_dict:
for num, key in enumerate(attr_dict):
self.tableAttributes.insertRow(self.tableAttributes.rowCount())
key_item = QTableWidgetItem(key)
key_item.setFlags(QtCore.Qt.ItemFlag.ItemIsSelectable
| QtCore.Qt.ItemFlag.ItemIsEnabled)
self.tableAttributes.setItem(num, 0, key_item)
val_item = QTableWidgetItem(str(attr_dict[key]))
if editable_values:
val_item.setFlags(QtCore.Qt.ItemFlag.ItemIsSelectable
| QtCore.Qt.ItemFlag.ItemIsEditable
| QtCore.Qt.ItemFlag.ItemIsEnabled)
else:
val_item.setFlags(QtCore.Qt.ItemFlag.ItemIsSelectable
| QtCore.Qt.ItemFlag.ItemIsEnabled)
self.tableAttributes.setItem(num, 1, val_item)
self.tableAttributes.blockSignals(False)
def update_attrs_to_file(self, item):
file_path = os.path.join(self.experiment_file_directory,
self.experiment_file_name + '.hdf5')
group_path = h5io.getPathFromTreeItem(
self.groupTree.selectedItems()[0])
attr_key = self.tableAttributes.item(item.row(), 0).text()
attr_val = item.text()
# update attr in file
h5io.changeAttribute(file_path=file_path,
group_path=group_path,
attr_key=attr_key,
attr_val=attr_val)
print('Changed attr {} to = {}'.format(attr_key, attr_val))
# %% # # # # # # # # ROI SELECTOR WIDGET # # # # # # # # # # # # # # # # # # #
def refreshLassoWidget(self, keep_paths=False):
self.roi_ax.clear()
init_lasso = False
if self.roi_image is not None:
if len(self.roi_mask) > 0:
newImage = plot_tools.overlayImage(
self.roi_image[:, :, self.current_z_slice],
self.roi_mask,
0.5,
self.colors,
z=self.current_z_slice)
else:
newImage = self.roi_image[:, :, self.current_z_slice]
self.roi_ax.imshow(newImage, cmap=cm.gray)
init_lasso = True
else:
self.roi_ax.imshow(self.blank_image)
self.roi_ax.set_axis_off()
self.roi_canvas.draw()
if not keep_paths:
self.roi_path_list = []
if init_lasso:
if self.roi_type == 'circle':
self.lasso_1 = EllipseSelector(self.roi_ax,
onselect=self.newEllipse,
button=1)
elif self.roi_type == 'freehand':
self.lasso_1 = LassoSelector(self.roi_ax,
onselect=self.newFreehand,
button=1)
self.lasso_2 = LassoSelector(self.roi_ax,
onselect=self.appendFreehand,
button=3)
else:
print(
'Warning ROI type not recognized. Choose circle or freehand'
)
def newFreehand(self, verts):
new_roi_path = path.Path(verts)
new_roi_path.z_level = self.zSlider.value()
new_roi_path.channel = self.current_channel
self.updateRoiSelection([new_roi_path])
def appendFreehand(self, verts):
print('Appending rois, hit Enter/Return to finish')
new_roi_path = path.Path(verts)
new_roi_path.z_level = self.zSlider.value()
new_roi_path.channel = self.current_channel
self.roi_path_list.append(new_roi_path)
def keyPressEvent(self, event):
if type(event) == QtGui.QKeyEvent:
if np.any([
event.key() == QtCore.Qt.Key.Key_Return,
event.key() == QtCore.Qt.Key.Key_Enter
]):
if len(self.roi_path_list) > 0:
event.accept()
self.updateRoiSelection(self.roi_path_list)
else:
event.ignore()
else:
event.ignore()
else:
event.ignore()
def newEllipse(self, pos1, pos2, definedRadius=None):
x1 = np.round(pos1.xdata)
x2 = np.round(pos2.xdata)
y1 = np.round(pos1.ydata)
y2 = np.round(pos2.ydata)
radiusX = np.sqrt((x1 - x2)**2) / 2
radiusY = np.sqrt((y1 - y2)**2) / 2
if self.roi_radius is not None:
radiusX = self.roi_radius
center = (np.round((x1 + x2) / 2), np.round((y1 + y2) / 2))
new_roi_path = path.Path.circle(center=center, radius=radiusX)
new_roi_path.z_level = self.zSlider.value()
new_roi_path.channel = self.current_channel
self.updateRoiSelection([new_roi_path])
def updateRoiSelection(self, new_roi_path):
mask = self.plugin.getRoiMaskFromPath(new_roi_path)
new_roi_resp = self.plugin.getRoiDataFromPath(roi_path=new_roi_path)
if mask.sum() == 0:
print('No pixels in the roi you just drew')
return
# update list of roi data
self.roi_mask.append(mask)
self.roi_path.append(new_roi_path) # list of lists of paths
self.roi_response.append(new_roi_resp)
# update slider to show most recently drawn roi response
self.current_roi_index = len(self.roi_response) - 1
self.roiSlider.setValue(self.current_roi_index)
# Update figures
self.redrawRoiTraces()
def sliderUpdated(self):
self.current_roi_index = self.roiSlider.value()
self.redrawRoiTraces()
def zSliderUpdated(self):
self.current_z_slice = self.zSlider.value()
if self.roi_image is not None:
self.refreshLassoWidget(keep_paths=True)
def redrawRoiTraces(self):
self.clearRoiArtists()
if self.current_roi_index < len(self.roi_response):
current_raw_trace = np.squeeze(
self.roi_response[self.current_roi_index])
fxn_name = self.RoiResponseTypeComboBox.currentText()
display_trace = getattr(self.plugin,
'getRoiResponse_{}'.format(fxn_name))(
[current_raw_trace])
self.responsePlot.plot(display_trace,
color=self.colors[self.current_roi_index],
linewidth=1,
alpha=0.5)
self.responsePlot.set_xlim([0, len(display_trace)])
y_min = np.nanmin(display_trace)
y_max = np.nanmax(display_trace)
self.responsePlot.set_ylim([y_min, y_max])
self.responseCanvas.draw()
self.refreshLassoWidget(keep_paths=False)
# %% # # # # # # # # LOADING / SAVING / COMPUTING ROIS # # # # # # # # # # # # # # # # # # #
def loadRois(self, roi_set_path):
file_path = os.path.join(self.experiment_file_directory,
self.experiment_file_name + '.hdf5')
self.roi_response, self.roi_image, self.roi_path, self.roi_mask = self.plugin.loadRoiSet(
file_path, roi_set_path)
self.zSlider.setValue(0)
self.zSlider.setMaximum(self.roi_image.shape[2] - 1)
def saveRois(self):
file_path = os.path.join(self.experiment_file_directory,
self.experiment_file_name + '.hdf5')
roi_set_name = self.le_roiSetName.text()
if roi_set_name in h5io.getAvailableRoiSetNames(
file_path, self.series_number):
buttonReply = QMessageBox.question(
self, 'Overwrite roi set',
"Are you sure you want to overwrite roi set: {}?".format(
roi_set_name),
QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No,
QMessageBox.StandardButton.No)
if buttonReply == QMessageBox.StandardButton.Yes:
self.plugin.saveRoiSet(file_path,
series_number=self.series_number,
roi_set_name=roi_set_name,
roi_mask=self.roi_mask,
roi_response=self.roi_response,
roi_image=self.roi_image,
roi_path=self.roi_path)
print('Saved roi set {} to series {}'.format(
roi_set_name, self.series_number))
self.populateGroups()
self.updateExistingRoiSetList()
else:
print('Overwrite aborted - pick a unique roi set name')
else:
self.plugin.saveRoiSet(file_path,
series_number=self.series_number,
roi_set_name=roi_set_name,
roi_mask=self.roi_mask,
roi_response=self.roi_response,
roi_image=self.roi_image,
roi_path=self.roi_path)
print('Saved roi set {} to series {}'.format(
roi_set_name, self.series_number))
self.populateGroups()
self.updateExistingRoiSetList()
def deleteRoi(self):
if self.current_roi_index < len(self.roi_response):
self.roi_mask.pop(self.current_roi_index)
self.roi_response.pop(self.current_roi_index)
self.roi_path.pop(self.current_roi_index)
self.roiSlider.setValue(self.current_roi_index - 1)
self.redrawRoiTraces()
def clearRois(self):
self.roi_mask = []
self.roi_response = []
self.roi_path = []
self.roi_image = None
self.clearRoiArtists()
self.redrawRoiTraces()
self.roi_ax.clear()
def clearRoiArtists(self):
for artist in self.responsePlot.lines + self.responsePlot.collections:
artist.remove()
def selectRoiType(self):
self.roi_type = self.RoiTypeComboBox.currentText().split(':')[0]
if 'circle' in self.RoiTypeComboBox.currentText():
self.roi_radius = int(
self.RoiTypeComboBox.currentText().split(':')[1])
else:
self.roi_radius = None
self.redrawRoiTraces()
def selectChannel(self):
self.current_channel = int(self.ChannelComboBox.currentText())
# show roi image
if self.series_number is not None:
if self.data_directory is not None: # user has selected a raw data directory
self.plugin.updateImageSeries(
data_directory=self.data_directory,
image_file_name=self.image_file_name,
series_number=self.series_number,
channel=self.current_channel)
self.roi_image = self.plugin.mean_brain
self.zSlider.setValue(0)
self.zSlider.setMaximum(self.roi_image.shape[2] - 1)
self.redrawRoiTraces()
else:
print('Select a data directory before drawing rois')
class MainWindow(QWidget):
def reload(self):
if len(self.file_list): return
self.webEngineView.reload()
def on_downloadRequestState(self,event):
if not len(self.file_list):
self.label.setText('Статус:...')
self.reload()
try:
for b in self.buttons: b.setEnabled(True)
except: pass
return
if event.value>1:
if event.value==2:
if len(self.file_list[-1])>5:
h0=self.file_list[-1][6]
h1=gethash(self.file_list[-1][-1])
if h0==h1:
try:
tm=mktime(datetime.strptime(self.file_list[-1][4],"%d.%m.%Y %H:%M").timetuple())
tm+=(int(self.file_list[-1][5])%60)
utime(self.file_list[-1][-1],(tm,tm))
except: pass
else:
print('Error. ("%s") - %i'%(self.file_list[-1][-1],event.value))
self.label.setText('Статус: ошибка скачивания файла ("'+self.file_list[-1][-1]+'")')
self.file_list.pop()
self.download_files()
else:
print('Error. ("%s") - %i'%(self.file_list[-1][-1],event.value))
self.label.setText('Статус: ошибка скачивания файла ("%s", код ошибки: %i)!'%(self.file_list[-1][-1],event.value))
self.reload()
try:
for b in self.buttons: b.setEnabled(True)
except: pass
def check_canceled(self,event):
if event: self.canceled_callback()
def on_downloadRequested(self,event):
self.webEngineView.page().runJavaScript(\
"""
function check_canceled(){
if(window.process_canceled) return 1;
else return 0;
};
check_canceled();
""",self.check_canceled)
if event.state().value==0:
event.stateChanged.connect(lambda event: self.on_downloadRequestState(event))
event.accept()
def download_files(self):
if len(self.file_list):
if not self.file_number: self.file_number=len(self.file_list)
tmp=self.file_list[-1]
if exists(tmp[-1]):
try: remove(tmp[-1])
except: pass
self.webEngineView.page().profile().downloadRequested.connect(lambda event:\
self.on_downloadRequested(event))
self.label.setText('Статус: скачивание файла "%s" ( %i из %i )'%\
(tmp[2],self.file_number-len(self.file_list)+1,self.file_number))
self.webEngineView.page().download(QUrl(tmp[3]),tmp[-1])
else:
self.webEngineView.page().profile().downloadRequested.disconnect()
self.label.setText('Статус:...')
self.reload()
try:
for b in self.buttons: b.setEnabled(True)
except: pass
def canceled_callback(self):
self.label.setText('Статус: операция отменена!')
self.file_list=[]
self.file_number=0
self.reload()
try:
for b in self.buttons: b.setEnabled(True)
except: pass
def getfilelist_error_callback(self):
self.label.setText('Статус: ошибка при получении списка файлов для скачивания!')
self.file_list=[]
self.file_number=0
self.reload()
try:
for b in self.buttons: b.setEnabled(True)
except: pass
def getfilelist_success_callback(self,links):
for link in links:
path=join(self.basepath,link[0],link[1])
try: makedirs(path,exist_ok=True)
except: return
link.append(realpath(join(path,link[2])))
self.file_list.append(link)
if len(self.file_list):
self.download_files()
def progress_callback(self,counter):
self.label.setText('Статус: получение списка файлов для скачивания (получено %d ссылок)...'%counter)
def getfileslist(self):
if len(self.file_list): return
self.url=self.webEngineView.url().toString()
if not('Files/prt0' in self.url): return
self.file_list=[]
self.file_number=0
try:
for b in self.buttons: b.setEnabled(False)
except: pass
self.label.setText('Статус: получение списка файлов для скачивания...')
self.webEngineView.page().runJavaScript(\
self.qwebchannel_js+self.busy+"""
function get_sig_links_details_parse(file_links,details_links,partitions,partition,section,data){
if(window.process_canceled){
cancel();
return;
};
var parser=new DOMParser();
var doc=parser.parseFromString(data,'text/html');
var req=doc.querySelectorAll('a[href*=\"GetFile\"]');
for(const e of req){
if(e.text.includes('.sig')){
file_links.push([pd_replace(partition),sec_replace(section),e.text,e.href]);
progress(file_links);
};
};
if(details_links.length) get_sig_links_details(file_links,details_links,partitions);
else success(file_links);
};
function start_parser(){
var file_links=[];
var details_links=[];
var url=window.location.href.toString();
var path='';
if(url.includes('ProjectDocuments')) path='ProjectDocuments';
else if(url.includes('RIIDocuments')) path='RIIDocuments';
else if(url.includes('SmetaDocuments')) path='SmetaDocuments';
else if(url.includes('IrdDocuments')) path='IrdDocuments';
var href=''.concat('https://lk.spbexp.ru/Grid/',path,'FilesRead/own',
window.location.href.toString().split('\/').pop());
var v=document.querySelector('span[style=\"font-weight:600; color:darkblue\"]');\
if(v==null) return;
var section=v.textContent;
var links=[[path,section,href]];
new QWebChannel(qt.webChannelTransport,(channel)=>{
window.qtwebchannel=channel.objects.backend;
});
get_pd_links_details(links,file_links,details_links,[]);
};
start_parser();
""",lambda x: None)
def getallfileslist(self):
if len(self.file_list): return
self.url=self.webEngineView.url().toString()
if not('lk.spbexp.ru/Zeg/Zegmain' in self.url or\
('lk.spbexp.ru/SF/' in self.url and '/prt0/' in self.url)): return
self.file_list=[]
self.file_number=0
try:
for b in self.buttons: b.setEnabled(False)
except: pass
self.label.setText('Статус: получение списка файлов для скачивания...')
self.webEngineView.page().runJavaScript(\
self.qwebchannel_js+self.busy+"""
function get_sig_links_details_parse(file_links,details_links,partitions,partition,section,data){
if(window.process_canceled){
cancel();
return;
};
var parser=new DOMParser();
var doc=parser.parseFromString(data,'text/html');
var req=doc.querySelectorAll('a[href*=\"GetFile\"]');
for(const e of req){
if(e.text.includes('.sig')){
file_links.push([pd_replace(partition),sec_replace(section),e.text,e.href]);
progress(file_links);
};
};
if(details_links.length) get_sig_links_details(file_links,details_links,partitions);
else get_file_links(file_links,details_links,partitions);
};
function get_pd_links_parse(path,file_links,details_links,partitions,data){
if(window.process_canceled){
cancel();
return;
};
var links=[];
for(const d of data.Data){
if(d['NumberOfFiles']){
var href=''.concat('https://lk.spbexp.ru/Grid/',path,'FilesRead/own',d['IDRow']);
if(path.includes('IrdDocuments')) links.push([path,d['Content'],href]);
else links.push([path,d['Nazvanie'],href]);
};
};
if(links.length) get_pd_links_details(links,file_links,details_links,partitions);
};
async function get_pd_links(path,file_links,details_links,partitions){
if(window.process_canceled){
cancel();
return;
};
var req=document.querySelector('a[href*="/Zeg/Zegmain1"]');
if(req){
var href=''.concat('https://lk.spbexp.ru/Grid/',path,'Read/own',req.pathname.split('\/').pop());
return await fetch(href)
.then(async (response) => {
const j=await response.json();
get_pd_links_parse(path,file_links,details_links,partitions,j);
}).catch((error) => { errlog(error); });
};
};
function get_file_links(file_links,details_links,partitions){
if(window.process_canceled){
cancel();
return;
};
if(partitions.length) get_pd_links(partitions.pop(),file_links,details_links,partitions);
else success(file_links);
};
function start_parser(){
var file_links=[];
var details_links=[];
var partitions=['ProjectDocuments','RIIDocuments','SmetaDocuments','IrdDocuments'];
new QWebChannel(qt.webChannelTransport,(channel)=>{
window.qtwebchannel=channel.objects.backend;
});
get_file_links(file_links,details_links,partitions);
};
start_parser();
""",lambda x: None)
def __init__(self):
super().__init__()
self.setWindowTitle('ExpGet')
self.busy="""
window.process_canceled=false;
function errlog(error){
window.qtwebchannel.backrun_error(error);
};
function success(file_links){
window.qtwebchannel.backrun_success(file_links);
};
function progress(file_links){
window.qtwebchannel.backrun_progress(file_links.length);
};
async function cancel(){
window.qtwebchannel.backrun_cancel();
};
function pd_replace(partition){
switch(partition){
case 'ProjectDocuments': return 'ПД';
case 'RIIDocuments': return 'РИИ';
case 'SmetaDocuments': return 'СД';
case 'IrdDocuments': return 'ИРД';
};
};
function sec_replace(section){
return section.trim().replaceAll('"','_').slice(0,64).trim();
};
async function get_sig_links_details(file_links,details_links,partitions){
if(window.process_canceled){
cancel();
return;
};
if(details_links.length){
var d=details_links.pop();
return await fetch(d[2])
.then(async (response) => {
const j=await response.text();
get_sig_links_details_parse(file_links,details_links,partitions,d[0],d[1],j);
}).catch((error) => { errlog(error); });
};
};
function get_pd_links_details_parse(links,file_links,details_links,partitions,partition,section,data){
if(window.process_canceled){
cancel();
return;
};
for(const d of data.Data){
file_links.push([pd_replace(partition),sec_replace(section),d['Nazvanie'],
''.concat('https://lk.spbexp.ru/File/GetFile/',d['IDRow']),d['sDateTo'],d['Number'],d['sMD5']]);
progress(file_links);
details_links.push([partition,section,''.concat('https://lk.spbexp.ru/SF/FileCspViewer/',d['IDRow'])]);
};
if(links.length) get_pd_links_details(links,file_links,details_links,partitions);
else get_sig_links_details(file_links,details_links,partitions);
};
async function get_pd_links_details(links,file_links,details_links,partitions){
if(window.process_canceled){
cancel();
return;
};
if(links.length){
var d=links.pop();
return await fetch(d[2])
.then(async (response) => {
const j=await response.json();
get_pd_links_details_parse(links,file_links,details_links,partitions,d[0],d[1],j);
}).catch((error) => { errlog(error); });
};
};
function button_click(){
window.process_canceled=true;
};
class ProgressRing extends HTMLElement{
constructor(){
super();
const stroke=this.getAttribute('stroke');
const radius=this.getAttribute('radius');
const normalizedRadius=radius-stroke*2;
this._circumference=normalizedRadius*2*Math.PI;
this._root=this.attachShadow({mode:'open'});
this._root.innerHTML=`
<svg height="${radius*2}" width="${radius*2}">
<circle
class="ring"
stroke="#3c3b3a"
stroke-width="${stroke*2}"
stroke-opacity="0.5"
fill="transparent"
r="${normalizedRadius}"
cx="${radius}"
cy="${radius}"
shape-rendering="geometricPrecision"
/>
<circle
class="ring"
stroke="#c8c5c3"
stroke-dasharray="${this._circumference} ${this._circumference}"
style="stroke-dashoffset:${this._circumference}"
stroke-width="${stroke}"
fill="transparent"
r="${normalizedRadius}"
cx="${radius}"
cy="${radius}"
shape-rendering="geometricPrecision"
/>
<circle
class="button"
stroke="#191919"
stroke-width="1"
fill="#f44336"
r="${normalizedRadius-stroke}"
cx="${radius}"
cy="${radius}"
shape-rendering="geometricPrecision"
onclick="button_click()"
/>
<text class="txt" x="50%" y="52%" text-rendering="geometricPrecision">STOP</text>
<circle
class="ring"
stroke="#000000"
stroke-width="1"
stroke-opacity="0.5"
fill="transparent"
r="${normalizedRadius+8}"
cx="${radius}"
cy="${radius}"
shape-rendering="geometricPrecision"
/>
</svg>
<style>
.ring {
transition: stroke-dashoffset 0.35s;
transform: rotate(-90deg);
transform-origin: 50% 50%;
pointer-events: none;
}
.button:hover {
fill: #ce000f;
opacity: 1;
}
.txt {
font: bold 40px sans-serif;
fill: #323232;
stroke: #4c4c4c;
stroke-width: 1px;
text-anchor: middle;
dominant-baseline: middle;
pointer-events: none;
}
</style>
`;
}
setProgress(percent){
const offset=this._circumference-(percent/100*this._circumference);
const circle=this._root.querySelectorAll('circle')[1];
circle.style.strokeDashoffset=offset;
}
setTransition(value){
const circle=this._root.querySelectorAll('circle')[1];
circle.style.transition='stroke-dashoffset '+value+'s';
}
static get observedAttributes(){ return ['progress','transition']; }
attributeChangedCallback(name,oldValue,newValue){
if(name==='progress') this.setProgress(newValue);
if(name==='transition') this.setTransition(newValue);
}
}
function overlay() {
[].forEach.call(document.querySelectorAll('nav'), function (el) {
el.style.visibility = 'hidden';
});
[].forEach.call(document.querySelectorAll('*[class*=popover]'), function (el) {
el.style.visibility = 'hidden';
});
var overlay=document.createElement("div");
overlay.style.opacity=0.9;
overlay.style.width='100%';
overlay.style.height='100%';
overlay.style.top='0px';
overlay.style.left='0px';
overlay.style.backgroundColor='#666666';
overlay.style.zIndex='1000';
overlay.style.position = 'absolute';
document.getElementsByTagName('body')[0].appendChild(overlay);
window.customElements.define('progress-ring',ProgressRing);
const circle=document.createElement("div");
circle.innerHTML=`<progress-ring stroke="8" radius="88" progress="0"></progress-ring>`
circle.style.top='50%';
circle.style.left='50%';
circle.style.marginTop='-88px'
circle.style.marginLeft='-88px'
circle.style.zIndex='2000';
circle.style.position='absolute';
window.cprogress=0;
document.body.appendChild(circle);
document.body.style.overflow='hidden';
};
overlay();
const interval=setInterval(()=>{
const el=document.querySelector('progress-ring');
if(window.cprogress==200){
el.setAttribute('transition','0');
window.cprogress=0;
el.setAttribute('progress',window.cprogress);
}
else{
if(window.cprogress==0) el.setAttribute('transition','0.35');
window.cprogress+=2;
el.setAttribute('progress',window.cprogress);
};
},100);
"""
self.webEngineView=QWebEngineView()
ws=self.webEngineView.page().profile().defaultProfile()
ws.setHttpCacheMaximumSize(0)
ws=self.webEngineView.settings()
ws.setAttribute(ws.WebAttribute.AutoLoadImages,True)
ws.setAttribute(ws.WebAttribute.PluginsEnabled,False)
qwebchannel_js=QFile(':/qtwebchannel/qwebchannel.js')
if not qwebchannel_js.open(QIODeviceBase.OpenModeFlag.ReadOnly):
raise SystemExit(
'Failed to load qwebchannel.js with error: %s' %
qwebchannel_js.errorString())
self.qwebchannel_js=bytes(qwebchannel_js.readAll()).decode('utf-8')
self.channel=QWebChannel()
self.handler=CallHandler()
self.handler.set_parent(self)
self.channel.registerObject('backend',self.handler)
self.webEngineView.page().setWebChannel(self.channel)
self.grid=QGridLayout()
self.grid.setContentsMargins(0,0,0,0)
self.grid.setVerticalSpacing(0)
self.grid.setHorizontalSpacing(0)
self.hbox=QHBoxLayout()
self.hbox.setSpacing(0)
self.reload_button=QPushButton('Обновить страницу')
self.get_data_button=QPushButton('Получить данные')
self.get_all_data_button=QPushButton('Скачать данные проекта')
self.buttons=[self.reload_button,self.get_data_button,self.get_all_data_button]
self.hbox.addWidget(self.reload_button,1)
self.hbox.addWidget(self.get_data_button,1)
self.hbox.addWidget(self.get_all_data_button,1)
self.grid.addLayout(self.hbox,0,0)
self.grid.addWidget(self.webEngineView,1,0)
self.label=QLabel('Статус:...')
self.label.setStyleSheet("QLabel {background-color:gray;color:black;}")
self.grid.addWidget(self.label,2,0)
self.grid.setRowStretch(0,0)
self.grid.setRowStretch(1,1)
self.grid.setRowStretch(2,0)
self.setLayout(self.grid)
self.webEngineView.load(QUrl('https://lk.spbexp.ru'))
self.webEngineView.loadFinished.connect(self.login)
self.file_list=[]
self.file_number=0
self.lock=False
self.basepath=realpath('out')
try: makedirs(self.basepath,exist_ok=True)
except: pass
self.reload_button.clicked.connect(self.reload)
self.get_data_button.clicked.connect(self.getfileslist)
self.get_all_data_button.clicked.connect(self.getallfileslist)
def goto_projects(self):
self.webEngineView.loadFinished.disconnect()
self.webEngineView.load(QUrl('https://lk.spbexp.ru/SF/Zayava/'))
def login_callback(self,result):
if len(result):
self.webEngineView.loadFinished.connect(self.goto_projects)
def login(self):
l=p=''
try:
f=open('login','r')
d=f.readlines()
f.close()
l=d[0].strip()
p=d[1].strip()
except:
self.webEngineView.loadFinished.disconnect()
return
self.webEngineView.page().runJavaScript("""
function find_login_form(){
var v=document.querySelector('input[id=Login]');
if(v!=null){
v.value=`%s`;
v=document.querySelector('input[id=Password]');
if(v!=null){
v.value=`%s`;
v=document.querySelector('button[type=submit]');
if(v!=null){
v.click();
return 'ok';
};
};
};
return '';
};
find_login_form();"""%(l,p),self.login_callback)