-
Notifications
You must be signed in to change notification settings - Fork 0
/
app.py
75 lines (65 loc) · 2.25 KB
/
app.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
from flask import Flask, render_template, url_for, redirect, session
from authlib.integrations.flask_client import OAuth
from authlib.common.security import generate_token
from dotenv import load_dotenv
import os
from db_functions import update_or_create_user
from flask_session import Session
load_dotenv()
GOOGLE_CLIENT_ID = os.getenv('GOOGLE_CLIENT_ID')
GOOGLE_CLIENT_SECRET = os.getenv('GOOGLE_CLIENT_SECRET')
app = Flask(__name__)
app.secret_key = os.urandom(12)
app.config["SESSION_PERMANENT"] = False
app.config["SESSION_TYPE"] = "filesystem"
Session(app)
oauth = OAuth(app)
@app.route('/')
def index():
return render_template('index.html')
@app.route('/google/')
def google():
CONF_URL = 'https://accounts.google.com/.well-known/openid-configuration'
oauth.register(
name='google',
client_id=GOOGLE_CLIENT_ID,
client_secret=GOOGLE_CLIENT_SECRET,
server_metadata_url=CONF_URL,
client_kwargs={
'scope': 'openid email profile'
}
)
# Redirect to google_auth function
###note, if running locally on a non-google shell, do not need to override redirect_uri
### and can just use url_for as below
redirect_uri = url_for('google_auth', _external=True)
print('REDIRECT URL: ', redirect_uri)
session['nonce'] = generate_token()
##, note: if running in google shell, need to override redirect_uri
## to the external web address of the shell, e.g.,
redirect_uri = 'https://5000-cs-1039191608401-default.cs-us-east1-pkhd.cloudshell.dev/google/auth/'
return oauth.google.authorize_redirect(redirect_uri, nonce=session['nonce'])
@app.route('/google/auth/')
def google_auth():
token = oauth.google.authorize_access_token()
user = oauth.google.parse_id_token(token, nonce=session['nonce'])
session['user'] = user
update_or_create_user(user)
print(" Google User ", user)
return redirect('/dashboard')
@app.route('/dashboard/')
def dashboard():
user = session.get('user')
if user:
return render_template('dashboard.html', user=user)
else:
return redirect('/')
@app.route('/logout')
def logout():
session.pop('user', None)
return redirect('/')
if __name__ == '__main__':
app.run(
debug=True,
port=5000
)