py.lib.aw_web_tools

Yohn Y. 2024-11-09 Child:74f5377d83ab

9:9c734271dcf2 Go to Latest

py.lib.aw_web_tools/src/aw_web_tools/authelia_helper.py

.. 1.202411.2 + Поддержка Authelia

Download raw file
View source Diff to previous Annotate
History
awgur@9 1 # coding: utf-8
awgur@9 2 """\
awgur@9 3 Поддержка работы с системой авторизации [authelia](https://www.authelia.com/)
awgur@9 4 """
awgur@9 5 import bottle
awgur@9 6 from typing import Optional, List
awgur@9 7 from collections import namedtuple
awgur@9 8
awgur@9 9 from . import Error
awgur@9 10
awgur@9 11
awgur@9 12 # Константы
awgur@9 13 HEADER_USER = 'Remote-User' # Заголовок по умолчанию, в котором система аутентификации передаёт имя УЗ пользователя
awgur@9 14 HEADER_GROUPS = 'Remote-Groups' # Заголовок по умолчанию, в котором система аутентификации передаёт группы пользовтеля
awgur@9 15 HEADER_EMAIL = 'Remote-Email' # Заголовок по умолчанию, в котором система аутентификации передаёт e-mail пользователя
awgur@9 16 HEADER_NAME = 'Remote-Name' # Заголовок по умолчанию, в котором система аутентификации передаёт имя пользователя
awgur@9 17
awgur@9 18
awgur@9 19 class AHError(Error):
awgur@9 20 """\
awgur@9 21 Базовый класс исключений модуля адаптера Authelia
awgur@9 22 """
awgur@9 23
awgur@9 24
awgur@9 25 class AHAuthError(AHError):
awgur@9 26 """\
awgur@9 27 Проблемы с авторизацией пользователя
awgur@9 28 """
awgur@9 29 def __init__(self):
awgur@9 30 super().__init__('Пользователь не авторизован')
awgur@9 31
awgur@9 32
awgur@9 33 AHUser = namedtuple('AHUser', ['uname', 'groups', 'email', 'name'])
awgur@9 34
awgur@9 35
awgur@9 36 class AutheliaHelper(object):
awgur@9 37 def __init__(self,
awgur@9 38 group_filter: Optional[List[str]] = None,
awgur@9 39 header_user: str = HEADER_USER,
awgur@9 40 header_groups: str = HEADER_GROUPS,
awgur@9 41 header_email: str = HEADER_EMAIL,
awgur@9 42 header_name: str = HEADER_NAME
awgur@9 43 ):
awgur@9 44 """\
awgur@9 45 :param group_filter: Фильтр передаваемый системой аутентификации групп, которые интересны нам. Ели ``None``
awgur@9 46 группы, передаваемые в заголовке не фильтруются
awgur@9 47 :param header_user: Имя заголовка с именем УЗ пользователя.
awgur@9 48 :param header_groups: Имя заголовка с группами пользователя
awgur@9 49 :param header_name: Имя заголовка с именем пользователя
awgur@9 50 """
awgur@9 51 self.header_user = header_user
awgur@9 52 self.header_groups = header_groups
awgur@9 53 self.header_email = header_email
awgur@9 54 self.header_name = header_name
awgur@9 55
awgur@9 56 self.group_filter = None
awgur@9 57 self.group_filter_set = None
awgur@9 58 if group_filter is not None:
awgur@9 59 self.group_filter = {}
awgur@9 60 for i in group_filter:
awgur@9 61 self.group_filter[i.lower()] = i
awgur@9 62
awgur@9 63 self.group_filter_set = set(self.group_filter.keys())
awgur@9 64
awgur@9 65 def __call__(self, request: bottle.BaseRequest = bottle.request) -> AHUser:
awgur@9 66 """\
awgur@9 67 Обработка запроса. Если заголовки системы авторизации присутствуют, информацией из них заполняется
awgur@9 68 объект ``AHUser``. Если данные не найдены, то будет возбуждено исключение.
awgur@9 69
awgur@9 70 :param request: Обрабатываемый запрос
awgur@9 71 :returns: Объект ``AHUser`` с данным авторизации.
awgur@9 72 """
awgur@9 73
awgur@9 74 uname = request.get_header(self.header_user)
awgur@9 75 email = request.get_header(self.header_email)
awgur@9 76 name = request.get_header(self.header_name)
awgur@9 77 groups = request.get_header(self.header_groups)
awgur@9 78
awgur@9 79 if uname is None:
awgur@9 80 raise AHAuthError()
awgur@9 81
awgur@9 82 if groups is not None:
awgur@9 83 groups = tuple(map(lambda x: x.stip(), groups.split(',')))
awgur@9 84 if self.group_filter is not None:
awgur@9 85 _groups = []
awgur@9 86 _buf = set(map(lambda x: x.lower(), groups))
awgur@9 87
awgur@9 88 for grp in _buf & self.group_filter_set:
awgur@9 89 try:
awgur@9 90 _groups.append(self.group_filter[grp])
awgur@9 91
awgur@9 92 except KeyError:
awgur@9 93 AHError(f'Ошибка в фильтре групп. Для ключа "{grp}" не нашлось значения, хотя оно присутствует '
awgur@9 94 f'в множестве возможных значений.')
awgur@9 95
awgur@9 96 return AHUser(uname=uname, groups=groups, name=name, email=email)