wikicookbook/wikicookbook.py

#!/usr/bin/python

"""
WikiCookBook

Easily enter recipes and export them to beautiful wiki pages.

Copyright (C) 2020 Marcel Kapfer <opensource@mmk2410.org>

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <https://www.gnu.org/licenses/>.
"""

import sys
import recipe
from writer.MoinMoinWriter import MoinMoinWriter
from PyQt5.QtWidgets import QApplication, QMainWindow, QMessageBox, QTableWidgetItem, QDialog
from ui_mainwindow import Ui_MainWindow
from ui_code_dialog import Ui_WikiCode
from ui_about_dialog import Ui_About
from ui_clear_dialog import Ui_ClearDialog

class AboutDialog(QDialog, Ui_About):
    def __init__(self, *args, **kwargs):
        super(AboutDialog, self).__init__(*args, **kwargs)
        self.setupUi(self)

        self.b_close.clicked.connect(self.close)

        self.show()

class ClearDialog(QDialog, Ui_ClearDialog):
    def __init__(self, *args, **kwargs):
        super(ClearDialog, self).__init__(*args, **kwargs)
        self.setupUi(self)

        self.b_yes.clicked.connect(self.accept)
        self.b_no.clicked.connect(self.reject)

        self.show()

class WikiCodeDialog(QDialog, Ui_WikiCode):
    def __init__(self, code, *args, **kwargs):
        super(WikiCodeDialog, self).__init__(*args, **kwargs)
        self.setupUi(self)
        self.copied = False

        self.e_code.setPlainText(code)

        self.b_close.clicked.connect(self.clear_fields)
        self.b_copy.clicked.connect(self.copy)

        self.show()

    def copy(self):
        self.e_code.selectAll()
        self.e_code.copy()
        self.copied = True

    def clear_fields(self):
        """
        Show the clear fields dialog if the code was at least copied once.
        """
        if self.copied:
            dialog = ClearDialog()
            dialog.rejected.connect(self.reject)
            dialog.accepted.connect(self.accept)
            dialog.exec_()
        else:
            self.close()


class MainWindow(QMainWindow, Ui_MainWindow):
    def __init__(self, *args, **kwargs):
        super(MainWindow, self).__init__(*args, **kwargs)
        self.setupUi(self)

        # Connect close actions
        self.b_cancel.clicked.connect(self.close)
        self.ma_close.triggered.connect(self.close)

        # Connect other menu actions
        self.ma_information.triggered.connect(self.about)
        self.ma_clear_fields.triggered.connect(self.clear_fields)

        # Conncect recipe name typing
        self.e_recipe_name.textChanged.connect(self.updateTitle)

        # Handle categories
        self.b_category_add.clicked.connect(self.category_add)
        self.e_category.returnPressed.connect(self.category_add)
        self.b_category_remove.clicked.connect(self.category_remove)

        # Handle ingredients
        self.b_ingredient_add.clicked.connect(self.ingredient_add)
        self.b_ingredient_remove.clicked.connect(self.ingredient_remove)

        # Handle utensils
        self.b_utensil_add.clicked.connect(self.utensil_add)
        self.e_utensil.returnPressed.connect(self.utensil_add)
        self.b_utensil_remove.clicked.connect(self.utensil_remove)

        # Handle steps
        self.b_step_add.clicked.connect(self.step_add)
        self.b_step_edit.clicked.connect(self.step_edit)
        self.b_step_replace.clicked.connect(self.step_replace)
        self.b_step_remove.clicked.connect(self.step_remove)

        # Handle notes
        self.b_note_add.clicked.connect(self.note_add)
        self.b_note_edit.clicked.connect(self.note_edit)
        self.b_note_replace.clicked.connect(self.note_replace)
        self.b_note_remove.clicked.connect(self.note_remove)

        # Handle tips
        self.b_tip_add.clicked.connect(self.tip_add)
        self.b_tip_edit.clicked.connect(self.tip_edit)
        self.b_tip_replace.clicked.connect(self.tip_replace)
        self.b_tip_remove.clicked.connect(self.tip_remove)

        # Handle variations
        self.b_variation_add.clicked.connect(self.variation_add)
        self.b_variation_edit.clicked.connect(self.variation_edit)
        self.b_variation_replace.clicked.connect(self.variation_replace)
        self.b_variation_remove.clicked.connect(self.variation_remove)

        # Handle ratings
        self.b_rating_add.clicked.connect(self.rating_add)
        self.b_rating_edit.clicked.connect(self.rating_edit)
        self.b_rating_replace.clicked.connect(self.rating_replace)
        self.b_rating_remove.clicked.connect(self.rating_remove)

        # Handle recipe creation
        self.b_create.clicked.connect(self.create_recipe)

        self.show()

    def updateTitle(self):
        """
        Update the window title so it contains the name of the current recipe.
        """
        current_recipe_name = self.e_recipe_name.text()
        if current_recipe_name:
            self.setWindowTitle(f"{current_recipe_name} - WikiCookBook")
        else:
            self.setWindowTitle("WikiCookBook")

    def category_add(self):
        if self.e_category.text():
            self.v_categories.addItem(f"Kategorie{self.e_category.text()}")
            self.e_category.setFocus()
            self.e_category.selectAll()

    def category_remove(self):
        self.v_categories.takeItem(self.v_categories.currentRow())

    def ingredient_add(self):
        if self.e_ingredient.text():
            if self.c_heading.checkState() == 2:
                current_amount = QTableWidgetItem("----------")
            else:
                current_amount = QTableWidgetItem(self.e_amount.text())
            current_ingredient = QTableWidgetItem(self.e_ingredient.text())
            self.v_ingredients.setRowCount(self.v_ingredients.rowCount() + 1)
            self.v_ingredients.setItem(self.v_ingredients.rowCount() - 1, 0, current_amount)
            self.v_ingredients.setItem(self.v_ingredients.rowCount() - 1, 1, current_ingredient)
            self.e_ingredient.setFocus()
            self.e_ingredient.selectAll()

    def ingredient_remove(self):
        self.v_ingredients.removeRow(self.v_ingredients.currentRow())

    def utensil_add(self):
        if self.e_utensil.text():
            self.v_utensils.addItem(self.e_utensil.text())
            self.e_utensil.setFocus()
            self.e_utensil.selectAll()

    def utensil_remove(self):
        self.v_utensils.takeItem(self.v_utensils.currentRow())

    def step_add(self):
        if self.e_step.toPlainText():
            self.v_steps.addItem(self.e_step.toPlainText())
            self.e_step.setFocus()
            self.e_step.selectAll()

    def step_edit(self):
        if self.v_steps.count() > 0:
            self.e_step.setText(self.v_steps.currentItem().text())

    def step_replace(self):
        if self.e_step.toPlainText() and self.v_steps.count() > 0:
            self.v_steps.currentItem().setText(self.e_step.toPlainText())

    def step_remove(self):
        self.v_steps.takeItem(self.v_steps.currentRow())

    def note_add(self):
        if self.e_note.toPlainText():
            self.v_notes.addItem(self.e_note.toPlainText())
            self.e_note.setFocus()
            self.e_note.selectAll()

    def note_edit(self):
        if self.v_notes.count() > 0:
            self.e_note.setText(self.v_notes.currentItem().text())

    def note_replace(self):
        if self.e_note.toPlainText() and self.v_notes.count() > 0:
            self.v_notes.currentItem().setText(self.e_note.toPlainText())

    def note_remove(self):
        self.v_notes.takeItem(self.v_notes.currentRow())

    def tip_add(self):
        if self.e_tip.toPlainText():
            self.v_tips.addItem(self.e_tip.toPlainText())
            self.e_tip.setFocus()
            self.e_tip.selectAll()

    def tip_edit(self):
        if self.v_tips.count() > 0:
            self.e_tip.setText(self.v_tips.currentItem().text())

    def tip_replace(self):
        if self.e_tip.toPlainText() and self.v_tips.count() > 0:
            self.v_tips.currentItem().setText(self.e_tip.toPlainText())

    def tip_remove(self):
        self.v_tips.takeItem(self.v_tips.currentRow())

    def variation_add(self):
        if self.e_variation.toPlainText():
            self.v_variations.addItem(self.e_variation.toPlainText())
            self.e_variation.setFocus()
            self.e_variation.selectAll()

    def variation_edit(self):
        if self.v_variations.count() > 0:
            self.e_variation.setText(self.v_variations.currentItem().text())

    def variation_replace(self):
        if self.e_variation.toPlainText() and self.v_variations.count() > 0:
            self.v_variations.currentItem().setText(self.e_variation.toPlainText())

    def variation_remove(self):
        self.v_variations.takeItem(self.v_variations.currentRow())

    def rating_add(self):
        if self.e_rating.toPlainText():
            self.v_ratings.addItem(self.e_rating.toPlainText())
            self.e_rating.setFocus()
            self.e_rating.selectAll()

    def rating_edit(self):
        if self.v_ratings.count() > 0:
            self.e_rating.setText(self.v_ratings.currentItem().text())

    def rating_replace(self):
        if self.e_rating.toPlainText() and self.v_ratings.count() > 0:
            self.v_ratings.currentItem().setText(self.e_rating.toPlainText())

    def rating_remove(self):
        self.v_ratings.takeItem(self.v_ratings.currentRow())

    def create_recipe(self):
        current_recipe = recipe.Recipe(self.e_recipe_name.text(),
                                       self.e_servings.text(),
                                       self.e_time.text(),
                                       self.e_rating_overall.value(),
                                       self.e_url.text())

        # Categories
        for row in range(self.v_categories.count()):
            current_recipe.add_category(self.v_categories.item(row).text())

        # Ingredients
        for row in range(self.v_ingredients.rowCount()):
            current_recipe.add_ingredient(
                self.v_ingredients.item(row, 0).text(),
                self.v_ingredients.item(row, 1).text()
            )

        # Utensils
        for row in range(self.v_utensils.count()):
            current_recipe.add_utensil(self.v_utensils.item(row).text())

        # Steps
        for row in range(self.v_steps.count()):
            current_recipe.add_step(self.v_steps.item(row).text())

        # Notes
        for row in range(self.v_notes.count()):
            current_recipe.add_note(self.v_notes.item(row).text())

        # Tips
        for row in range(self.v_tips.count()):
            current_recipe.add_tip(self.v_tips.item(row).text())

        # Variations
        for row in range(self.v_variations.count()):
            current_recipe.add_variation(self.v_variations.item(row).text())

        # Ratings
        for row in range(self.v_ratings.count()):
            current_recipe.add_rating(self.v_ratings.item(row).text())

        writer = MoinMoinWriter()
        code = current_recipe.wikicode(writer)

        dialog = WikiCodeDialog(code)
        dialog.accepted.connect(self.clear)
        dialog.exec_()

    def clear(self):
        """
        Clear all fields
        """

        # Tab: Basics
        self.e_recipe_name.setText("")
        self.e_servings.setText("")
        self.e_time.setText("")
        self.e_rating_overall.setValue(3)
        self.e_url.setText("")
        self.e_category.setText("")
        self.v_categories.clear()

        # Tab: Ingredients
        self.e_ingredient.setText("")
        self.e_amount.setText("")
        self.c_heading.setChecked(False)
        self.v_ingredients.clear()

        # Tab: Utensils
        self.e_utensil.setText("")
        self.v_utensils.clear()

        # Tab: Steps
        self.e_step.setText("")
        self.v_steps.clear()

        # Tab: Notes
        self.e_note.setText("")
        self.v_notes.clear()

        # Tab: Tips
        self.e_tip.setText("")
        self.v_tips.clear()

        # Tab: Variations
        self.e_variation.setText("")
        self.v_variations.clear()

        # Tab: Ratings
        self.e_rating.setText("")
        self.v_ratings.clear()

    def clear_fields(self):
        """
        Show the clear fields dialog.
        """
        dialog = ClearDialog()
        dialog.accepted.connect(self.clear)
        dialog.exec_()

    def about(self):
        """
        Show an about dialog.
        """
        dialog = AboutDialog()
        dialog.exec_()

    def close(self):
        """
        Close the application. A warning is shown before.
        """
        reply = QMessageBox.question(self, "Sicher?",
            "Möchten Sie wirklich schließen?",
            QMessageBox.Yes | QMessageBox.No, QMessageBox.No)
        if reply == QMessageBox.Yes:
            super.close()


if __name__ == "__main__":
    app = QApplication([])
    app.setApplicationName("WikiCookBook")
    window = MainWindow()
    sys.exit(app.exec_())