generated from Code-Institute-Org/gitpod-full-template
-
Notifications
You must be signed in to change notification settings - Fork 4
/
05-trees-forms.txt
37 lines (30 loc) · 1.07 KB
/
05-trees-forms.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
Check results
=============
Code
=============
"""form for super user to add products"""
# pylint: disable=no-member
from django import forms
from .widgets import CustomClearableFileInput
from .models import Tree, Feature, Enviroment
class TreeForm(forms.ModelForm):
"""
Form to add trees to the shop
"""
class Meta:
"""
Meta data for product form
"""
model = Tree
fields = '__all__'
# replaces default styling with customised styling of the image input
image = forms.ImageField(
label='Image', required=False, widget=CustomClearableFileInput)
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
features = Feature.objects.all()
f_friendly_names = [(f.id, f.get_friendly_name()) for f in features]
enviroments = Enviroment.objects.all()
e_friendly_names = [(e.id, e.get_friendly_name()) for e in enviroments]
self.fields['feature'].choices = f_friendly_names
self.fields['enviroment'].choices = e_friendly_names