boris-behav-obs 8.16.5__py3-none-any.whl → 9.7.1__py3-none-any.whl

boris/view_df.py

@@ -0,0 +1,104 @@
+from .view_df_ui import Ui_Form
+from PySide6.QtWidgets import QWidget, QFileDialog
+from PySide6.QtCore import Qt, QAbstractTableModel
+
+from . import config as cfg
+from pathlib import Path
+from . import dialog
+
+try:
+    import pyreadr
+
+    flag_pyreadr_loaded = True
+except ModuleNotFoundError:
+    flag_pyreadr_loaded = False
+
+
+class DataFrameModel(QAbstractTableModel):
+    def __init__(self, dataframe):
+        super().__init__()
+        self._dataframe = dataframe
+
+    def rowCount(self, parent=None):
+        return self._dataframe.shape[0]
+
+    def columnCount(self, parent=None):
+        return self._dataframe.shape[1]
+
+    def data(self, index, role=Qt.DisplayRole):
+        if role == Qt.DisplayRole:
+            return str(self._dataframe.iat[index.row(), index.column()])
+        return None
+
+    def headerData(self, section, orientation, role=Qt.DisplayRole):
+        if role == Qt.DisplayRole:
+            if orientation == Qt.Horizontal:
+                return self._dataframe.columns[section]
+            elif orientation == Qt.Vertical:
+                return self._dataframe.index[section]
+        return None
+
+
+class View_df(QWidget, Ui_Form):
+    def __init__(self, plugin_name: str, plugin_version: str, df, parent=None):
+        super().__init__()
+        self.plugin_name = plugin_name
+        self.df = df
+
+        self.setupUi(self)
+        self.lb_plugin_info.setText(f"{plugin_name} v. {plugin_version}")
+        self.setWindowTitle(f"{plugin_name} v. {plugin_version}")
+
+        self.pb_close.clicked.connect(self.close)
+        self.pb_save.clicked.connect(self.save)
+
+        model = DataFrameModel(self.df)
+        self.tv_df.setModel(model)
+
+    def save(self):
+        file_formats = (
+            cfg.TSV,
+            cfg.CSV,
+            cfg.ODS,
+            cfg.XLSX,
+            # cfg.XLS,
+            cfg.HTML,
+            # cfg.TBS,
+            cfg.PANDAS_DF,
+            cfg.RDS,
+        )
+
+        file_dialog_options = QFileDialog.Options()
+        file_dialog_options |= QFileDialog.DontConfirmOverwrite
+
+        file_name, filter_ = QFileDialog().getSaveFileName(
+            None, f"Save {self.plugin_name}", "", ";;".join(file_formats), options=file_dialog_options
+        )
+        if not file_name:
+            return
+
+        outputFormat = cfg.FILE_NAME_SUFFIX[filter_]
+        if Path(file_name).suffix != "." + outputFormat:
+            file_name = f"{file_name}.{outputFormat}"
+            if Path(file_name).exists():
+                if (
+                    dialog.MessageDialog(cfg.programName, f"The file {file_name} already exists.", [cfg.CANCEL, cfg.OVERWRITE])
+                    == cfg.CANCEL
+                ):
+                    return
+
+        if filter_ == cfg.TSV:
+            self.df.to_csv(file_name, sep="\t", index=False)
+        if filter_ == cfg.CSV:
+            self.df.to_csv(file_name, sep=";", index=False)
+        if filter_ == cfg.XLSX:
+            self.df.to_excel(file_name, index=False)
+        if filter_ == cfg.ODS:
+            self.df.to_excel(file_name, index=False, engine="odf")
+        if filter_ == cfg.HTML:
+            self.df.to_html(file_name, index=False)
+        if filter_ == cfg.PANDAS_DF:
+            self.df.to_pickle(file_name)
+
+        if filter_ == cfg.RDS and flag_pyreadr_loaded:
+            pyreadr.write_rds(file_name, self.df)

boris/view_df_ui.py

@@ -0,0 +1,75 @@
+# -*- coding: utf-8 -*-
+
+################################################################################
+## Form generated from reading UI file 'view_df.ui'
+##
+## Created by: Qt User Interface Compiler version 6.8.0
+##
+## WARNING! All changes made in this file will be lost when recompiling UI file!
+################################################################################
+
+from PySide6.QtCore import (QCoreApplication, QDate, QDateTime, QLocale,
+    QMetaObject, QObject, QPoint, QRect,
+    QSize, QTime, QUrl, Qt)
+from PySide6.QtGui import (QBrush, QColor, QConicalGradient, QCursor,
+    QFont, QFontDatabase, QGradient, QIcon,
+    QImage, QKeySequence, QLinearGradient, QPainter,
+    QPalette, QPixmap, QRadialGradient, QTransform)
+from PySide6.QtWidgets import (QApplication, QHBoxLayout, QHeaderView, QLabel,
+    QPushButton, QSizePolicy, QSpacerItem, QTableView,
+    QVBoxLayout, QWidget)
+
+class Ui_Form(object):
+    def setupUi(self, Form):
+        if not Form.objectName():
+            Form.setObjectName(u"Form")
+        Form.resize(400, 300)
+        self.verticalLayout_2 = QVBoxLayout(Form)
+        self.verticalLayout_2.setObjectName(u"verticalLayout_2")
+        self.lb_plugin_info = QLabel(Form)
+        self.lb_plugin_info.setObjectName(u"lb_plugin_info")
+
+        self.verticalLayout_2.addWidget(self.lb_plugin_info)
+
+        self.verticalLayout = QVBoxLayout()
+        self.verticalLayout.setObjectName(u"verticalLayout")
+        self.tv_df = QTableView(Form)
+        self.tv_df.setObjectName(u"tv_df")
+
+        self.verticalLayout.addWidget(self.tv_df)
+
+        self.horizontalLayout = QHBoxLayout()
+        self.horizontalLayout.setObjectName(u"horizontalLayout")
+        self.horizontalSpacer = QSpacerItem(40, 20, QSizePolicy.Policy.Expanding, QSizePolicy.Policy.Minimum)
+
+        self.horizontalLayout.addItem(self.horizontalSpacer)
+
+        self.pb_save = QPushButton(Form)
+        self.pb_save.setObjectName(u"pb_save")
+
+        self.horizontalLayout.addWidget(self.pb_save)
+
+        self.pb_close = QPushButton(Form)
+        self.pb_close.setObjectName(u"pb_close")
+
+        self.horizontalLayout.addWidget(self.pb_close)
+
+
+        self.verticalLayout.addLayout(self.horizontalLayout)
+
+
+        self.verticalLayout_2.addLayout(self.verticalLayout)
+
+
+        self.retranslateUi(Form)
+
+        QMetaObject.connectSlotsByName(Form)
+    # setupUi
+
+    def retranslateUi(self, Form):
+        Form.setWindowTitle(QCoreApplication.translate("Form", u"Form", None))
+        self.lb_plugin_info.setText(QCoreApplication.translate("Form", u"TextLabel", None))
+        self.pb_save.setText(QCoreApplication.translate("Form", u"Save results", None))
+        self.pb_close.setText(QCoreApplication.translate("Form", u"Close", None))
+    # retranslateUi
+