etesync-server/django_etebase/utils.py

37 lines
1008 B
Python
Raw Normal View History

import typing as t
from dataclasses import dataclass
from django.core.exceptions import PermissionDenied
2020-12-29 11:22:36 +00:00
from myauth.models import UserType, get_typed_user_model
2020-07-13 11:30:18 +00:00
from . import app_settings
2020-12-29 11:22:36 +00:00
User = get_typed_user_model()
2020-07-13 11:30:18 +00:00
@dataclass
class CallbackContext:
"""Class for passing extra context to callbacks"""
url_kwargs: t.Dict[str, t.Any]
2020-12-29 11:22:36 +00:00
user: t.Optional[UserType] = None
def get_user_queryset(queryset, context: CallbackContext):
custom_func = app_settings.GET_USER_QUERYSET_FUNC
2020-07-13 11:30:18 +00:00
if custom_func is not None:
return custom_func(queryset, context)
2020-07-13 11:30:18 +00:00
return queryset
def create_user(context: CallbackContext, *args, **kwargs):
custom_func = app_settings.CREATE_USER_FUNC
if custom_func is not None:
2020-12-28 15:46:20 +00:00
return custom_func(context, *args, **kwargs)
return User.objects.create_user(*args, **kwargs)
def create_user_blocked(*args, **kwargs):
raise PermissionDenied("Signup is disabled for this server. Please refer to the README for more information.")