-
Notifications
You must be signed in to change notification settings - Fork 0
/
app.py
91 lines (72 loc) · 2.51 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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
# coding: utf-8
from flask import Flask, request, redirect, Response
from flask_login import LoginManager, UserMixin, login_user, logout_user, login_required, current_user
from flask_firebaseauth import FirebaseAuth
app = Flask(__name__)
app.config['SECRET_KEY'] = "secret"
app.config['FIREBASE_API_KEY'] = '<The API key>'
app.config['FIREBASE_PROJECT_ID'] = '<The project identifier, eg. `foobar`>'
app.config['FIREBASE_AUTH_SIGN_IN_OPTIONS'] = 'email,facebook,github,google,twitter' #Comma-separated list of enabled providers.
login_manager = LoginManager()
login_manager.init_app(app)
auth = FirebaseAuth(app)
app.register_blueprint(auth.blueprint, url_prefix='/auth')
class User(UserMixin):
# user database
users = {
'[email protected]': {'name': 'debug01'},
'[email protected]': {'name': 'debug02'},
'[email protected]': {'name': 'debug03'},
'[email protected]': {'name': 'debug04'},
}
def __init__(self, id, name):
self.id = id
self.name = name
@classmethod
def get(self, id):
user = self.users.get(id)
if user is not None:
return User(id, user['name'])
@auth.production_loader
def production_loader(token):
user = User.get(token['email'])
if user is not None:
login_user(user, True)
@auth.development_loader
def development_loader(email):
user = User.get(email)
if user is not None:
login_user(user, True)
@auth.unloader
def sign_out():
logout_user()
@login_manager.user_loader
def user_loader(user_id):
return User.get(user_id)
@login_manager.request_loader
def request_loader(request):
token = request.headers.get('Authorization')
if token is None:
token = request.args.get('token')
return User.get(token)
@login_manager.unauthorized_handler
def authentication_required():
auth.language_code = 'ja' # https://github.com/firebase/firebaseui-web/blob/master/LANGUAGES.md
return redirect(auth.url_for('widget', mode='select', next=request.url))
@app.route('/')
def homepage():
return Response("home: <a href='/login/'>Login</a> <a href='/protected/'>Protected</a> <a href='/logout/'>Logout</a>")
@app.route('/login/')
@login_required
def login():
return redirect("/")
@app.route('/logout/')
def logout():
return auth.sign_out()
@app.route('/protected/')
@login_required
def protected():
return Response("<h1>Protected Page</h1><a href='/logout/'>Logout</a><br />" + current_user.name)
if __name__ == '__main__':
auth.debug = True
app.run(debug=True)