Mercurial Repositories > Django Tags Menu / file revision

# -*- coding: utf-8 -*-

# Copyright (c) 2014 - 2023 Detlev Offenbach <detlev@die-offenbachs.de>
#

"""
Module implementing a dialog to enter the parameters for the if tag.
"""

from PyQt6.QtCore import pyqtSlot
from PyQt6.QtWidgets import QDialog, QDialogButtonBox

from .Ui_IfTagInputDialog import Ui_IfTagInputDialog


class IfTagInputDialog(QDialog, Ui_IfTagInputDialog):
    """
    Class implementing a dialog to enter the parameters for the if tag.
    """

    def __init__(self, parent=None):
        """
        Constructor

        @param parent reference to the parent widget
        @type QWidget
        """
        super().__init__(parent)
        self.setupUi(self)

        self.buttonBox.button(QDialogButtonBox.StandardButton.Ok).setEnabled(False)

    @pyqtSlot(str)
    def on_ifEdit_textChanged(self, txt):
        """
        Private slot to handle changes of the 'if' expression.

        @param txt text of the line edit
        @type str
        """
        self.buttonBox.button(QDialogButtonBox.StandardButton.Ok).setEnabled(bool(txt))

    def getTag(self):
        """
        Public method to retrieve the tag.

        @return tag
        @rtype str
        """
        tag = "{{% if {0} %}}\n".format(self.ifEdit.text())
        elifText = self.elifEdit.toPlainText()
        if elifText:
            for expression in elifText.splitlines():
                if expression.strip():
                    tag += "{{% elif {0} %}}\n".format(expression.strip())
        if self.elseCheckBox.isChecked():
            tag += "{% else %}\n"
        tag += "{% endif %}"

        return tag