1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210
| import os
import sys
import pandas as pd
from PyQt6 import QtWidgets, QtCore
class PandasModel(QtCore.QAbstractTableModel):
def __init__(self, data):
super().__init__()
self._data = data
def rowCount(self, parent=None):
return len(self._data.index)
def columnCount(self, parent=None):
return len(self._data.columns)
def data(self, index: QtCore.QModelIndex, role: int = QtCore.Qt.ItemDataRole.DisplayRole):
if role == QtCore.Qt.ItemDataRole.DisplayRole:
return str(self._data.iloc[index.row(), index.column()])
def headerData(self, section: int, orientation: QtCore.Qt.Orientation,
role: int = QtCore.Qt.ItemDataRole.DisplayRole):
if role == QtCore.Qt.ItemDataRole.DisplayRole:
if orientation == QtCore.Qt.Orientation.Horizontal:
return self._data.columns[section]
else:
return str(self._data.index[section])
class MainWindow(QtWidgets.QMainWindow):
def __init__(self):
super().__init__()
self.resize(1015, 593)
# Créez un widget QTabWidget
self.tab_widget = QtWidgets.QTabWidget()
# Active les boutons de fermeture sur les onglets
self.tab_widget.setTabsClosable(True)
# Connecte le signal tabCloseRequested à la méthode close_tab
self.tab_widget.tabCloseRequested.connect(self.close_tab)
# Crée un menu principal
menu = self.menuBar()
file_menu = menu.addMenu("Fichier")
# Ajoute les commandes "Importer" et "Quitter" au sous-menu "Fichier"
file_menu.addAction("Importer", self.import_file)
# noinspection PyArgumentList
file_menu.addAction("Quitter", self.close)
# Ajoute un nouveau menu "Outils" à côté du menu "Fichier"
tools_menu = menu.addMenu("Outils")
# Ajoute la commande "Descriptions similaires"
find_similar_action = tools_menu.addAction("Descriptions similaires")
# Stocke l'action find_similar_action dans une variable d'instance pour y accéder plus tard
self.find_similar_action = find_similar_action
# Connecte l'action find_similar_action à la méthode find_similar
self.find_similar_action.triggered.connect(self.find_similar)
# Crée une barre de recherche
search_bar = QtWidgets.QLineEdit()
search_bar.setPlaceholderText("Rechercher...")
search_bar.setFixedWidth(189) # 5 cm en pixels
search_bar.setStyleSheet("QLineEdit { border: 1px solid gray; border-radius: 10px; padding: 0 8px; }")
spacer = QtWidgets.QWidget()
spacer.setSizePolicy(QtWidgets.QSizePolicy.Policy.Expanding, QtWidgets.QSizePolicy.Policy.Expanding)
# Crée un widget QToolBar pour contenir la barre de recherche
tool_bar = QtWidgets.QToolBar()
tool_bar.addWidget(spacer)
tool_bar.addWidget(search_bar)
# Créez une disposition verticale pour organiser les widgets
layout = QtWidgets.QVBoxLayout()
# Ajoutez le QToolBar et le QTabWidget à la disposition
layout.addWidget(tool_bar)
layout.addWidget(self.tab_widget)
# Crée un widget central pour contenir la disposition
central_widget = QtWidgets.QWidget()
central_widget.setLayout(layout)
self.setCentralWidget(central_widget)
# Permet le déplacement des onglets pour changer l'ordre des pages
self.tab_widget.setMovable(True)
def import_file(self):
filename, _ = QtWidgets.QFileDialog.getOpenFileName(self, "Sélectionner un fichier", "",
"Fichiers Excel (*.xlsx *.xls *.xlsm)")
if filename:
# Lit le fichier Excel et stocke les données dans un DataFrame
df = pd.read_excel(filename)
# Vérifie si le DataFrame a une colonne SIMILARITY
if "SIMILARITY" in df.columns:
# Convertit la colonne SIMILARITY en chaîne de caractères
df['SIMILARITY'] = df['SIMILARITY'].astype(str)
# Stocke les données du DataFrame dans une variable d'instance pour y accéder plus tard
self._data = df
# Créez le premier onglet
tab1 = QtWidgets.QWidget()
tab1_layout = QtWidgets.QVBoxLayout()
self.tab1_table = QtWidgets.QTableView()
tab1_layout.addWidget(self.tab1_table)
tab1.setLayout(tab1_layout)
self.tab_widget.addTab(tab1, "Importé: " + os.path.splitext(os.path.basename(filename))[0])
# Définissez le modèle pour le QTableView en utilisant les données de votre DataFrame
model = PandasModel(df)
self.tab1_table.setModel(model)
# Permet le déplacement des onglets pour changer l'ordre des pages
self.tab_widget.setMovable(True)
def close_tab(self, index):
self.tab_widget.removeTab(index)
def find_similar(self):
import_tab_exists = False
for i in range(self.tab_widget.count()):
if self.tab_widget.tabText(i).startswith("Importé: "):
import_tab_exists = True
break
if not import_tab_exists:
QtWidgets.QMessageBox.warning(self, "Attention", "Veuillez importer des données avant d'utiliser cette fonctionnalité.")
return
current_tab_index = self.tab_widget.currentIndex()
if current_tab_index != -1 and self.tab_widget.tabText(current_tab_index).startswith("Extrait: "):
QtWidgets.QMessageBox.warning(self, "Attention", "La fonctionnalité Descriptions similaires n'est pas disponible pour les onglets contenant des données extraites.")
return
current_tab = self.tab_widget.currentWidget()
if not current_tab or not current_tab.findChild(QtWidgets.QTableView):
QtWidgets.QMessageBox.warning(self, "Attention", "La fonctionnalité Descriptions similaires n'est disponible que pour les onglets contenant des données importées.")
return
if not self.tab_widget.findChild(QtWidgets.QTableView):
QtWidgets.QMessageBox.warning(self, "Attention", "Veuillez importer des données avant d'utiliser cette fonctionnalité.")
return
if "SIMILARITY" not in self._data.columns:
QtWidgets.QMessageBox.warning(self, "Attention",
"Le fichier Excel importé doit contenir une colonne SIMILARITY.")
return
if not hasattr(self, "_data"):
QtWidgets.QMessageBox.warning(self, "Attention",
"Veuillez importer des données avant d'utiliser cette fonctionnalité.")
return
# Récupère l'index de la ligne et de la colonne de la cellule sélectionnée
selected_indexes = self.tab1_table.selectedIndexes()
if len(selected_indexes) != 1:
QtWidgets.QMessageBox.warning(self, "Attention", "Veuillez sélectionner une seule cellule.")
return
row = selected_indexes[0].row()
col = selected_indexes[0].column()
# Vérifie si la colonne sélectionnée est la colonne SIMILARITY
if self._data.columns[col] != "SIMILARITY":
QtWidgets.QMessageBox.warning(self, "Attention", "Veuillez sélectionner une cellule dans la colonne SIMILARITY.")
return
tab_name = f"extraire: {self._data.iloc[row]['ITEM']}"
for i in range(self.tab_widget.count()):
if self.tab_widget.tabText(i) == tab_name:
QtWidgets.QMessageBox.warning(self, "Attention", "Un onglet avec les mêmes données extraites existe déjà.")
return
# Vérifie si la valeur de la cellule sélectionnée n'est pas 0
if self._data.iloc[row, col] == "0":
QtWidgets.QMessageBox.warning(self, "Attention", "La cellule sélectionnée ne doit pas contenir 0.")
return
# Récupère les numéros de la colonne ITEM mentionnés dans la cellule sélectionnée
item_numbers = self._data.iloc[row, col].split(" ")
# Crée un nouveau DataFrame contenant les entêtes, la ligne sélectionnée et les lignes correspondant aux numéros de la colonne ITEM
new_df = pd.DataFrame(columns=self._data.columns)
new_df = new_df.append(self._data.iloc[row])
for item_number in item_numbers:
new_df = new_df.append(self._data[self._data["ITEM"] == int(item_number)])
# Crée un nouveau QTableView pour afficher les données du nouveau DataFrame
new_tab_table = QtWidgets.QTableView()
model = PandasModel(new_df)
new_tab_table.setModel(model)
# Crée un nouvel onglet avec le nouveau QTableView
new_tab = QtWidgets.QWidget()
new_tab_layout = QtWidgets.QVBoxLayout()
new_tab_layout.addWidget(new_tab_table)
new_tab.setLayout(new_tab_layout)
self.tab_widget.addTab(new_tab, f"extraire: {self._data.iloc[row]['ITEM']}")
app = QtWidgets.QApplication(sys.argv)
window = MainWindow()
window.show()
app.exec() |
Partager