Refactor Flask SSO app to use Sessions (#6)

* Update Flask apps with caveat regarding port use with Mac OS Monterey and removed buggy user image logic from logged in page on SSO app

* Refactor Flask SSO app to use sessions and clean up unused static files on dsync app

* Update readme to reflect need for secret key env variable

Co-authored-by: Adam Wolfman <[email protected]>
Co-authored-by: Adam Wolfman <[email protected]>

Author: 3 people

python-flask-directory-sync-example/app.py

@@ -52,6 +52,17 @@ def directory_groups():
 
     return render_template('groups.html', groups=groups)
 
+@app.route('/update_org', methods=['GET', 'POST'])
+def update_org():
+    print(request)
+    response = workos_client.organizations.update_organization(
+        organization="org_01FGM2T96YX19Z4HENZ1AC7848",
+        name="Planet Express",
+        domains=["newboom.com", "workos.com", "hotmail.com", "boom.com"]
+    )
+    print(json.dumps(response))
+    return json.dumps(response)
+
 @app.route('/webhooks', methods=['GET', 'POST'])
 def webhooks():
     print(request)

python-flask-directory-sync-example/static/images/mountain_image.jpeg

@@ -54,8 +54,9 @@ An example Flask application demonstrating how to use the [WorkOS Python SDK](ht
 
 8. Once the Nano text editor opens, you can directly edit the `.env` file by listing the environment variables:
    ```bash
-   export WORKOS_API_KEY=<value found in step 6>
-   export WORKOS_CLIENT_ID=<value found in step 6>
+   WORKOS_API_KEY=<value found in step 6>
+   WORKOS_CLIENT_ID=<value found in step 6>
+   APP_SECRET_KEY=<any string value you\'d like>
    ```
 
    To exit the Nano text editor, type `CTRL + x`. When prompted to "Save modified buffer", type `Y`, then press the `Enter` or `Return` key.

python-flask-directory-sync-example/static/images/workos_logo.png

@@ -1,12 +1,13 @@
 import os
 
-from flask import (Flask, redirect, render_template, request, url_for)
+from flask import (Flask, session, redirect, render_template, request, url_for)
 import workos
 
 
 # Flask Setup
 DEBUG = False
 app = Flask(__name__)
+app.secret_key = os.getenv('APP_SECRET_KEY')
 
 # WorkOS Setup
 
@@ -15,11 +16,16 @@
 workos.base_api_url = 'http://localhost:7000/' if DEBUG else workos.base_api_url
 
 # Enter Connection ID here
+
 CUSTOMER_CONNECTION_ID = 'xxx'
 
+
 @app.route('/')
 def login():
-    return render_template('login.html')
+    if session:
+        return render_template('login_successful.html', first_name=session['first_name'], raw_profile=session['raw_profile'])
+    else:
+        return render_template('login.html')
 
 @app.route('/auth')
 def auth():
@@ -36,11 +42,18 @@ def auth():
 @app.route('/auth/callback')
 def auth_callback():
     code = request.args.get('code')
-    print(code)
     profile = workos.client.sso.get_profile_and_token(code)
     p_profile = profile.to_dict()
-    first_name = p_profile['profile']['first_name']
-
-    raw_profile = p_profile['profile']
+    session['first_name'] = p_profile['profile']['first_name']
+    session['raw_profile'] = p_profile['profile']
+    session['session_id'] = p_profile['profile']['id']
+    print('session', session)
+    return redirect('/')
+    
 
-    return render_template('login_successful.html', first_name=first_name, raw_profile=raw_profile)
+@app.route('/logout')
+def logout():
+    # remove the username from the session if it is there
+    session.clear()
+    print(session)
+    return redirect('/')

python-flask-sso-example/README.md

@@ -31,6 +31,7 @@ <h2>Raw Profile Response Details</h2>
                     <div class="text_box">
                         <p>{{raw_profile}}</p>
                     </div>
+                    <a href="/logout"><button class="button">Log Out</button></a>
                 </div>
 
             </div>