# -*- coding: utf-8 -*-
from qgis.PyQt.QtWidgets import (
    QDialog, QVBoxLayout, QFormLayout, QHBoxLayout,
    QLineEdit, QCheckBox, QPushButton
)


class AddEtablissementDialog(QDialog):
    def __init__(self, parent=None):
        super().__init__(parent)
        self.setWindowTitle("Ajouter un établissement")
        layout = QVBoxLayout()
        self.setLayout(layout)

        form = QFormLayout()
        self.nom_edit = QLineEdit()
        self.nom_edit.setPlaceholderText("Nom de l'établissement")
        self.commune_edit = QLineEdit()
        self.commune_edit.setPlaceholderText("Commune")
        self.prive_check = QCheckBox("Établissement privé")

        form.addRow("Nom* :", self.nom_edit)
        form.addRow("Commune :", self.commune_edit)
        form.addRow("", self.prive_check)
        layout.addLayout(form)

        btns = QHBoxLayout()
        ok = QPushButton("Ajouter")
        cancel = QPushButton("Annuler")
        btns.addWidget(ok)
        btns.addWidget(cancel)
        layout.addLayout(btns)

        ok.clicked.connect(self.accept)
        cancel.clicked.connect(self.reject)

    def values(self):
        return (
            self.nom_edit.text().strip(),
            self.prive_check.isChecked(),
            self.commune_edit.text().strip() or None,
        )