from dal import autocomplete
from django.contrib import messages
from django.contrib.auth.mixins import PermissionRequiredMixin
from django.contrib.contenttypes.models import ContentType
from django.db.models import Q
from django.shortcuts import get_object_or_404, redirect
from django.urls import reverse
from django.utils.html import format_html
from django.views import View
from django.views.generic.edit import FormView
from apis_core.apis_entities.forms import EntitiesMergeForm
from apis_core.apis_metainfo.models import RootObject
from apis_core.generic.helpers import generate_search_filter
from apis_core.generic.views import GenericModelMixin, Update
[docs]
class EntitiesUpdate(Update):
[docs]
def get_context_data(self, *args, **kwargs):
context = super().get_context_data(*args, **kwargs)
context["mergeform"] = EntitiesMergeForm(instance=self.get_object())
return context
[docs]
class EntitiesDuplicate(GenericModelMixin, PermissionRequiredMixin, View):
permission_action_required = "create"
[docs]
def get(self, request, *args, **kwargs):
source_obj = get_object_or_404(self.model, pk=kwargs["pk"])
newobj = source_obj.duplicate()
messages.success(
request,
format_html(
"<a href={}>{}</a> was successfully duplicated to the current object:",
source_obj.get_absolute_url(),
source_obj,
),
)
return redirect(
reverse(
"apis_core:apis_entities:generic_entities_edit_view",
kwargs={
"pk": newobj.id,
"contenttype": newobj.__class__.__name__.lower(),
},
)
)
[docs]
class EntitiesMerge(GenericModelMixin, PermissionRequiredMixin, FormView):
permission_action_required = "create"
form_class = EntitiesMergeForm
template_name = "entity_merge.html"
template_name_suffix = "_merge"
[docs]
def get_object(self, *args, **kwargs):
return get_object_or_404(self.model, pk=self.kwargs.get("pk"))
[docs]
def get_context_data(self, *args, **kwargs):
context = super().get_context_data(*args, **kwargs)
context["object"] = self.get_object()
return context
[docs]
def get_success_url(self):
return reverse(
"apis_core:apis_entities:generic_entities_edit_view",
args=[self.get_object().__class__.__name__.lower(), self.get_object().id],
)
[docs]
class EntitiesAutocomplete(autocomplete.Select2QuerySetView):
"""
This endpoint allows us to use autocomplete over multiple model classes.
It takes a parameter `entities` which is a list of ContentType natural
keys and searches for the query in all instances of those entities
(using `generate_search_filter`, which means it uses a different search
approach for every model).
The return values of the endpoint are then prefixed with the id of the
contenttype of the results, separated by an underscore.
Example:
Using this endpoint with the parameters:
?entities=apis_ontology.person&entities=apis_ontology.place&q=ammer
gives you all the persons and places that have `ammer` in their names
and labels.
"""
[docs]
def get_result_value(self, result) -> str:
content_type = ContentType.objects.get_for_model(result)
return f"{content_type.id}_" + super().get_result_value(result)
[docs]
def get_queryset(self):
q = Q()
if entities := self.request.GET.getlist("entities"):
for entity in entities:
app_label, model = entity.split(".")
content_type = get_object_or_404(
ContentType, app_label=app_label, model=model
)
name = (
RootObject.objects_inheritance.get_queryset()._get_ancestors_path(
content_type.model_class()
)
)
q |= Q(**{f"{name}__isnull": False}) & generate_search_filter(
content_type.model_class(), self.q, prefix=f"{name}__"
)
return RootObject.objects_inheritance.select_subclasses().filter(q)