Change set_context function, using **kwargs

This commit is contained in:
Степаненко Ольга 2021-05-06 15:23:22 +03:00
parent b9cb22e2e5
commit bb74072c32

View File

@ -35,8 +35,7 @@ from main.serializers import ProfileSerializer, ZendeskUserSerializer
from .models import UserProfile from .models import UserProfile
def setup_context(profile_lit: bool = False, control_lit: bool = False, work_lit: bool = False, def setup_context(**kwargs) -> Dict[str, Any]:
registration_lit: bool = False, login_lit: bool = False, stats_lit: bool = False) -> Dict[str, Any]:
""" """
Функция добавления в контекст статуса пользователя. Функция добавления в контекст статуса пользователя.
@ -48,15 +47,12 @@ def setup_context(profile_lit: bool = False, control_lit: bool = False, work_lit
:param stats_lit: True, при получении пользователем прав администратора (просмотр статистики), иначе False :param stats_lit: True, при получении пользователем прав администратора (просмотр статистики), иначе False
:return: Контекст (context) :return: Контекст (context)
""" """
context = {}
context = { for key in ('profile_lit', 'control_lit', 'work_lit', 'registration_lit', 'login_lit', 'stats_lit'):
'profile_lit': profile_lit, if key in kwargs:
'control_lit': control_lit, context.update({key: True})
'work_lit': work_lit, else:
'registration_lit': registration_lit, context.update({key: False})
'login_lit': login_lit,
'stats_lit': stats_lit,
}
return context return context