Changeset 2106 for trunk/loreality/loreality/controllers/squadron.py

Timestamp:

07/06/10 12:38:59 (23 months ago)

Author:

dart

Message:

LoReality?:
o Update on squadron/mission

File:

• trunk/loreality/loreality/controllers/squadron.py (modified) (5 diffs)

trunk/loreality/loreality/controllers/squadron.py

@@ -12 +12 @@
 
 import loreality.lib.helpers as h
-from loreality.model.form import CampaignForm, MissionForm
+from loreality.model.form import *
 
 from authkit.authorize.pylons_adaptors import authorize
@@ -22 +22 @@
 import uuid
 
+class SquField:
+    name = ""
+    formtype = ""
+    def __init__( self, **kwargs):
+        self.__dict__.update( kwargs )
+
+
 class SquadronController(BaseController):
+    squfields = [
+            SquField(name="name"          ,t_name=_("name")          ,formtype="text"),
+            SquField(name="plane"         ,t_name=_("plane")   ,formtype="text"),
+            ]
     #--------------------------------------------------------------
     def _get_pilots(self, already_pilots=[]):
@@ -38 +49 @@
         c.heading = _("Squadrons")
         c.squadrons = squadrons
+        c.squfields = self.squfields
         c.available_pilots = self._get_pilots()
 
@@ -54 +66 @@
         c.heading = squadron.name
         c.squadron = squadron
+        c.squfields = self.squfields
         c.available_pilots = self._get_pilots(squadron.pilots)
 
         html = render('/derived/page/squadron.html')
         return  htmlfill.render(html, defaults=values, errors=errors)
+    #--------------------------------------------------------------
+    def process(self, id=None):
+        action = request.params.getone('action')
+        values = dict(request.params)
+        # Don't use the values field for repopulation
+        del values['action']
+        if action=='Modify':
+            # Assume we are trying to save the form
+            schema = SquadronForm()
+            try:
+                result = schema.to_python(dict(request.params), c)
+            except Invalid, e:
+                return self.edit(
+                        id=id,
+                    values=values,
+                    errors=variabledecode.variable_encode(
+                        e.unpack_errors() or {},
+                        add_repetitions=False
+                    )
+                )
+            else:
+                squadron_q = meta.Session.query(model.Squadron)
+                squadron = squadron_q.filter_by(id=uuid.UUID(id)).first()
+                if squadron is None:
+                    abort(404)
+
+                for f in self.squfields:
+                    setattr(squadron, f.name, result[f.name] )
+
+                meta.Session.commit()
+                return self.view(id=id) 
+        else:
+            raise Exception('Invalid action %s'%action)
 
     #--------------------------------------------------------------
@@ -72 +118 @@
         c.heading = "Edit %s"%squadron.name
         c.squadron = squadron
+        c.squfields = self.squfields
         c.available_pilots = self._get_pilots(squadron.pilots)
+
+        if not values: values = {}
+
+        for f in self.squfields:
+            values[f.name] = getattr(squadron, f.name )
 
         html = render('/derived/page/squadron_edit.html')