first draft of using class-based views

Author: natea

sim/seedbank/forms.py

@@ -0,0 +1,17 @@
+from django import forms
+from django.forms.models import inlineformset_factory
+
+from seedbank.models import Seed, Location
+
+# inlineformset_factory creates a Class from a parent model (Seed)
+# to a child model (Location)
+# SeedLocationFormSet = inlineformset_factory(
+#     Seed,
+#     Location,
+# )
+
+
+class SeedForm(forms.ModelForm):
+
+    class Meta:
+        model = Seed

sim/seedbank/models.py

@@ -3,47 +3,71 @@
 # Create your models here.
 
 class Location(models.Model):
+    """
+    TODO: Description of class
+    """
     location_name = models.CharField(max_length=45, blank=False)
     location_code = models.CharField(max_length=45, blank=False)
     location_type = models.CharField(max_length=45)
 
 
 class Family(models.Model):
+    """
+    TODO: Description of class
+    """
     family = models.CharField(max_length=45, blank=False)
 
 
 class ScientificName(models.Model):
+    """
+    TODO: Description of class
+    """
     scientific_name = models.CharField(max_length=45, blank=False)
 
 
 class Variety(models.Model):
+    """
+    TODO: Description of class
+    """
     variety = models.CharField(max_length=45, blank=False)
     seed_color = models.CharField(max_length=75)
-    parts_to_harvest =  models.TextField()
+    parts_to_harvest = models.TextField()
     unique_characteristics = models.TextField()
     planting_intructions = models.TextField()
 
     scientific_name = models.ForeignKey(ScientificName)
 
 
 class Germination(models.Model):
+    """
+    TODO: Description of class
+    """
     germ_rate = models.DecimalField(decimal_places=2, max_digits=2, blank=False)
     germ_date = models.DateField(blank=False)
     germ_method = models.TextField()
 
 
 class CommonName(models.Model):
+    """
+    TODO: Description of class
+    """
     common_name = models.CharField(max_length=45)
     langs = (("English", "English"), ("Thai", "Thai"))
     language = models.CharField(max_length=45, choices=langs, default="English")
     scientific_name = models.ForeignKey(ScientificName)
 
 
 class SupplierType(models.Model):
+    """
+    TODO: Description of class
+    """
     supplier_type = models.CharField(max_length=45)
 
 
 class Supplier(models.Model):
+    """
+    TODO: Description of class
+    """
     supplier_name = models.CharField(max_length=45, blank=False)
     contact_first_name = models.CharField(max_length=45)
     contact_last_name = models.CharField(max_length=45)
@@ -64,17 +88,22 @@ class Supplier(models.Model):
 
 
 class Seed(models.Model):
+    """
+    TODO: Description of class
+    """
     accession_num = models.CharField(max_length=8, blank=False)
     lot_num = models.CharField(max_length=5, blank=False)
     aquisition_date = models.DateField(blank=False)
-    cost = models.DecimalField(decimal_places=2, max_digits=6, blank=False)  
+    cost = models.DecimalField(decimal_places=2, max_digits=6, blank=False)
 
     aquisition_location = models.ForeignKey(Location)
     family = models.ForeignKey(Family)
     variety = models.ForeignKey(Variety)
     scientific_name = models.ForeignKey(ScientificName)
     germination = models.ForeignKey(Germination)
 
-    # I think this one is supposed to be here... ?  
-    supplier = models.ForeignKey(Supplier) 
+    # I think this one is supposed to be here... ?
+    supplier = models.ForeignKey(Supplier)
 
+    def get_absolute_url(self):
+        return reverse('seeds-view', kwargs={'pk': self.id})

sim/seedbank/templates/delete_seed.html

@@ -0,0 +1,10 @@
+<h1>Delete Seed</h1>
+
+<p>Are you sure you want to delete the seed {{ seed }}?</p>
+
+<form action="{% url "seeds-delete" pk=seed.id %}" method="POST">
+	{% csrf_token %}
+
+	<input type="submit" value="Yes, delete." />
+	<a href="{% url "seeds-list" %}">No, cancel.</a>
+</form>

sim/seedbank/templates/edit_seed.html

@@ -0,0 +1,19 @@
+{% if seed.id %}
+<h1>Edit Seed</h1>
+{% else %}
+<h1>Add Seed</h1>
+{% endif %}
+
+<form action="{{ action }}" method="POST">
+	{% csrf_token %}
+	<ul>
+		{{ form.as_ul }}
+	</ul>
+	<input type="submit" value="Save" />
+</form>
+
+{% if seed.id %}
+<a href="{% url "seeds-delete" pk=seed.id %}">Delete</a>
+{% endif %}
+
+<a href="{% url "seeds-list" %}">Back to list</a>

sim/seedbank/templates/seed.html

@@ -0,0 +1,21 @@
+<h1>{{ seed }}</h1>
+
+<p>Accession number: {{ seed.accession_num }}</p>
+<p>Lot number: {{ seed.lot_num }}</p>
+<p>Acquisition date: {{ seed.aquisition_date }}</p>
+<p>Cost: {{ seed.cost }}</p>
+<p>Location: {{ seed.aquisition_location }}</p>
+<p>Family: {{ seed.family }}</p>
+<p>Variety: {{ seed.variety }}</p>
+<p>Scientific name: {{ seed.scientific_name }}</p>
+<p>Germination: {{ seed.germination }}</p>
+<p>Supplier: {{ seed.supplier }}</p>
+
+{# <ul>
+	{% for location in seed.location_set.all %}
+	<li>{{ location.name }}<br />
+		{{ location.code }}<br />
+		{{ location.type }}
+	</li>
+	{% endfor %}
+</ul> #}

sim/seedbank/templates/seed_list.html

@@ -0,0 +1,11 @@
+<h1>Seeds</h1>
+
+<ul>
+	{% for seed in object_list %}
+	<li><a href="{{ seed.get_absolute_url }}">{{ seed }}</a>
+		(<a href="{% url "seeds-edit" pk=seed.id %}">edit</a>)
+	</li>
+	{% endfor %}
+</ul>
+
+<a href="{% url 'seeds-new' %}">add seed</a>

sim/seedbank/urls.py

@@ -0,0 +1,20 @@
+from django.conf.urls import patterns, url
+
+from seedbank import views
+
+urlpatterns = patterns('',
+
+    url(r'^$', views.ListSeedView.as_view(),
+        name='seeds-list',),
+    url(r'^new$', views.CreateSeedView.as_view(),
+        name='seeds-new',),
+    url(r'^edit/(?P<pk>\d+)/$', views.UpdateSeedView.as_view(),
+        name='seeds-edit',),
+    url(r'^delete/(?P<pk>\d+)/$', views.DeleteSeedView.as_view(),
+        name='seeds-delete',),
+    url(r'^(?P<pk>\d+)/$', views.SeedView.as_view(),
+        name='seeds-view',),
+    # url(r'^edit/(?P<pk>\d+)/locations$', views.EditSeedLocationView.as_view(),
+    #     name='seeds-edit-locations',),
+
+)

sim/seedbank/views.py

@@ -1 +1,73 @@
 # Create your views here.
+
+from django.views.generic import (
+    ListView,
+    CreateView,
+    UpdateView,
+    DeleteView,
+    DetailView,
+)
+
+from django.core.urlresolvers import reverse
+
+from seedbank.models import Seed
+from seedbank import forms
+
+
+class ListSeedView(ListView):
+    model = Seed
+    template_name = 'seed_list.html'
+
+
+class CreateSeedView(CreateView):
+    model = Seed
+    template_name = 'edit_seed.html'
+    form_class = forms.SeedForm
+
+    def get_success_url(self):
+        return reverse('seeds-list')
+
+    def get_context_data(self, **kwargs):
+        context = super(CreateSeedView, self).get_context_data(**kwargs)
+        context['action'] = reverse('seeds-new')
+
+        return context
+
+
+class UpdateSeedView(UpdateView):
+    model = Seed
+    template_name = 'edit_seed.html'
+    form_class = forms.SeedForm
+
+    def get_success_url(self):
+        return reverse('seeds-list')
+
+    def get_context_data(self, **kwargs):
+        context = super(UpdateSeedView, self).get_context_data(**kwargs)
+        context['action'] = reverse('seeds-edit',
+                                    kwargs={'pk': self.get_object().id})
+
+        return context
+
+
+class DeleteSeedView(DeleteView):
+    model = Seed
+    template_name = 'delete_seed.html'
+
+    def get_success_url(self):
+        return reverse('seeds-list')
+
+
+class SeedView(DetailView):
+    model = Seed
+    template_name = 'seed.html'
+
+
+# class EditSeedLocationView(UpdateView):
+#     model = Seed
+#     template_name = 'edit_location.html'
+#     form_class = forms.SeedLocationFormSet
+
+#     def get_success_url(self):
+#         # redirect to the Seed view
+#         return self.get_object().get_absolute_url()

sim/sim/urls.py

@@ -5,6 +5,8 @@
 from django.contrib import admin
 admin.autodiscover()
 
+#from seedbank import urls as seedbank_urls
+
 urlpatterns = patterns('',
     url(r'^$', TemplateView.as_view(template_name='base.html')),
 
@@ -17,4 +19,5 @@
 
     # Uncomment the next line to enable the admin:
     url(r'^admin/', include(admin.site.urls)),
+    url(r'^seeds/', include('seedbank.urls')),
 )