Переопределить импортированные переменные класса - django / python - PullRequest
0 голосов
/ 18 февраля 2019

Мне нужно переопределить переменные (или передать динамические данные) в импортируемый класс.

filters.py

import django_filters
from .models import Gate, Tram, OperationArea, Bogie
from distutils.util import strtobool
from django import forms


class GateFilter(django_filters.FilterSet):

    # Prepare dynamic lists with choices
    tram_list = [(id, number) for id, number in Tram.objects.all().values_list('id', 'number')]
    bogie_list = [(id, number) for id, number in Bogie.objects.all().values_list('id', 'number')]
    area_list = [(id, area) for id, area in OperationArea.objects.all().values_list('id', 'area')]
    # Generate fields
    tram = django_filters.MultipleChoiceFilter(choices=tram_list, label=u'Tramwaj')
    car = django_filters.MultipleChoiceFilter(choices=Gate.CAR_SYMBOLS, label=u'Człon')
    bogie = django_filters.MultipleChoiceFilter(choices=bogie_list, label=u'Wózek')
    bogie_type = django_filters.MultipleChoiceFilter(choices=Gate.BOGIE_TYPES, label=u'Typ wózka')
    area = django_filters.MultipleChoiceFilter(choices=area_list, label=u'Obszar')
    operation_no = django_filters.CharFilter(label=u'Numer operacji', widget=forms.TextInput(attrs={'size': '16px'}))
    status = django_filters.MultipleChoiceFilter(choices=Gate.GATE_STATUSES, label=u'Status')
    rating = django_filters.MultipleChoiceFilter(choices=Gate.GATE_GRADES, label=u'Ocena')

    class Meta:
        pass

views.py

from .filters import GateFilter

class GateListView(generic.ListView):

    queryset = None
    gate_type = None
    template_name = 'qapp/gate/list.html'
    context_object_name = 'gate_list'
    paginate_by = 20

    def get_queryset(self):
        # Type is stored in database as big-letter word, so 'bjc' != 'BJC'.
        if self.gate_type.upper() == 'BJW':
            ordering = ['bogie', 'bogie_type']
        else:
            ordering = ['tram', 'car']
        queryset = Gate.objects.filter(type=self.gate_type.upper()).order_by(*ordering)
        self.gate_list = GateFilter(self.request.GET, queryset=queryset)
        return self.gate_list.qs.distinct()

    def get_context_data(self, **kwargs):
        context = super(GateListView, self).get_context_data(**kwargs)
        # Return Gate.type to template.
        context['gate_type'] = self.gate_type
        # Return object (for generating form) to template.
        context['gate_list_filter'] = self.gate_list
        return context

Как выв filters.py можно увидеть, что данные для переменных tram_list, bogie_list и area_list являются динамическими (извлекаются из базы данных).

Но при импорте этого класса в views.py эти данные становятся статическими.

Я попытался переопределить эти значения:

  • с помощью декоратора @classmethod в классе GateFilter и вызов его перед установкой self.gate_list объекта,
  • в views.py с использованием GateFilter.tram_list (и остальных),

Не повезло.

Я не могу использовать функцию reload () из-за типа импорта ( из .filters import GateFilter )).

В настоящее время для списков обновлений в filters.py мне нужно перезапустить все приложение.Это неприемлемо для бизнес-логики моего приложения.

1 Ответ

0 голосов
/ 18 февраля 2019

Это неправильный подход.Скорее, вы должны использовать фильтры, которые знают о наборе запросов и оценивают их при необходимости: ModelChoiceFilter и ModelMultipleChoiceFilter.

class GateFilter(django_filters.FilterSet):
    team = django_filters.ModelMultipleChoiceFilter(queryset=Tram.objects.all())
...