Module saePisan.view.components.CsvDialogOption

Classes

class CSVOptionsDialog (parent=None)

Expand source code

class CSVOptionsDialog(QDialog):
    """
    A dialog for selecting CSV file options and previewing the CSV content.
    Attributes:
        file_path (str): The path to the selected CSV file.
        separator (str): The separator used in the CSV file.
        header (bool): Whether the first row is treated as a header.
    Methods:
        __init__(parent=None):
            Initializes the dialog with default values and sets up the UI.
        init_ui():
            Sets up the user interface components of the dialog.
        select_file():
            Opens a file picker dialog to select a CSV file and updates the preview.
        update_preview():
            Updates the preview table based on the selected file, separator, and header options.
        get_csv_options():
            Returns the selected CSV options if the dialog is accepted.
    """
    
    def __init__(self, parent=None):
        super().__init__(parent)
        self.setWindowTitle("CSV Options")

        self.file_path = None
        self.separator = ","
        self.header = True

        self.init_ui()

    def init_ui(self):
        layout = QVBoxLayout()

        # File picker
        self.file_label = QLabel("No file selected")
        layout.addWidget(self.file_label)
        self.file_button = QPushButton("Select File")
        self.file_button.clicked.connect(self.select_file)
        layout.addWidget(self.file_button)

        # Separator input
        layout.addWidget(QLabel("Separator (e.g. , or ; For tab use \\t)"))
        self.separator_input = QLineEdit(",")
        self.separator_input.textChanged.connect(self.update_preview)
        layout.addWidget(self.separator_input)

        # Header checkbox
        self.header_checkbox = QCheckBox("First row as header")
        self.header_checkbox.setChecked(True)  # Default header True
        self.header_checkbox.toggled.connect(self.update_preview)
        layout.addWidget(self.header_checkbox)

        # Preview Table
        self.preview_label = QLabel("Preview")
        layout.addWidget(self.preview_label)
        self.preview_table = QTableView()
        layout.addWidget(self.preview_table)

        # OK/Cancel buttons
        self.buttons = QDialogButtonBox(QDialogButtonBox.StandardButton.Ok | QDialogButtonBox.StandardButton.Cancel)
        self.buttons.accepted.connect(self.accept)
        self.buttons.rejected.connect(self.reject)
        layout.addWidget(self.buttons)

        self.setLayout(layout)

    def select_file(self):
        """File picker untuk memilih file CSV."""
        file_path, _ = QFileDialog.getOpenFileName(self, "Open CSV File", "", "CSV Files (*.csv)")
        if file_path:
            self.file_path = file_path
            self.file_label.setText(f"Selected: {file_path}")
            self.update_preview()

    def update_preview(self):
        """Perbarui preview tabel berdasarkan input."""
        if not self.file_path:
            self.preview_table.setModel(None)
            return

        sep = self.separator_input.text()
        if sep == r"\t":  # Jika input adalah string literal "\t"
            sep = "\t"
        hdr = True if self.header_checkbox.isChecked() else False
        try:
            preview_data = pl.read_csv(self.file_path, separator=sep, has_header=hdr, ignore_errors=True, null_values=["NA", "NULL", "na", "null"]).head(10)
            model = QStandardItemModel()
            
            model.setHorizontalHeaderLabels(preview_data.columns)

            for row in preview_data.to_numpy():
                items = [QStandardItem(str(cell)) for cell in row]
                model.appendRow(items)

            self.preview_table.setModel(model)
        except Exception as e:
            print(f"Error: {e}")
            self.preview_table.setModel(None)

    def get_csv_options(self):
        """Mengembalikan opsi CSV jika dialog diterima."""
        if self.exec() == QDialog.DialogCode.Accepted and self.file_path:
            return self.file_path, self.separator_input.text(), self.header_checkbox.isChecked()
        return None, None, None

A dialog for selecting CSV file options and previewing the CSV content.

Attributes

file_path : str

The path to the selected CSV file.

separator : str

The separator used in the CSV file.

header : bool

Whether the first row is treated as a header.

Methods

init(parent=None): Initializes the dialog with default values and sets up the UI. init_ui(): Sets up the user interface components of the dialog. select_file(): Opens a file picker dialog to select a CSV file and updates the preview. update_preview(): Updates the preview table based on the selected file, separator, and header options. get_csv_options(): Returns the selected CSV options if the dialog is accepted.

Ancestors

• PyQt6.QtWidgets.QDialog
• PyQt6.QtWidgets.QWidget
• PyQt6.QtCore.QObject
• PyQt6.sip.wrapper
• PyQt6.QtGui.QPaintDevice
• PyQt6.sip.simplewrapper

Methods

def get_csv_options(self)

Expand source code

def get_csv_options(self):
    """Mengembalikan opsi CSV jika dialog diterima."""
    if self.exec() == QDialog.DialogCode.Accepted and self.file_path:
        return self.file_path, self.separator_input.text(), self.header_checkbox.isChecked()
    return None, None, None

Mengembalikan opsi CSV jika dialog diterima.

def init_ui(self)

Expand source code

def init_ui(self):
    layout = QVBoxLayout()

    # File picker
    self.file_label = QLabel("No file selected")
    layout.addWidget(self.file_label)
    self.file_button = QPushButton("Select File")
    self.file_button.clicked.connect(self.select_file)
    layout.addWidget(self.file_button)

    # Separator input
    layout.addWidget(QLabel("Separator (e.g. , or ; For tab use \\t)"))
    self.separator_input = QLineEdit(",")
    self.separator_input.textChanged.connect(self.update_preview)
    layout.addWidget(self.separator_input)

    # Header checkbox
    self.header_checkbox = QCheckBox("First row as header")
    self.header_checkbox.setChecked(True)  # Default header True
    self.header_checkbox.toggled.connect(self.update_preview)
    layout.addWidget(self.header_checkbox)

    # Preview Table
    self.preview_label = QLabel("Preview")
    layout.addWidget(self.preview_label)
    self.preview_table = QTableView()
    layout.addWidget(self.preview_table)

    # OK/Cancel buttons
    self.buttons = QDialogButtonBox(QDialogButtonBox.StandardButton.Ok | QDialogButtonBox.StandardButton.Cancel)
    self.buttons.accepted.connect(self.accept)
    self.buttons.rejected.connect(self.reject)
    layout.addWidget(self.buttons)

    self.setLayout(layout)

def select_file(self)

Expand source code

def select_file(self):
    """File picker untuk memilih file CSV."""
    file_path, _ = QFileDialog.getOpenFileName(self, "Open CSV File", "", "CSV Files (*.csv)")
    if file_path:
        self.file_path = file_path
        self.file_label.setText(f"Selected: {file_path}")
        self.update_preview()

File picker untuk memilih file CSV.

def update_preview(self)

Expand source code

def update_preview(self):
    """Perbarui preview tabel berdasarkan input."""
    if not self.file_path:
        self.preview_table.setModel(None)
        return

    sep = self.separator_input.text()
    if sep == r"\t":  # Jika input adalah string literal "\t"
        sep = "\t"
    hdr = True if self.header_checkbox.isChecked() else False
    try:
        preview_data = pl.read_csv(self.file_path, separator=sep, has_header=hdr, ignore_errors=True, null_values=["NA", "NULL", "na", "null"]).head(10)
        model = QStandardItemModel()
        
        model.setHorizontalHeaderLabels(preview_data.columns)

        for row in preview_data.to_numpy():
            items = [QStandardItem(str(cell)) for cell in row]
            model.appendRow(items)

        self.preview_table.setModel(model)
    except Exception as e:
        print(f"Error: {e}")
        self.preview_table.setModel(None)

Perbarui preview tabel berdasarkan input.