Remove domain from SSO app (#9)

awolfden · Adam Wolfman · web-flow · commit d46a22c5c760 · 2021-12-15T11:27:08.000-07:00
Co-authored-by: Adam Wolfman &lt;adamwolfman@Adams-MBP-3.domain&gt;
diff --git a/python-django-sso-example/README.md b/python-django-sso-example/README.md
@@ -90,11 +90,11 @@ An example Django application demonstrating how to use the [WorkOS Python SDK](h
    . . .
    ```
 
-11. In `python-django-sso-example/sso/views.py` change the `CUSTOMER_EMAIL_DOMAIN` string value to an email domain that makes sense for your testing purposes if the default `gmail.com` isn't relevant.
+11. In `python-django-sso-example/sso/views.py` change the `CONNECTION_ID` string value to the connection ID that you are targeting. This can be found in the WorkOS Dashboard under the Connection Settings. 
 
 12. The final setup step is to start the server.
    ```bash
-   (env) $ python3 manage.py runserver
+   (env) $ python3 manage.py runserver --insecure
    ```
 
    You'll know the server is running when you see no warnings or errors in the CLI, and output similar to the following is displayed:
@@ -125,7 +125,7 @@ If you get stuck, please reach out to us at support@workos.com so we can help.
 
 ## Testing the Integration
 
-1. Naviagte to the `python-django-sso-example` directory, which contains the `manage.py` file. Source the virtual environment we created earlier, if it isn't still activated from the steps above. Start the Django server locally.
+13. Naviagte to the `python-django-sso-example` directory, which contains the `manage.py` file. Source the virtual environment we created earlier, if it isn't still activated from the steps above. Start the Django server locally.
 
    ```bash
    $ cd ~/Desktop/python-django-sso-example/
diff --git a/python-django-sso-example/requirements.txt b/python-django-sso-example/requirements.txt
@@ -8,3 +8,4 @@ requests==2.25.1
 sqlparse==0.4.2
 urllib3==1.26.5
 workos==1.2.0
+python-dotenv
diff --git a/python-django-sso-example/sso/templates/sso/login_successful.html b/python-django-sso-example/sso/templates/sso/login_successful.html
@@ -6,7 +6,6 @@
     <body class="container_success">
         <div class="logged_in_nav">
             <div class="flex">
-                <img src={{image}} alt="Profile Image">
                 <p>You're logged in {{first_name}}, welcome!</p>
             </div>
             <div>
diff --git a/python-django-sso-example/sso/views.py b/python-django-sso-example/sso/views.py
@@ -15,8 +15,8 @@
 # Constants
 # Required: Fill in either domain or customer_ID or both, at least one must be populated to generate auth connection.
 # For testing purposes we fitted domain with gmail.com as an example, please edit and add domains as needed for your testing.
-CUSTOMER_EMAIL_DOMAIN = 'gmail.com'
-CONNECTION_ID = ''
+
+CONNECTION_ID = 'xxx'
 REDIRECT_URI = os.getenv('REDIRECT_URI')
 
 
@@ -27,10 +27,9 @@ def login(request):
 def auth(request):
 
     authorization_url = workos.client.sso.get_authorization_url(
-        domain = CUSTOMER_EMAIL_DOMAIN,
+        connection= CONNECTION_ID,
         redirect_uri= REDIRECT_URI,
-        state={},
-        connection= CONNECTION_ID
+        state={},        
     )
     return redirect(authorization_url)
 
@@ -42,16 +41,10 @@ def auth_callback(request):
     print(p_profile)
     first_name = p_profile['profile']['first_name']
 
-    if "picture" in p_profile['profile']['raw_attributes']:
-        image = p_profile['profile']['raw_attributes']['picture']
-    else: 
-        image = "../static/images/workos_logo.png"
-
     raw_profile = p_profile['profile']
 
     return render(request, 'sso/login_successful.html', {
         "p_profile": p_profile,
         "first_name": first_name,
-        "image": image,
         "raw_profile": raw_profile
     })
