Как реализовать фильтрацию?

Question

[Игорь Малышев]

В простом примере ниже, в файле "city_list.html" необходимо сделать вывод городов для своих стран. Кто знает, как это осуществить, помогите, пожалуйста!
models.py

class Country(models.Model):
	name = models.CharField(max_length=50)
	country_slug = models.SlugField(max_length=50, unique=True)

	def __str__(self):
		return self.name

	def get_absolute_url(self):
		return reverse('country', kwargs={'country_slug': self.country_slug})

class City(models.Model):
	city_country = models.ForeignKey(Country)
	name = models.CharField(max_length=50)
	city_slug = models.SlugField(max_length=50, unique=True)

	def __str__(self):
		return self.name

urls.py

urlpatterns = [
    url(r'^admin/', include(admin.site.urls)),
    url(r'^$', 'example.views.home'),
    url(r'^(?P<country_slug>[-\w]+)$', CityListView.as_view(), name='country'),
]

views.py

def home(request):
	return render_to_response("home.html", {'countries': Country.objects.all()})


class CityListView(ListView):
	model = City
	context_object_name = "cities"
	template_name = "city_list.html"

home.html

{% for country in countries %}
<li><a href="{{ country.get_absolute_url }}">{{ country.name }}</a></li>
	{% endfor %}

Comments

[Roman K]

Не ясно, что нужно.

[Игорь Малышев]

Roman Kitaev: в базе данных есть список стран, который выведен на главную страницу. Через ForeignKey к каждой стране привязаны города. Необходимо, чтобы при клике на определенную страну осуществлялся выводился список принадлежащих ей городов.

Answer 1

[Oscar Django]

class CityListView(ListView):
    model = City
    context_object_name = "cities"
    template_name = "city_list.html"

    def get_queryset(self, *args, **kwargs):
        qs = super(CityListView, self).get_queryset(*args, **kwargs)

        country = Country.objects.get(country_slug=self.kwargs.get('country_slug'))
        return qs.filter(city_country=country)

Comments

[Игорь Малышев]

дописал Ваш метод, получил ошибку name 'country_slug' is not defined

[Roman K]

Игорь Малышев: Значит, что-то неправильно написал
Oscar Django: Я бы ещё проверку на существование слаги поставил. 404 выдавать логичнее, чем 500.

[Игорь Малышев]

Спасибо!

[Oscar Django]

Roman Kitaev Ну я самый простой вариант привел, чтобы у человека заработало. Чем больше кода тем больше способов ошибиться.

[Roman K]

Oscar Django: Дык там кода не больше.
Заменить

country = Country.objects.get(country_slug=self.kwargs.get('country_slug'))

На

country = get_object_or_404(Country, country_slug=self.kwargs.get('country_slug'))