Как динамически установить начальное значение ForeignKey в CreateView? - PullRequest
Новая
       21

Как динамически установить начальное значение ForeignKey в CreateView?

1 голос
/ 10 июля 2020

Я хочу динамически установить начальное значение ForeignKey в CreateView. Но есть ли простой способ сделать это? И я пробовал как это (как ответ по этой ссылке). Но это не работает. Как передать альбом (ForeignKey) в поле CreateView?

models.py 

class Album(models.Model):
    credit = models.CharField(max_length=250)
    album_title = models.CharField(max_length=100)
    logo = models.FileField()

    def get_absolute_url(self):
        return reverse('picture:detail', kwargs={'pk': self.pk})

    def __str__(self):
        return self.album_title + ' - ' + self.credit


class Item(models.Model):
    album = models.ForeignKey(Album, on_delete=models.CASCADE)
    file_type = models.CharField(choices=TYPE_CHOICES, max_length=1)
    caption = models.CharField(max_length=100)

    class Meta:
        ordering = ('upload_date', 'caption')

    def get_absolute_url(self):
        return reverse('picture:item-detail', kwargs={ 'id': self.album_id , 'pk': self.pk})

    def __str__(self):
        return self.caption

views.py 

class ItemCreate(CreateView):
    model = Item
    fields = ['album', 'file_type', 'caption']

    def get_initial(self):
        album = get_object_or_404(Album, pk=self.kwargs.get('album.pk'))
        return {
            'album': album,
            'file_type': 't',
        }

urls.py 

urlpatterns = [
# /picture/<album_id>/
    url(r'^(?P<pk>[0-9]+)/$', views.DetailView.as_view(), name='detail'),
# /picture/<album_id>/<pic_id>
    url(r'^(?P<id>[0-9]+)/(?P<pk>[0-9]+)/$', views.ItemDetailView.as_view(), name='item-detail'),
# /picture/<album_id>/pic/add
    url(r'^(?P<id>[0-9]+)/pic/add/$', views.ItemCreate.as_view(), name='item-add'),
]

1 Ответ

0 голосов
/ 12 июля 2020

Если вы хотите получить объект (album), где album.pk == id (id параметр URL-адреса 'item-add'), попробуйте это, 

class ItemCreate(CreateView):
    model = Item
    fields = ['album', 'file_type', 'caption']

    def get_initial(self):
        album = get_object_or_404(Album, pk=self.kwargs['id'])
        return {
            'album': album,
            'file_type': 't',
        }
...