django：店舗がスタッフを新規登録する時に、店舗IDを自動的に紐づけたい

前提・実現したいこと

店舗管理者が店舗スタッフを新規登録する際に、自動的に店舗IDが紐づけられるようにしたいです。

店舗管理者の登録は、djangoのデフォルトのUserモデルを使っています。
店舗管理者と店舗は、ManyToManyFieldで、
店舗とスタッフは、ForeignKeyで紐づけています。

該当のソースコード

※不要そうなところは省略してあります。
models.py

python
1from django.contrib.auth.models import User
2...
3
4class Store(models.Model):
5    name = models.CharField(verbose_name='店舗名', max_length=255)
6    user = models.ManyToManyField(User, verbose_name='店舗管理者')
7
8
9class Staff(models.Model):
10    name = models.CharField(verbose_name='名前', max_length=255)
11    store_id = models.ForeignKey(
12        Store, verbose_name='店舗ID', on_delete=models.CASCADE,)

views.py

python
1class StaffCreateView(CreateView):
2    model = Staff
3    form_class = StaffCreateForm
4    template_name = '../staff_create.html'
5    success_url = reverse_lazy('sample:index')

forms.py

python
1class StaffCreateForm(forms.ModelForm):
2    
3    class Meta:
4        model = Staff
5        fields = '__all__'

発生している問題・エラーメッセージ　　試したこと

今のままでは、スタッフ登録はできるのですが、店舗IDを自分で選ぶ仕組みになっています。
これを、自分で選択することなく、店舗IDを紐づけたいです。

fields = 'all'の部分をexclude = ['store_id']にすると、
自分で店舗IDを選ぶ必要はなくなるのですが、以下のエラーが出てしまいます。

列"store_id_id"内のNULL値はNOT NULL制約違反です

お分かりになる方がおりましたら、何卒宜しくお願い致します。

gh640

2018/12/18 12:34

方向性としては `StaffCreateForm` か `StaffCreateView` でベースクラスのメソッドを上書きして実現する形になると思いますが、自動的に紐付ける店舗 ID というのをどこから持ってくる形にするのかによって実装しやすい方法は変わってくるものと思います。例えばよくある形はスタッフ作成フォームの URL に店舗 ID が含まれていてそれを使う形でしょうか。そのあたりのご要望をご説明されると具体的な回答がもらえやすくなるのではないかと思います。

hiranohirano

2018/12/18 13:07

ご回答ありがとうございます。スタッフ作成フォームのURLに含まれる店舗IDを使う形で考えています。 urls.pyのurlpatternsで、新規作成のURLは下記のようにしています。 path('user/store<int:pk>/staff/create', StaffCreateView.as_view(), name='user_staff_create'),

gh640

2018/12/18 13:42

ご返信ありがとうございます。ご返信いただいた内容を受けて回答させていただきましたので、またご覧になってみください。

行動規範の内容に同意します

回答1件

ベストアンサー

ご返信ありがとうございます。

バージョンに言及されていませんが、 Python 3 / Django 2.1.x と仮定して回答させていただきますね。

urls.pyのurlpatternsで、新規作成のURLは下記のようにしています。
path('user/storeint:pk/staff/create', StaffCreateView.as_view(), name='user_staff_create'),

とのことですので、 dispatch() メソッドの引数の kwargs から URL パラメータ pk が取得できるものと思います。それを適当な名前のアトリビュート（＝祖先クラスで使われていない & 将来的にも衝突の可能性の低い名前のアトリビュート）に格納しておけば後々利用できるようになります。

また、フォームの post 時にユーザの入力値が不正でなければ form_valid() というメソッドが呼ばれ、その中で新しいモデルレコードが DB に保存されるようになっています。ですので、 form_valid() を上書きして、（ dispatch() であらかじめ取得しておいた） store をその中でセットするようにすると、おそらくお望みのことができるものと思います。

実装サンプルです:

python
1# （省略）
2from django.shortcuts import get_object_or_404
3
4
5class StaffCreateView(CreateView):
6    # （省略）
7
8    # URL のパラメータ `pk` から対象の `Store` を取得する
9    def dispatch(self, request, *args, **kwargs):
10        self.target_store = get_object_or_404(Store, pk=kwargs['pk'])
11        return super().dispatch(request, *args, **kwargs)
12
13    # `staff` の保存直前に URL から取得した `store` フィールドをセットする
14    def form_valid(self, form):
15        form.instance.store = self.target_store
16        return super().form_valid(form)