90 lines
3.9 KiB
Python
90 lines
3.9 KiB
Python
from django.shortcuts import redirect,render
|
||
from django.http import HttpResponse
|
||
from keycloak import KeycloakOpenID, KeycloakGetError
|
||
from django.conf import settings
|
||
import logging
|
||
|
||
# Настройка логгера для текущего модуля
|
||
logger = logging.getLogger(__name__)
|
||
|
||
# Функция для начала процесса аутентификации с Keycloak
|
||
def keycloak_login(request):
|
||
try:
|
||
keycloak_openid = KeycloakOpenID(
|
||
server_url=settings.KEYCLOAK_CONFIG['SERVER_URL'],
|
||
client_id=settings.KEYCLOAK_CONFIG['CLIENT_ID'],
|
||
realm_name=settings.KEYCLOAK_CONFIG['REALM'],
|
||
client_secret_key=settings.KEYCLOAK_CONFIG['CLIENT_SECRET']
|
||
)
|
||
auth_url = keycloak_openid.auth_url(redirect_uri=settings.KEYCLOAK_CONFIG['CALLBACK_URL'])
|
||
logger.info("Redirecting to Keycloak for authentication.")
|
||
return redirect(auth_url)
|
||
except Exception as e:
|
||
logger.error(f"Error during Keycloak login: {e}")
|
||
return HttpResponse("Ошибка при попытке аутентификации через Keycloak.")
|
||
|
||
# Функция обратного вызова для обработки ответа от Keycloak
|
||
def keycloak_callback(request):
|
||
code = request.GET.get('code')
|
||
if code:
|
||
try:
|
||
keycloak_openid = KeycloakOpenID(
|
||
server_url=settings.KEYCLOAK_CONFIG['SERVER_URL'],
|
||
client_id=settings.KEYCLOAK_CONFIG['CLIENT_ID'],
|
||
realm_name=settings.KEYCLOAK_CONFIG['REALM'],
|
||
client_secret_key=settings.KEYCLOAK_CONFIG['CLIENT_SECRET']
|
||
)
|
||
|
||
token = keycloak_openid.token(
|
||
grant_type=['authorization_code'],
|
||
code=code,
|
||
redirect_uri=settings.KEYCLOAK_CONFIG['CALLBACK_URL']
|
||
)
|
||
|
||
userinfo = keycloak_openid.userinfo(token['access_token'])
|
||
logger.info(f"User authenticated with Keycloak: {userinfo}")
|
||
|
||
request.session['keycloak_user_id'] = userinfo['sub']
|
||
request.session['username'] = userinfo.get('preferred_username', 'Guest')
|
||
|
||
return redirect('http://127.0.0.1:8000/products/')
|
||
except KeycloakGetError as e:
|
||
logger.error(f"Keycloak authentication error: {e}")
|
||
return HttpResponse("Ошибка аутентификации.")
|
||
except Exception as e:
|
||
logger.error(f"Unexpected error during Keycloak callback: {e}")
|
||
return HttpResponse("Неожиданная ошибка.")
|
||
else:
|
||
logger.warning("Authentication code not provided.")
|
||
return HttpResponse("Код аутентификации не предоставлен.")
|
||
|
||
# Функция для выхода из системы
|
||
def keycloak_logout(request):
|
||
try:
|
||
request.session.flush()
|
||
logger.info("User logged out.")
|
||
return redirect('http://127.0.0.1:8000/')
|
||
except Exception as e:
|
||
logger.error(f"Error during logout: {e}")
|
||
return HttpResponse("Ошибка при выходе из системы.")
|
||
|
||
|
||
|
||
|
||
|
||
def profile(request):
|
||
# Проверяем, аутентифицирован ли пользователь
|
||
if 'keycloak_user_id' in request.session:
|
||
# Извлекаем данные пользователя из сессии
|
||
user_id = request.session['keycloak_user_id']
|
||
username = request.session.get('username', 'Гость')
|
||
logger.info(f"Keycloak user ID: {request.session['keycloak_user_id']}, Username: {request.session['username']},{username}")
|
||
|
||
|
||
|
||
# Отображаем шаблон, передавая в него данные пользователя
|
||
return render(request, 'main/product/profile.html', {'username': username})
|
||
else:
|
||
# Если пользователь не аутентифицирован, перенаправляем на страницу входа
|
||
return redirect('keycloak_login')
|