From 39686b58074d87f738b5deeee3fcd2691fc91747 Mon Sep 17 00:00:00 2001
From: Xavier Fiechter <xavierfiechter@gmail.com>
Date: Fri, 3 Feb 2023 18:38:47 +0100
Subject: [PATCH] .

---
 labelbase/forms.py | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

diff --git a/labelbase/forms.py b/labelbase/forms.py
index 7e648ea..ab72dc1 100644
--- a/labelbase/forms.py
+++ b/labelbase/forms.py
@@ -1,9 +1,16 @@
 
-from django.forms import ModelForm
+from django import forms
+from .models import Label, Labelbase
 
-from .models import Label
-
-class LabelForm(ModelForm):
+class LabelForm(forms.ModelForm):
     class Meta:
         model = Label
         fields = ['labelbase', 'type', 'ref', 'label']
+    def __init__(self, *args, **kwargs):
+        self.request = kwargs.pop('request')
+        super(LabelForm, self).__init__(*args, **kwargs)
+        self.fields['labelbase'] = forms.ModelChoiceField(
+            queryset=Labelbase.objects.filter(user_id=request.user.id),
+            required=True,
+            widget=forms.Select(attrs={'class': 'chzn-select'}),
+        )