codex 1.6.19__py3-none-any.whl → 1.7.0a0__py3-none-any.whl

codex/_vendor/haystack/forms.py

@@ -1,135 +0,0 @@
-from django import forms
-from django.utils.encoding import smart_str
-from django.utils.text import capfirst
-from django.utils.translation import gettext_lazy as _
-
-from ..haystack import connections
-from ..haystack.constants import DEFAULT_ALIAS
-from ..haystack.query import EmptySearchQuerySet, SearchQuerySet
-from ..haystack.utils import get_model_ct
-from ..haystack.utils.app_loading import haystack_get_model
-
-
-def model_choices(using=DEFAULT_ALIAS):
-    choices = [
-        (get_model_ct(m), capfirst(smart_str(m._meta.verbose_name_plural)))
-        for m in connections[using].get_unified_index().get_indexed_models()
-    ]
-    return sorted(choices, key=lambda x: x[1])
-
-
-class SearchForm(forms.Form):
-    q = forms.CharField(
-        required=False,
-        label=_("Search"),
-        widget=forms.TextInput(attrs={"type": "search"}),
-    )
-
-    def __init__(self, *args, **kwargs):
-        self.searchqueryset = kwargs.pop("searchqueryset", None)
-        self.load_all = kwargs.pop("load_all", False)
-
-        if self.searchqueryset is None:
-            self.searchqueryset = SearchQuerySet()
-
-        super().__init__(*args, **kwargs)
-
-    def no_query_found(self):
-        """
-        Determines the behavior when no query was found.
-
-        By default, no results are returned (``EmptySearchQuerySet``).
-
-        Should you want to show all results, override this method in your
-        own ``SearchForm`` subclass and do ``return self.searchqueryset.all()``.
-        """
-        return EmptySearchQuerySet()
-
-    def search(self):
-        if not self.is_valid():
-            return self.no_query_found()
-
-        if not self.cleaned_data.get("q"):
-            return self.no_query_found()
-
-        sqs = self.searchqueryset.auto_query(self.cleaned_data["q"])
-
-        if self.load_all:
-            sqs = sqs.load_all()
-
-        return sqs
-
-    def get_suggestion(self):
-        if not self.is_valid():
-            return None
-
-        return self.searchqueryset.spelling_suggestion(self.cleaned_data["q"])
-
-
-class HighlightedSearchForm(SearchForm):
-    def search(self):
-        return super().search().highlight()
-
-
-class FacetedSearchForm(SearchForm):
-    def __init__(self, *args, **kwargs):
-        self.selected_facets = kwargs.pop("selected_facets", [])
-        super().__init__(*args, **kwargs)
-
-    def search(self):
-        sqs = super().search()
-
-        # We need to process each facet to ensure that the field name and the
-        # value are quoted correctly and separately:
-        for facet in self.selected_facets:
-            if ":" not in facet:
-                continue
-
-            field, value = facet.split(":", 1)
-
-            if value:
-                sqs = sqs.narrow('%s:"%s"' % (field, sqs.query.clean(value)))
-
-        return sqs
-
-
-class ModelSearchForm(SearchForm):
-    def __init__(self, *args, **kwargs):
-        super().__init__(*args, **kwargs)
-        self.fields["models"] = forms.MultipleChoiceField(
-            choices=model_choices(),
-            required=False,
-            label=_("Search In"),
-            widget=forms.CheckboxSelectMultiple,
-        )
-
-    def get_models(self):
-        """Return a list of the selected models."""
-        search_models = []
-
-        if self.is_valid():
-            for model in self.cleaned_data["models"]:
-                search_models.append(haystack_get_model(*model.split(".")))
-
-        return search_models
-
-    def search(self):
-        sqs = super().search()
-        return sqs.models(*self.get_models())
-
-
-class HighlightedModelSearchForm(ModelSearchForm):
-    def search(self):
-        return super().search().highlight()
-
-
-class FacetedModelSearchForm(ModelSearchForm):
-    selected_facets = forms.CharField(required=False, widget=forms.HiddenInput)
-
-    def search(self):
-        sqs = super().search()
-
-        if hasattr(self, "cleaned_data") and self.cleaned_data["selected_facets"]:
-            sqs = sqs.narrow(self.cleaned_data["selected_facets"])
-
-        return sqs.models(*self.get_models())

codex/_vendor/haystack/generic_views.py

@@ -1,141 +0,0 @@
-from django.conf import settings
-from django.core.paginator import Paginator
-from django.views.generic import FormView
-from django.views.generic.edit import FormMixin
-from django.views.generic.list import MultipleObjectMixin
-
-from .forms import FacetedSearchForm, ModelSearchForm
-from .query import SearchQuerySet
-
-RESULTS_PER_PAGE = getattr(settings, "HAYSTACK_SEARCH_RESULTS_PER_PAGE", 20)
-
-
-class SearchMixin(MultipleObjectMixin, FormMixin):
-    """
-    A mixin that allows adding in Haystacks search functionality into
-    another view class.
-
-    This mixin exhibits similar end functionality as the base Haystack search
-    view, but with some important distinctions oriented around greater
-    compatibility with Django's built-in class based views and mixins.
-
-    Normal flow:
-
-        self.request = request
-
-        self.form = self.build_form()
-        self.query = self.get_query()
-        self.results = self.get_results()
-
-        return self.create_response()
-
-    This mixin should:
-
-        1. Make the form
-        2. Get the queryset
-        3. Return the paginated queryset
-
-    """
-
-    template_name = "search/search.html"
-    load_all = True
-    form_class = ModelSearchForm
-    context_object_name = None
-    paginate_by = RESULTS_PER_PAGE
-    paginate_orphans = 0
-    paginator_class = Paginator
-    page_kwarg = "page"
-    form_name = "form"
-    search_field = "q"
-    object_list = None
-
-    def get_queryset(self):
-        if self.queryset is None:
-            self.queryset = SearchQuerySet()
-        return self.queryset
-
-    def get_form_kwargs(self):
-        """
-        Returns the keyword arguments for instantiating the form.
-        """
-        kwargs = {"initial": self.get_initial()}
-        if self.request.method == "GET":
-            kwargs.update({"data": self.request.GET})
-        kwargs.update(
-            {"searchqueryset": self.get_queryset(), "load_all": self.load_all}
-        )
-        return kwargs
-
-    def form_invalid(self, form):
-        context = self.get_context_data(
-            **{self.form_name: form, "object_list": self.get_queryset()}
-        )
-        return self.render_to_response(context)
-
-    def form_valid(self, form):
-        self.queryset = form.search()
-        context = self.get_context_data(
-            **{
-                self.form_name: form,
-                "query": form.cleaned_data.get(self.search_field),
-                "object_list": self.queryset,
-            }
-        )
-        return self.render_to_response(context)
-
-
-class FacetedSearchMixin(SearchMixin):
-    """
-    A mixin that allows adding in a Haystack search functionality with search
-    faceting.
-    """
-
-    form_class = FacetedSearchForm
-    facet_fields = None
-    date_facet_fields = None
-
-    def get_form_kwargs(self):
-        kwargs = super().get_form_kwargs()
-        kwargs.update({"selected_facets": self.request.GET.getlist("selected_facets")})
-        return kwargs
-
-    def get_context_data(self, **kwargs):
-        context = super().get_context_data(**kwargs)
-        context.update({"facets": self.queryset.facet_counts()})
-        return context
-
-    def get_queryset(self):
-        qs = super().get_queryset()
-        for field in self.facet_fields:
-            qs = qs.facet(field)
-
-        if self.date_facet_fields:
-            for field in self.date_facet_fields:
-                qs = qs.date_facet(**field)
-
-        return qs
-
-
-class SearchView(SearchMixin, FormView):
-    """A view class for searching a Haystack managed search index"""
-
-    def get(self, request, *args, **kwargs):
-        """
-        Handles GET requests and instantiates a blank version of the form.
-        """
-        form_class = self.get_form_class()
-        form = self.get_form(form_class)
-
-        if form.is_valid():
-            return self.form_valid(form)
-        else:
-            return self.form_invalid(form)
-
-
-class FacetedSearchView(FacetedSearchMixin, SearchView):
-    """
-    A view class for searching a Haystack managed search index with
-    facets
-    """
-
-    pass