Changeset 2109 for trunk/loreality/loreality/controllers/messager.py

Timestamp:

07/12/10 14:48:12 (23 months ago)

Author:

dart

Message:

LoReality?:
o Messager is working

File:

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

trunk/loreality/loreality/controllers/messager.py

@@ -22 +22 @@
 
 class MessagerController(BaseController):
+    formfields = [
+            FormField(name="subject"       ,t_name=_("subject")       ,formtype="text"),
+            FormField(name="text"          ,t_name=_("text")          ,formtype="textarea"),
+            ]
 
-    def index(self):
+    #--------------------------------------------------------------
+    def _get_pilots(self):
+        pilot_q = meta.Session.query(model.Pilot)
+
+        me = pilot_q.filter_by(login=request.environ['REMOTE_USER']).first()
+        pilots = pilot_q.all()
+        names = []
+        for p in pilots:
+            if p is not me:
+                names.append( [str(p.id), p.name] )
+        return names
+
+    #--------------------------------------------------------------
+    def index(self, opened=None):
         pilot_q = meta.Session.query(model.Pilot)
         pilot = pilot_q.filter_by(login=request.environ['REMOTE_USER']).first()
@@ -30 +47 @@
 
         c.messages = pilot.messages 
+        c.message_open = opened
         return render('/derived/page/messager.html')
+    #--------------------------------------------------------------
+    def open(self, id):
+        return self.index(opened=uuid.UUID(id))
+    #--------------------------------------------------------------
+    def delete(self, id):
+        message_q = meta.Session.query(model.Message)
+        message = message_q.filter_by(id=uuid.UUID(id)).first()
+        if message is None:
+            abort(404)
+        if message.unread:
+            pilot = message.pilot
+            if pilot:
+                pilot.messages_unread -= 1
+            
+        meta.Session.delete( message )
+        meta.Session.commit()
+        return self.index()
+    #--------------------------------------------------------------
+    def create(self, id=None, values=None, errors=None ):
+        current_pilot = None
+        if id is None:
+            #- Make new message
+            pilot_q = meta.Session.query(model.Pilot)
+            pilot = pilot_q.filter_by(login=request.environ['REMOTE_USER']).first()
+            if pilot is None:
+                return " pilot" 
+                abort(404)
+
+            message = model.Message()
+            message.from_ = ""
+            message.subject = ""
+            message.text = ""
+            message.from_ = pilot.id
+            meta.Session.add( message )
+            meta.Session.commit()
+        else: 
+            message_q = meta.Session.query(model.Message)
+            message = message_q.filter_by(id=uuid.UUID(id)).first()
+            if message is None:
+                abort(404)
+            current_pilot = str(message.from_)
+        #- Here we must have a message
+        c.message = message
+
+        if not values: values = {}
+        values['subject'] = message.subject
+        values['Pilot'] = current_pilot
+
+        c.formfields = self.formfields
+        c.available_pilots = self._get_pilots()
+
+        html = render('/derived/page/messager_edit.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=='Send':
+            # Assume we are trying to save the form
+            schema = MessageForm()
+            try:
+                result = schema.to_python(dict(request.params), c)
+            except Invalid, e:
+                return self.create(
+                        id=id,
+                    values=values,
+                    errors=variabledecode.variable_encode(
+                        e.unpack_errors() or {},
+                        add_repetitions=False
+                    )
+                )
+            else:
+                message_q = meta.Session.query(model.Message)
+                message = message_q.filter_by(id=uuid.UUID(id)).first()
+                if message is None:
+                    abort(404)
+
+                for f in self.formfields:
+                    setattr(message, f.name, result[f.name] )
+
+                pilot_q = meta.Session.query(model.Pilot)
+                pilotids = request.params.getall('Pilot')
+                for pilotid in pilotids:
+                    pilot = pilot_q.filter_by(id=uuid.UUID(pilotid)).first()
+                    msg = message.copy()
+                    msg.from_ = pilot.id
+                    pilot.send_message( msg )
+                meta.Session.commit()
+                return self.index() 
+        else:
+            raise Exception('Invalid action %s'%action)