Changeset 2100 for trunk/loreality/loreality/controllers/campaign.py

Timestamp:

07/02/10 14:49:40 (23 months ago)

Author:

dart

Message:

LoReality?:
o Manage campaign, mission, and becin pilots

File:

• trunk/loreality/loreality/controllers/campaign.py (modified) (2 diffs)

trunk/loreality/loreality/controllers/campaign.py

@@ -8 +8 @@
 import loreality.model as model
 
+import loreality.lib.helpers as h
+from loreality.model.form import *
+
+from authkit.authorize.pylons_adaptors import authorize
+from pylons.decorators import validate
+
+from formencode import htmlfill, variabledecode
+from formencode.validators import Invalid
+import uuid
 
 log = logging.getLogger(__name__)
@@ -13 +22 @@
 class CampaignController(BaseController):
 
-    def index(self):
+    def index(self, values=None, errors=None ):
         c.title = "LoReality"
         c.heading = "Campaign"
-        c.campaign_title = "My Campaign"
         c.campaigns = meta.Session.query(model.Campaign).all()
 
-        return  render('/derived/page/campaign.html')
+        html = render('/derived/page/campaign.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=='Add campaign':
+            # Assume we are trying to save the form
+            schema = CampaignNameForm()
+            try:
+                result = schema.to_python(dict(request.params), c)
+            except Invalid, e:
+                return self.index(
+                    values=values,
+                    errors=variabledecode.variable_encode(
+                        e.unpack_errors() or {},
+                        add_repetitions=False
+                    )
+                )
+            else:
+                camp = model.Campaign( result['name_campaign'] )
+                meta.Session.add(camp)
+                meta.Session.commit()
+                return self.index() 
+        elif action=='Add mission':
+            # Assume we are trying to save the form
+            schema = MissionNameForm()
+            try:
+                result = schema.to_python(dict(request.params), c)
+            except Invalid, e:
+                return self.index(
+                    values=values,
+                    errors=variabledecode.variable_encode(
+                        e.unpack_errors() or {},
+                        add_repetitions=False
+                    )
+                )
+            else:
+                campaign_q = meta.Session.query(model.Campaign)
+                campaign = campaign_q.filter_by(id=uuid.UUID(id)).first()
+                if campaign is None:
+                    abort(404)
+
+                mis = model.Mission( result["name_mission"] )
+                meta.Session.add( mis )
+                campaign.missions.append( mis )
+
+                meta.Session.commit()
+                return self.edit(id)
+        elif action=='Modify':
+            # Assume we are trying to save the form
+            schema = CampaignForm()
+            try:
+                result = schema.to_python(dict(request.params), c)
+            except Invalid, e:
+                return self.index(
+                    values=values,
+                    errors=variabledecode.variable_encode(
+                        e.unpack_errors() or {},
+                        add_repetitions=False
+                    )
+                )
+            else:
+                campaign_q = meta.Session.query(model.Campaign)
+                campaign = campaign_q.filter_by(id=uuid.UUID(id)).first()
+                if campaign is None:
+                    abort(404)
+
+                campaign.name = result["name"]
+                meta.Session.commit()
+                return self.edit(id=id) 
+        else:
+            raise Exception('Invalid action %s'%action)
+
+    @authorize(h.auth.is_creator)
+    def edit(self, id=None, values=None, errors=None  ):
+        if id is None:
+            abort(404)
+        campaign_q = meta.Session.query(model.Campaign)
+        campaign = campaign_q.filter_by(id=uuid.UUID(id)).first()
+        if campaign is None:
+            abort(404)
+
+        c.title = "LoReality"
+        c.heading = "Edit %s"%campaign.name
+        c.campaign = campaign
+
+        if not values: values = {}
+        values["name"] = campaign.name
+
+        html = render('/derived/page/campaign_edit.html')
+        return  htmlfill.render(html, defaults=values, errors=errors)
+
+    @authorize(h.auth.is_creator)
+    def delete(self, id):
+        values = dict(request.params)
+        missionid = values["missionid"]
+        if id is None:
+            abort(404)
+        campaign_q = meta.Session.query(model.Campaign)
+        campaign = campaign_q.filter_by(id=uuid.UUID(id)).first()
+        if campaign is None:
+            abort(404)
+
+        if missionid is None:
+            abort(404)
+        mission_q = meta.Session.query(model.Mission)
+        mission = mission_q.filter_by(id=uuid.UUID(missionid)).first()
+        if mission is None:
+            abort(404)
+       
+        #- del now
+        campaign.missions.remove( mission )
+        meta.Session.delete( mission )
+        meta.Session.commit()
+        return self.edit( id )
+
+    @authorize(h.auth.is_creator)
+    def deletecampaign(self, id):
+        if id is None:
+            abort(404)
+        campaign_q = meta.Session.query(model.Campaign)
+        campaign = campaign_q.filter_by(id=uuid.UUID(id)).first()
+        if campaign is None:
+            abort(404)
+
+        #- del now
+        for mis in campaign.missions:
+            campaign.missions.remove( mis )
+            meta.Session.delete( mis )
+        meta.Session.delete( campaign )
+        meta.Session.commit()
+        return self.index()