Add support for sending new messages.

tailbone/views/messages.py

@@ -30,12 +30,13 @@ from rattail import enum
 from rattail.db import model
 
 import formalchemy
+from formalchemy.helpers import hidden_field
 from pyramid import httpexceptions
-from webhelpers.html import tags
+from webhelpers.html import tags, HTML
 
 from tailbone import forms
 from tailbone.db import Session
-from tailbone.views import MasterView
+from tailbone.views import MasterView, AutocompleteView
 
 
 class SubjectFieldRenderer(formalchemy.FieldRenderer):
@@ -56,13 +57,31 @@ class SenderFieldRenderer(forms.renderers.UserFieldRenderer):
         return super(SenderFieldRenderer, self).render_readonly(**kwargs)
 
 
+class RecipientsField(formalchemy.Field):
+    """
+    Custom field for recipients, used when sending new messages.
+    """
+    is_collection = True
+
+    def sync(self):
+        if not self.is_readonly():
+            message = self.parent.model
+            for uuid in set(self._deserialize()):
+                user = Session.query(model.User).get(uuid)
+                if user:
+                    message.add_recipient(user, status=enum.MESSAGE_STATUS_INBOX)
+
+
 class RecipientsFieldRenderer(formalchemy.FieldRenderer):
 
+    def render(self, **kwargs):
+        return HTML.tag('ul')
+
     def render_readonly(self, **kwargs):
         recipients = self.raw_value
         if not recipients:
             return ''
-        recips = filter(lambda r: r.recipient is not self.request.user, recipients)
+        recips = [r for r in recipients if r.recipient is not self.request.user]
         recips = sorted([r.recipient.display_name for r in recips])
         if len(recips) < len(recipients):
             recips.insert(0, 'you')
@@ -87,7 +106,6 @@ class MessagesView(MasterView):
     Base class for message views.
     """
     model_class = model.Message
-    creatable = False
     editable = False
     deletable = False
     checkboxes = True
@@ -151,16 +169,27 @@ class MessagesView(MasterView):
         return {}
 
     def configure_fieldset(self, fs):
-        fs.configure(
-            include=[
+        if self.creating:
+            fs.append(RecipientsField('recipients', label="To", renderer=RecipientsFieldRenderer))
+            fs.configure(include=[
+                fs.recipients,
                 fs.subject,
+                fs.body.textarea(size='50x10'),
+            ])
+        else:
+            fs.configure(include=[
                 fs.sender.with_renderer(SenderFieldRenderer).label("From"),
                 fs.recipients.with_renderer(RecipientsFieldRenderer).label("To"),
                 fs.sent.with_renderer(forms.renderers.DateTimeFieldRenderer(self.rattail_config)),
-                fs.body,
+                fs.subject,
+                fs.body.textarea(size='50x10'),
             ])
         if self.viewing:
-            del fs.body
+            del fs.body         # ..really?
+
+    def before_create(self, form):
+        message = form.fieldset.model
+        message.sender = self.request.user
 
     def get_recipient(self, message):
         for recip in message.recipients:
@@ -265,6 +294,22 @@ class SentView(MessagesView):
         g.filters['sender'].default_active = False
 
 
+class RecipientsAutocomplete(AutocompleteView):
+    """
+    Autocomplete AJAX view for the recipients field when sending a new message.
+    """
+
+    def query(self, term):
+        return Session.query(model.User)\
+                      .join(model.Person)\
+                      .filter(model.User.active == True)\
+                      .filter(model.Person.display_name.ilike('%{}%'.format(term)))\
+                      .order_by(model.Person.display_name)
+
+    def display(self, user):
+        return user.person.display_name
+
+
 def includeme(config):
 
     config.add_tailbone_permission('messages', 'messages.list', "List/Search Messages")
@@ -279,6 +324,10 @@ def includeme(config):
     config.add_view(ArchiveView, attr='index', route_name='messages.archive',
                     permission='messages.list')
 
+    # move (single)
+    config.add_route('messages.move', '/messages/{uuid}/move')
+    config.add_view(MessagesView, attr='move', route_name='messages.move')
+
     # move bulk
     config.add_route('messages.move_bulk', '/messages/move-bulk')
     config.add_view(MessagesView, attr='move_bulk', route_name='messages.move_bulk')
@@ -288,10 +337,9 @@ def includeme(config):
     config.add_view(SentView, attr='index', route_name='messages.sent',
                     permission='messages.list')
 
-    # view
-    config.add_route('messages.view', '/messages/{uuid}')
-    config.add_view(MessagesView, attr='view', route_name='messages.view')
+    # recipients autocomplete
+    config.add_route('messages.recipients',  '/messages/recipients')
+    config.add_view(RecipientsAutocomplete, route_name='messages.recipients',
+                    renderer='json', permission='messages.create')
 
-    # move (single)
-    config.add_route('messages.move', '/messages/{uuid}/move')
-    config.add_view(MessagesView, attr='move', route_name='messages.move')
+    MessagesView.defaults(config)