From e6a6e2c6aff84b5152584ace80740c9a87fac448 Mon Sep 17 00:00:00 2001
From: Pierre-Yves Chibon <pingou@pingoured.fr>
Date: Mar 31 2014 09:57:04 +0000
Subject: After having successfully logged-in on FAS, save/update user information locally


Signed-off-by: Pierre-Yves Chibon <pingou@pingoured.fr>

---

diff --git a/progit/__init__.py b/progit/__init__.py
index 7a84c4e..b5cc35c 100644
--- a/progit/__init__.py
+++ b/progit/__init__.py
@@ -169,10 +169,27 @@ def rst2html(rst_string):
         return progit.doc_utils.convert_doc(unicode(rst_string))
 
 
+@FAS.postlogin
+def set_user(return_url):
+    ''' After login method. '''
+    try:
+        progit.lib.set_up_user(
+            session=SESSION,
+            username=flask.g.fas_user.username,
+            fullname=flask.g.fas_user.fullname,
+            user_email=flask.g.fas_user.email,
+        )
+        SESSION.commit()
+    except SQLAlchemyError, err:
+        flask.flash(
+            'Could not set up you as a user properly, please contact '
+            'an admin', 'error')
+    return flask.redirect(return_url)
+
+
 @APP.route('/login/', methods=('GET', 'POST'))
 def auth_login():
     """ Method to log into the application using FAS OpenID. """
-
     return_point = flask.url_for('index')
     if 'next' in flask.request.args:
         return_point = flask.request.args['next']