Как написать CreateView для модели с ForeignKey?

Question

[Fedor_PV]

Здравствуйте, помоги плиз новичку - есть такая модель: страны и города внутри стран, каждый город это Foreign Key для определенной страны. Очень просто создал CreateView для страны, вопросов нет, но не могу создать CreateView для города - мешает Foreign Key. В документации примеров нет - есть только туманный пример про внешний ключ автора

form.instance.created_by = self.request.user

Этот пример не помогает в данном случае. Как быть?

models.py

class Countries(models.Model):
....
    country_name=models.CharField(max_length=30, choices=COUNTRY_CHOICE)
....

class Cities(models.Model):
    country=models.ForeignKey(Countries, on_delete=models.CASCADE)
    city_name=models.CharField(max_length=100)
....

urls.ry

path('countries/<int: country_id>/new_city/', login_required(CityCreate.as_view()), name='new_city')

view.py

class CityCreate(CreateView):
    model=Cities
    fields=['city_name', 'tags']
    success_url=reverse_lazy('countries:cities')
    template_name='countries/new_city.html'

def get(self, request, *args, **kwargs):
    self.country=Countries.objects.get(id=self.kwargs['country_id'])
    return super(CityCreate, self).get(request, *args, **kwargs) 

def get_context_data(self, **kwargs):
    context=super(CityCreate, self).get_context_data(**kwargs)
    context['country']=self.country
    return context

def form_valid(self, form):
    form.instance.country=self.country
    return super().form_valid(form)

countries/new_city.html:

country: {{ country }}

<form action="{% url 'countries:new_city' country.id %}" method='post'  class="form">
 {% csrf_token %}

 </form>

Answer 1

[Fedor_PV]

В общем вот код, который работает (другие подходы не помогли):

class CityCreate(CreateView):
    model = Cities
    fields = ['city_name', 'tags']
    template_name = 'countries/new_city.html'

    def get_success_url(self):
        return reverse('countries:cities', args=[self.country.id])

    def get_context_data(self, **kwargs):
        self.country = get_object_or_404(Countries, id=self.kwargs['country_id'])
        kwargs['country'] = self.country
        return super().get_context_data(**kwargs)

    def form_valid(self, form):
        self.country = get_object_or_404(Countries, id=self.kwargs['country_id'])
        form.instance.country = self.country
        messages.success(self.request, 'The city has been added to the list of visited places, thank you') 
        return super().form_valid(form)

Answer 2

[Евгений]

Во-первых, соблюдайте, пожалуйста PEP-8 и Django Coding Style. Это уважение к тем, кому приходится читать ваш код. Также, обратите внимание, что обычно модели называются в единственном числе: City, а не Cities. Country, а не Countries.

Во-вторых, при данном использовании CreateView форма создается автоматически. Вам же нужна своя форма, которая будет назначать страну. Создайте форму для City, в конструкторе которой будет сохраняться Country из запроса. А в CreateView вместо атрибута fields используйте form_class для использования вашей формы.

Comments

[Fedor_PV]

Евгений, спасибо большое за ответ. А подскажите - в данном случае, когда создаешь "свою форму" в файле forms.py по прежнему нужно использовать наследование от forms.ModelForm или использовать forms.Form?

Получается класс, определяющий форму в forms.py может принимать переменную country.id как и функция во вьюшке ?

[Евгений]

Fedor_PV, Вам нужно обязательно изучить официальную документацию Django. Вы задаете вопросы, которые, в общем, тривиальны — значит, вы не читали официальный tutorial Django и официальную документацию. Без этого невозможно хорошо писать используя Django.

В вашем случае лучше/проще использовать ModelForm. Класс формы может принимать начальные значения, в которых можно передать country_id. https://docs.djangoproject.com/en/2.0/topics/forms...

Кстати, мой ответ неверный, можно даже обойтись без собственного класса формы — достаточно переопределить метод get_initial, чтобы передать в форму нужное значение country.

[Fedor_PV]

Евгений, ничего не получается. Ровным счетом ничего. Программа бесконечно ругается на все подряд, при попытках влезть в непонятный get_initial. Более-менее удалось решить вопрос простым оверрайдом def form_valid(self, form) - не знаю как тут правильно код вставлять, чтобы было красиво по РЕР-8, в комментариях не получается:

class CityCreate(CreateView):

model=Cities
fields=['city_name', 'tags']
success_url=reverse_lazy('countries:cities', kwargs={'country_id': ???})
template_name='countries/new_city.html'

def form_valid(self, form):
self.country=Countries.objects.get(id=self.kwargs['country_id'])
form.instance.country=self.country
return super(CityCreate, self).form_valid(form)

Такой оверрайд наконец то правильно привязывает город к нужной стране, сохраняет данные в базу, как бы все хорошо. НО остался вопрос как быть с перенаправлением обратно на страницу городов, принадлежащих одной стране? То есть как передать функции reverse_lazy параметр country_id ? В моем коде выше я выделил его тремя знаками вопроса. Ничего не получается, никакие self.kwargs['country_id'] программа не понимает, я не могу найти ответ нигде.

[Евгений]

Эту задачу можно разными способами решить. Через initial мне кажется наименее корявым, но остальные, в общем, не сильно хуже. Не знаю, почему у вас не получилось, возможно, забыли поле для country указать. Initial только на поля формы влияет, а не модели.

Еще раз повторюсь: читать документацию. Там все это написано, и про url'ы тоже. Простым понятным языком, с примерами.

success_url здесь не годится, надо переопределять метод get_success_url. В нем уже можно передать reverse нужный country_id.

[Fedor_PV]

В общем вот код, который работает:

class CityCreate(CreateView):
    model = Cities
    fields = ['city_name', 'tags']
    template_name = 'countries/new_city.html'

    def get_success_url(self):
        return reverse('countries:cities', args=[self.country.id])

    def get_context_data(self, **kwargs):
        self.country = get_object_or_404(Countries, id=self.kwargs['country_id'])
        kwargs['country'] = self.country
        return super().get_context_data(**kwargs)

    def form_valid(self, form):
        self.country = get_object_or_404(Countries, id=self.kwargs['country_id'])
        form.instance.country = self.country
        messages.success(self.request, 'The city has been added to the list of visited places, thank you') 
        return super().form_valid(form)

[Евгений]

Fedor_PV, вместо Countries.objects.get во view лучше использовать django.shortcuts.get_object_or_404.
В таком случае, если где-то произошла какая-то ошибка, то пользователь увидит не Internal Server Error, а ошибку 404, что во многих случаях понятнее и предпочтительнее.

Вместо messages.add_message можно написать короче: messages.success. Также есть пакеты расширения для Django, которые автоматически добавляют success message (SuccessMessageMixin).

Если вы используете Python 3 (который, собс-но, и нужно сейчас использовать везде, Python 2 уже unsupported, для новых проектов его использовать категорически нельзя), то вместо: super(Klass, self) можно просто писать super().

И прочтите же PEP-8, наконец. Это первый документ после Python Tutorial, который должен прочитать каждый программист на Python.

По обеим сторонам от оператора присваивания должны стоять пробелы (в отличие от вызова с kwargs).

[Fedor_PV]

Евгений, я все исправил, спасибо вам большое!