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