-
Notifications
You must be signed in to change notification settings - Fork 9
/
admin.py
48 lines (37 loc) · 1.34 KB
/
admin.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
"""
The core application: Admin interface configuration.
"""
import logging
import requests
from django.conf import settings
from django.contrib import admin
from . import models
logger = logging.getLogger(__name__)
GOOGLE_USER_INFO_URL = "https://www.googleapis.com/oauth2/v3/userinfo"
for model in [
models.EligibilityType,
models.EligibilityVerifier,
models.PaymentProcessor,
models.PemData,
models.TransitAgency,
]:
logger.debug(f"Register {model.__name__}")
admin.site.register(model)
def pre_login_user(user, request):
logger.debug(f"Running pre-login callback for user: {user.username}")
token = request.session.get("google_sso_access_token")
if token:
headers = {
"Authorization": f"Bearer {token}",
}
# Request Google user info to get name and email
response = requests.get(GOOGLE_USER_INFO_URL, headers=headers, timeout=settings.REQUESTS_TIMEOUT)
user_data = response.json()
logger.debug(f"Updating user data from Google for user with email: {user_data['email']}")
user.first_name = user_data["given_name"]
user.last_name = user_data["family_name"]
user.username = user_data["email"]
user.email = user_data["email"]
user.save()
else:
logger.warning("google_sso_access_token not found in session.")