From e313608090b8f7e1e1de082f4f2c31c670217903 Mon Sep 17 00:00:00 2001
From: Doug Le Tough <doug.letough@free.fr>
Date: Wed, 9 May 2018 10:26:17 +0200
Subject: [PATCH] "list staffers pdf"

---
 participate.py | 35 +++++++++++++++++++++++++++++++----
 1 file changed, 31 insertions(+), 4 deletions(-)

diff --git a/participate.py b/participate.py
index a5be8a8..7c23864 100755
--- a/participate.py
+++ b/participate.py
@@ -1040,11 +1040,38 @@ def update_staff_slot(TURN_ID, SLOT_ID):
 @check_session
 def list_users():
   """ Users list """
-  page = str(request.url_rule)
-  menu = get_menu(page)
   message = check_user_info()
-  staffers = Tetawebapp_users.query.filter(Tetawebapp_users.is_admin==0, Tetawebapp_users.link_id==None).order_by(Tetawebapp_users.name).all()
-  return render_template('list_users.html', menu=menu, staffers=staffers, message=message)
+  try:
+    if session['is_admin']:
+      page = str(request.url_rule)
+      menu = get_menu(page)
+      staffers = Tetawebapp_users.query.filter(Tetawebapp_users.is_admin==0, Tetawebapp_users.link_id==None).order_by(Tetawebapp_users.name).all()
+      return render_template('list_users.html', menu=menu, staffers=staffers, message=message)
+  # User is not admin
+    return render_template('login_or_register.html', message="Utilisateur ou mot de passe invalide")
+  except:
+    # User is not logged in
+    return render_template('login_or_register.html', message="Utilisateur ou mot de passe invalide")
+
+@app.route("/users/pdf", methods=['GET', 'POST'])
+@check_session
+def print_users():
+  """ Print user list """
+  message = check_user_info()
+  try:
+    if session['is_admin']:
+      page = str(request.url_rule)
+      menu = get_menu(page)
+      staffs = Tetawebapp_staffs.query(Tetawebapp_staffs.user_id).distinct()
+      staffers = Tetawebapp_users.query.filter(Tetawebapp_users.is_admin==0, Tetawebapp_users.link_id==None).order_by(Tetawebapp_users.name).all()
+      sstaffers = [staffer for staffer in staffers if staffer.id in staffs]
+      return sstaffers
+  # User is not admin
+    return render_template('login_or_register.html', message="Utilisateur ou mot de passe invalide")
+  except:
+    # User is not logged in
+    return render_template('login_or_register.html', message="Utilisateur ou mot de passe invalide")
+
 
 @app.route("/account/<ID>", methods=['GET', 'POST'])
 @check_session