【django】CreateViewをidごとに分けて作成したい

Question

djangoで店舗毎に商品を宣伝できるようにするサイトを想定して作ろうとしています。
出品するときに企業名を自動で入力されるようにしたいです。
今のままですと、出品者が企業名を選択して出品するので他の企業の人も勝手に出品できてしまいます。
例えば、企業Aが出品する場合、/create/1/から商品を登録し、企業Bは/create/2/から出品するといったページを分ける方法。
(/create/1/から出品した場合、テーブルにshop_id = 1が入れられる。)
もしくは、認証機能を利用して、企業Aのアカウントでログインしている場合は企業Aをformの企業名に自動で追加する。
という2つの方法を考えました。
しかし、具体的に作ろうと思うと知識不足もあり、難しい状態です。
上記の二つの方法以外でも、良い方法がありましたらよろしくお願いします。

#models.py

from django.db import models
from django.db.models.deletion import PROTECT
from django.db.models.fields import IntegerField
from accounts.models import CustomUser

# Create your models here.
class Shop(models.Model):
  #お店の概要
  user = models.OneToOneField(CustomUser, verbose_name="ユーザー", on_delete=PROTECT)
  shop_name = models.CharField(verbose_name="店の名前", max_length=30, unique=True)
  shop_place = models.CharField(verbose_name="店の場所", max_length=60)
  shop_style = models.CharField(verbose_name="お店の種類", max_length=10)
  shop_describe = models.TextField(verbose_name="お店の詳細説明", max_length=300, default="記載がありません")
  opening_time = models.TextField(verbose_name="お店の営業時間", max_length=300, default="記載がありません")
  website = models.URLField(verbose_name="お店のホームページ", blank=True, null=True)
  #photo1は縦横5:7の画像にするとぴったり
  photo1 = models.ImageField(verbose_name="写真1", blank=True, null=True)
  photo2 = models.ImageField(verbose_name="写真1", blank=True, null=True)
  photo3 = models.ImageField(verbose_name="写真1", blank=True, null=True)

  class Meta:
    verbose_name_plural = "店舗情報"

  def __str__(self):
    return self.shop_name


class Products(models.Model):
  user = models.ForeignKey(CustomUser , verbose_name="ユーザー", on_delete=PROTECT)
  shop = models.ForeignKey(Shop, verbose_name='企業名', on_delete=PROTECT)
  product = models.CharField(verbose_name="商品名", max_length=20)
  price = models.IntegerField(verbose_name="価格")
  product_detail = models.TextField(verbose_name="商品説明", max_length=60, blank=True, null=True)
  #画像サイズは縦横2:3がベスト
  product_photo = models.ImageField(verbose_name="商品写真")

  class Meta:
    verbose_name_plural = "商品情報"

  def __str__(self):
    return self.product

#urls.py

from django.urls import path
from .views import IndexView, ShopDetailView, CreateInfoView

app_name = "shoplist"
urlpatterns = [
  path("", IndexView.as_view(), name="index"),
  path("detail/<int:pk>/", ShopDetailView.as_view(), name="detail"),
  path("create/", CreateInfoView.as_view(), name="create"),
]

#views.py

from django.contrib.auth.mixins import LoginRequiredMixin
from django.shortcuts import render
from django.views import generic
from .models import Shop, Products
from .forms import CreateInfoForm
from django.urls import reverse_lazy
from django.contrib import messages
# Create your views here.


class IndexView(generic.ListView):
  model = Shop
  template_name = "index.html"

class ShopDetailView(generic.DetailView):
    model = Shop
    template_name = "detail.html"

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context['products_list'] = Products.objects.filter(shop=self.object)
        return context

class CreateInfoView(LoginRequiredMixin, generic.CreateView):
  model = Products
  template_name = "create.html"
  form_class = CreateInfoForm
  success_url = reverse_lazy("shoplist:index")

  def form_valid(self, form):
    diary = form.save(commit=False)
    diary.user = self.request.user
    diary.save()
    messages.success(self.request, "作成しました。")
    return super().form_valid(form)

  def form_invalid(self, form): #バリデーションに問題があった場合実行
      messages.error(self.request, "作成に失敗しました。")
      return super().form_invalid(form)

#forms.py

from .models import Products
from django import forms

class CreateInfoForm(forms.ModelForm):
  class Meta:
    model = Products
    fields = ("shop", "product", "product_detail", "price", "product_photo")

    def __init__(self, *args, **kwargs):
      super().__init__(*args, **kwargs)
      for field in self.fields.values():
          field.widget.attrs["class"] = "form-control"

{% extends "base.html" %}
{% load static %}

{% block header %}
    <!-- Responsive navbar-->
    <!-- Header-->
    <header class="bg-dark py-1">
        <div class="container px-4 px-lg-5 my-5">
            <div class="text-center text-white">
                <h1 class="display-4 fw-bolder">商品登録</h1>
                <p class="lead fw-normal text-white-50 mb-0">商品を登録するための仮ページです</p>
            </div>
        </div>
    </header>
{% endblock %}
{% block main %}
<div class="container">
    <div class="row">
        <div class="w-100">
            <form method="POST" enctype="multipart/form-data">
                {% csrf_token %}
                <div class="table-responsive">
                <table class="table my-3">
                    {{ form.as_table }}
                </table>
                <button class="btn btn-primary mb-3" type="submit">作成</button>
            </form>
        </div>
    </div>
</div>
{% endblock %}
    {% block footer %}
    {% endblock %}

Answer 1

ShopモデルとCustomUserモデルが1対1で紐づけていることを考えると、

• フォームのPOST時点ではshopを考えない
• フォームオブジェクトからのProducts保存時に認証済みユーザーをもとにShopを特定して紐付ける

といったアプローチは可能そうな気がします。

# forms.py
class CreateInfoForm(forms.ModelForm):
  class Meta:
    model = Products
+   # shopをフォームプロパティから除外
+   fields = ("product", "product_detail", "price", "product_photo")
-   fields = ("shop", "product", "product_detail", "price", "product_photo")

    def __init__(self, *args, **kwargs):
      super().__init__(*args, **kwargs)
      for field in self.fields.values():
          field.widget.attrs["class"] = "form-control"

# views.py
class CreateInfoView(LoginRequiredMixin, generic.CreateView):
  model = Products
  template_name = "create.html"
  form_class = CreateInfoForm
  success_url = reverse_lazy("shoplist:index")

  def form_valid(self, form):
    diary = form.save(commit=False)
    diary.user = self.request.user
+   # shop をuserから直接参照
+   diary.shop = self.request.user.shop
    diary.save()
    messages.success(self.request, "作成しました。")
    return super().form_valid(form)

  def form_invalid(self, form): #バリデーションに問題があった場合実行
      messages.error(self.request, "作成に失敗しました。")
      return super().form_invalid(form)

Comments

[seyu0930]

ありがとうございます！うまく行きました！