Как получить доступ к разным моделям в шаблоне? - PullRequest
Новая
       11

Как получить доступ к разным моделям в шаблоне?

0 голосов
/ 04 мая 2019

У меня есть 3 модели

Нужно ли менять соединения в моделях и сделать поле ключа не идентификатором, а именем? 

class Category(models.Model):
    name = models.CharField(max_length=150, unique=True)
    description = models.CharField(max_length=250)

class Company(models.Model):
    name = models.CharField(max_length=150, unique=True)
    country = models.CharField(max_length=50)

class Motobike(models.Model):
    name = models.CharField(max_length=150)
    company = models.ForeignKey('Company', on_delete=models.CASCADE)
    category = models.ForeignKey('Category', on_delete=models.CASCADE)

И тесты: 

def test_category(setup):
    client = Client()
    category_id = Category.objects.get(name='Мотоциклы').id
    response = client.get(f'/categories/{category_id}/')
    assert response.status_code == 200
    response_data = json.loads(response.content.decode('utf-8'))
    assert len(response_data) == 2
    assert response_data[1]['name'] == 'Ninja Turbo'
    assert response_data[1]['vendor'] == 'Kawasaki'
    assert response_data[1]['category'] == 'Мотоциклы'
    assert response_data[1]['description'] == ''

    response = client.get(f'/categories/25/')
    assert response.status_code == 404

В связи с этим я делаю так: 

 class CategoryView(DetailView):
     model = Category
     template_name = 'bikes_site/categories_detail.html'

     def get_context_data(self, id, **kwargs):
        context = get_object_or_404(self.model, id)
        context['motobikes'] =   Motobike.objects.filter(category_id=id).all()
    return context

Я получаю сообщение об ошибке:

get_context_data() отсутствует 1 обязательный позиционный аргумент: 'id'

1 Ответ

1 голос
/ 04 мая 2019

Сигнатура функции для get_context_data неверна, она должна быть 

def get_context_data(self, **kwargs):
    //todo

вашему подробному виду должно нравиться это 

 class CategoryView(DetailView):
     model = Category
     template_name = 'bikes_site/categories_detail.html'
     pk_url_kwarg = "id"

     def get_context_data(self, **kwargs):
       context = super().get_context_data(**kwargs) 
       category = self.get_object()
       context['motobikes'] =   Motobike.objects.filter(category_id=category.pk)
       return context
...