class AttributeCreatorWidget(CustomScrollableListItem):
def __init__(self, document_base_creator_widget) -> None:
super(AttributeCreatorWidget,
self).__init__(document_base_creator_widget)
self.document_base_creator_widget = document_base_creator_widget
self.setFixedHeight(40)
self.setStyleSheet("background-color: white")
self.layout = QHBoxLayout(self)
self.layout.setContentsMargins(20, 0, 20, 0)
self.layout.setSpacing(10)
self.name = QLineEdit()
self.name.setFont(CODE_FONT_BOLD)
self.name.setStyleSheet("border: none")
self.layout.addWidget(self.name)
self.delete_button = QPushButton()
self.delete_button.setIcon(QIcon("aset_ui/resources/trash.svg"))
self.delete_button.setFlat(True)
self.delete_button.clicked.connect(self._delete_button_clicked)
self.layout.addWidget(self.delete_button)
def update_item(self, item, params=None):
self.name.setText(item)
def _delete_button_clicked(self):
self.document_base_creator_widget.delete_attribute(self.name.text())
def enable_input(self):
self.delete_button.setEnabled(True)
def disable_input(self):
self.delete_button.setEnabled(False)
class DocumentBaseCreatorWidget(MainWindowContent):
def __init__(self, main_window) -> None:
super(DocumentBaseCreatorWidget,
self).__init__(main_window, "Create Document Base")
self.documents = MainWindowContentSection(self, "Documents:")
self.layout.addWidget(self.documents)
self.documents_explanation = QLabel(
"Enter the path of the directory that contains the documents as .txt files."
)
self.documents_explanation.setFont(LABEL_FONT)
self.documents.layout.addWidget(self.documents_explanation)
self.path_widget = QFrame()
self.path_layout = QHBoxLayout(self.path_widget)
self.path_layout.setContentsMargins(20, 0, 20, 0)
self.path_layout.setSpacing(10)
self.path_widget.setStyleSheet("background-color: white")
self.path_widget.setFixedHeight(40)
self.documents.layout.addWidget(self.path_widget)
self.path = QLineEdit()
self.path.setFont(CODE_FONT_BOLD)
self.path.setStyleSheet("border: none")
self.path_layout.addWidget(self.path)
self.edit_path_button = QPushButton()
self.edit_path_button.setIcon(QIcon("aset_ui/resources/folder.svg"))
self.edit_path_button.setFlat(True)
self.edit_path_button.clicked.connect(self._edit_path_button_clicked)
self.path_layout.addWidget(self.edit_path_button)
self.attributes = MainWindowContentSection(self, "Attributes:")
self.layout.addWidget(self.attributes)
self.labels_explanation = QLabel("Enter the attribute names.")
self.labels_explanation.setFont(LABEL_FONT)
self.attributes.layout.addWidget(self.labels_explanation)
self.create_attribute_button = QPushButton("New Attribute")
self.create_attribute_button.setFont(BUTTON_FONT)
self.create_attribute_button.clicked.connect(
self._create_attribute_button_clicked)
self.attribute_names = []
self.attributes_list = CustomScrollableList(
self, AttributeCreatorWidget, self.create_attribute_button)
self.attributes.layout.addWidget(self.attributes_list)
self.buttons_widget = QWidget()
self.buttons_layout = QHBoxLayout(self.buttons_widget)
self.buttons_layout.setContentsMargins(0, 0, 0, 0)
self.buttons_layout.setSpacing(10)
self.buttons_layout.setAlignment(Qt.AlignmentFlag.AlignRight)
self.attributes.layout.addWidget(self.buttons_widget)
self.cancel_button = QPushButton("Cancel")
self.cancel_button.setFont(BUTTON_FONT)
self.cancel_button.clicked.connect(self._cancel_button_clicked)
self.buttons_layout.addWidget(self.cancel_button)
self.create_document_base_button = QPushButton("Create Document Base")
self.create_document_base_button.setFont(BUTTON_FONT)
self.create_document_base_button.clicked.connect(
self._create_document_base_button_clicked)
self.buttons_layout.addWidget(self.create_document_base_button)
def enable_input(self):
self.edit_path_button.setEnabled(True)
self.create_attribute_button.setEnabled(True)
self.cancel_button.setEnabled(True)
self.create_document_base_button.setEnabled(True)
self.attributes_list.enable_input()
def disable_input(self):
self.edit_path_button.setDisabled(True)
self.create_attribute_button.setDisabled(True)
self.cancel_button.setDisabled(True)
self.create_document_base_button.setDisabled(True)
self.attributes_list.disable_input()
def initialize_for_new_document_base(self):
self.path.setText("")
self.attribute_names = []
self.attributes_list.update_item_list([])
def delete_attribute(self, attribute_name):
self.attribute_names = []
for attribute_widget in self.attributes_list.item_widgets[:self.
attributes_list
.
num_visible_item_widgets]:
self.attribute_names.append(attribute_widget.name.text())
self.attribute_names.remove(attribute_name)
self.attributes_list.update_item_list(self.attribute_names)
self.attributes_list.last_item_widget().name.setFocus()
def _edit_path_button_clicked(self):
path = str(
QFileDialog.getExistingDirectory(
self, "Choose a directory of text files."))
if path != "":
path = f"{path}/*.txt"
self.path.setText(path)
def _create_attribute_button_clicked(self):
self.attribute_names = []
for attribute_widget in self.attributes_list.item_widgets[:self.
attributes_list
.
num_visible_item_widgets]:
self.attribute_names.append(attribute_widget.name.text())
self.attribute_names.append("")
self.attributes_list.update_item_list(self.attribute_names)
self.attributes_list.last_item_widget().name.setFocus()
def _cancel_button_clicked(self):
self.main_window.show_start_menu_widget()
self.main_window.enable_global_input()
def _create_document_base_button_clicked(self):
self.attribute_names = []
for attribute_widget in self.attributes_list.item_widgets[:self.
attributes_list
.
num_visible_item_widgets]:
self.attribute_names.append(attribute_widget.name.text())
self.main_window.create_document_base_task(self.path.text(),
self.attribute_names)
