from wagtail.admin.forms import WagtailAdminPageForm


class UniwebHomePageAdminForm(WagtailAdminPageForm):
    def clean(self):
        cleaned_data = super().clean()

        # If one logo is set, then both logos must be set.
        if (cleaned_data["dark_logo"] or cleaned_data["light_logo"]) and (
            not cleaned_data["dark_logo"] or not cleaned_data["light_logo"]
        ):
            self.add_error(
                "dark_logo",
                "Pokud je jedno logo nastaveno, musí být obě loga nastaveny.",
            )
            self.add_error(
                "light_logo",
                "Pokud je jedno logo nastaveno, musí být obě loga nastaveny.",
            )