Django-фильтр как показать только некоторые объекты в выпадающем списке?
Мой сайт работает следующим образом: каждый менеджер может иметь несколько подменеджеров, эти подменеджеры могут иметь несколько агентов (таким образом, агенты косвенно связаны с менеджером, см. models.py, чтобы лучше понять отношения между ними). Я хочу показывать на странице профиля менеджера (см. views.py) все MembershipCard, созданные его/ее связанными агентами. Я пытаюсь реализовать фильтр для поиска, например, карт, созданных определенным агентом, у меня получается, но я хотел бы показывать в выпадающем списке только агентов, связанных с менеджером, сейчас выпадающий список показывает всех агентов в базе данных
models.py
class StandardProfile(models.Model):
name = models.CharField(max_length=200, null=True)
surname = models.CharField(max_length=200, null=True)
phone_number = models.CharField(max_length=200, null=True)
class Meta:
abstract = True
class Manager(StandardProfile):
user = models.OneToOneField(User, null=True, on_delete=models.CASCADE)
class SubManager(StandardProfile):
user = models.OneToOneField(User, null=True, on_delete=models.CASCADE)
manager = models.ForeignKey(Capo, null=True, on_delete = models.SET_NULL)
class Agent(StandardProfile):
user = models.OneToOneField(User, null=True, on_delete=models.CASCADE)
manager = models.ForeignKey(Manager, null=True, on_delete = models.SET_NULL)
subManager = models.ForeignKey(SubManager, null=True, blank=True, on_delete = models.SET_NULL)
class MembershipCard(models.Model):
agent = models.ForeignKey(Agent, null=True,blank=True, on_delete = models.SET_NULL)
client = models.ForeignKey(Client, null=True,blank=True, on_delete = models.SET_NULL)
creation_date = models.DateTimeField(auto_now_add=True, null=True)
activation_date = models.DateTimeField(null=True,blank=True)
expiration_date = models.DateTimeField(null=True,blank=True)
views.py
@login_required(login_url='login')
def profilePage(request, pk): #www.mysite.com/profilePage/<pk>
user = User.objects.get(id=pk) #getting the user from <pk>
cards = MembershipCard.objects.filter(agent__manager=user.manager)
myFilter = MembershipCardFilter(request.GET,queryset=cards,user=user)
cards = myFilter.qs
#page_obj is used for Pagination, and contains the cards, i removed this part of code for better readability, can add it if needed
context = {'page_obj': page_obj,"user": user,"myFilter":myFilter}
return render(request, 'polls/profilePage.html',context)
filters.py
class MembershipCardFilter(django_filters.FilterSet):
class Meta:
model = MembershipCard
fields = ['agent','agent__subManager']
exclude = ['creation_date']
Читая ответы на похожие вопросы, я думаю, что мне нужно изменить метод __init__
в классе CardFilter, я пытался адаптировать некоторые ответы к моему случаю, но это не сработало по некоторым причинам. Любой ответ/комментарий будет принят с благодарностью!
PS: Я не знаю, понятно ли название, не стесняйтесь предложить лучшее
Вы можете попробовать скормить выпадающий список агентов во время инициализации следующим образом (не проверено!):
class MembershipCardFilter(django_filters.FilterSet):
agent= django_filters.ModelChoiceFilter(
queryset=Agent.objects.none(),
)
class Meta:
model = MembershipCard
fields = ['agent','agent__subManager']
exclude = ['creation_date']
def __init__(self, *args, **kwargs):
user = kwargs.get("user")
agents = Agent.objects.filter(manager__user=user)
super().__init__(*args, **kwargs)
self.filters["agent"].queryset = agents