From b85a481677a0307cc51324961b33954f787cb2ba Mon Sep 17 00:00:00 2001
From: yomguy <yomguy@parisson.com>
Date: Thu, 12 Apr 2012 10:54:47 +0200
Subject: [PATCH] add messages to training users

---
 teleforma/templates/telemeta/users.html  | 8 ++++++++
 teleforma/templatetags/teleforma_tags.py | 7 +++++++
 teleforma/views.py                       | 6 ++++--
 3 files changed, 19 insertions(+), 2 deletions(-)

diff --git a/teleforma/templates/telemeta/users.html b/teleforma/templates/telemeta/users.html
index a5382100..cbea2e08 100644
--- a/teleforma/templates/telemeta/users.html
+++ b/teleforma/templates/telemeta/users.html
@@ -1,6 +1,7 @@
 {% extends "telemeta/base.html" %}
 {% load i18n %}
 {% load telemeta_utils %}
+{% load teleforma_tags %}
 
 {% block head_title %}{% trans "Users" %} - {{ block.super }}{% endblock %}
 
@@ -22,6 +23,13 @@
 </ul>
 </div>
 </div>
+
+{% if users %}
+<div class="module_action">
+<a href="{% url postman_write all_users|to_recipients %}" class="component_icon button" id="action_green">{% trans "New message" %}</a>
+</div>
+{% endif %}
+
 </div>
 
 <div class="desk_course">
diff --git a/teleforma/templatetags/teleforma_tags.py b/teleforma/templatetags/teleforma_tags.py
index 322d8c7c..56e31aa4 100644
--- a/teleforma/templatetags/teleforma_tags.py
+++ b/teleforma/templatetags/teleforma_tags.py
@@ -49,3 +49,10 @@ class ValueFromSettings(template.Node):
 def trainings(user):
     student = user.student.get()
     return training.student.all()
+
+@register.filter
+def to_recipients(users):
+    list = []
+    for user in users:
+        list.append(user.username)
+    return ':'.join(list)
diff --git a/teleforma/views.py b/teleforma/views.py
index cc7c00fa..c2efa52d 100755
--- a/teleforma/views.py
+++ b/teleforma/views.py
@@ -131,18 +131,20 @@ class UsersView(ListView):
     def get_context_data(self, **kwargs):
         context = super(UsersView, self).get_context_data(**kwargs)
         context['trainings'] = Training.objects.all()
+        context['all_users'] = User.objects.all()
         return context
 
 class UsersTrainingView(UsersView):
 
     def get_queryset(self):
         users = User.objects.all().select_related(depth=2)
-        trainings = Training.objects.filter(id=self.args[0])
-        return User.objects.filter(student__training__in=trainings)
+        self.trainings = Training.objects.filter(id=self.args[0])
+        return User.objects.filter(student__training__in=self.trainings)
 
     def get_context_data(self, **kwargs):
         context = super(UsersTrainingView, self).get_context_data(**kwargs)
         context['training'] = Training.objects.get(id=self.args[0])
+        context['all_users'] = User.objects.filter(student__training__in=self.trainings).all()
         return context
 
 class UsersXLSExport(object):
-- 
2.39.5