Source code for web.views.correspondent_views.CorrespondentUpdateOrDeleteView

# SPDX-License-Identifier: AGPL-3.0-or-later
#
# Eonvelope - a open-source self-hostable email archiving server
# Copyright (C) 2024 David Aderbauer & The Eonvelope Contributors
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) 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 Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.

"""Module with the :class:`web.views.CorrespondentUpdateView` view."""

from __future__ import annotations

from typing import TYPE_CHECKING, override

from django.contrib.auth.mixins import LoginRequiredMixin
from django.urls import reverse_lazy

from core.models import Correspondent
from web.forms import BaseCorrespondentForm
from web.views.base import UpdateOrDeleteView

from .CorrespondentFilterView import CorrespondentFilterView

if TYPE_CHECKING:
    from django.db.models import QuerySet




class CorrespondentUpdateOrDeleteView(LoginRequiredMixin, UpdateOrDeleteView):
    """View for updating or deleting a single :class:`core.models.Correspondent` instance."""

    URL_NAME = Correspondent.get_edit_web_url_name()
    model = Correspondent
    form_class = BaseCorrespondentForm
    template_name = "web/correspondent/correspondent_edit.html"
    delete_success_url = reverse_lazy("web:" + CorrespondentFilterView.URL_NAME)



    @override
    def get_queryset(self) -> QuerySet[Correspondent]:
        """Restricts the queryset to objects owned by the requesting user."""
        return Correspondent.objects.filter(  # type: ignore[misc]  # user auth is checked by LoginRequiredMixin, we also test for this
            user=self.request.user
        ).distinct()