Changeset 4491 for waeup/branches/ulif-rewrite

Timestamp:

31 Jul 2009, 07:19:54 (15 years ago)

Author:

uli

Message:

Add add viewlets/forms for courses in departments.

File:

• waeup/branches/ulif-rewrite/src/waeup/university/department.py (modified) (2 diffs)

waeup/branches/ulif-rewrite/src/waeup/university/department.py

@@ -5 +5 @@
 from zope.component.interfaces import IFactory
 from zope.interface import implementedBy
-from waeup.interfaces import IDepartment
+from waeup.interfaces import IDepartment, ICourse
 from waeup.utils.helpers import getName
 from waeup.widgets.interfaces import ITableProvider
-from waeup.viewlets import MainArea, LeftSidebar, Index, FormWrapMixin, Manage
+from waeup.viewlets import (MainArea, LeftSidebar, Index, FormWrapMixin,
+                            Manage, AddCourse)
 
 class Department(grok.Container):
@@ -129 +130 @@
     def render(self):
         return u'<div class="portlet"><a href="manage">Settings</a></div>'
+
+class AddCourseForm(grok.AddForm):
+    grok.context(IDepartment)
+    form_fields = grok.AutoFields(ICourse)
+    label = 'Add a course'
+
+    @grok.action('Add course')
+    def addCourse(self, **data):
+        course = createObject(u'waeup.Course')
+        self.applyData(course, **data)
+        try:
+            self.context.courses.addCourse(course)
+        except DuplicationError:
+            self.status = Invalid('The name chosen already exists '
+                                  'in the database')
+            return
+        self.redirect(self.url(self.context))
+
+class AddCourse(FormWrapMixin, grok.Viewlet):
+    """A viewlet that wraps the `AddFacultyForm`.
+    """
+    grok.viewletmanager(MainArea)
+    grok.context(IDepartment)
+    grok.view(AddCourse)
+    grok.require('waeup.manageUniversity')
+
+    formview_name = 'addcourseform' # The name of the formview we
+                                    # want to be rendered in this
+                                    # viewlet.