etesync-server/etebase_fastapi/utils.py

48 lines
1.2 KiB
Python
Raw Normal View History

2020-12-27 16:54:06 +00:00
import dataclasses
import typing as t
from fastapi import status, Query
from pydantic import BaseModel as PyBaseModel
from django.db.models import QuerySet
from django.core.exceptions import ObjectDoesNotExist
2020-12-27 16:54:06 +00:00
from django.contrib.auth import get_user_model
from django_etebase.models import AccessLevels
2020-12-28 08:00:35 +00:00
from .exceptions import HttpError, HttpErrorOut
2020-12-27 16:54:06 +00:00
User = get_user_model()
Prefetch = t.Literal["auto", "medium"]
PrefetchQuery = Query(default="auto")
class BaseModel(PyBaseModel):
class Config:
json_encoders = {
bytes: lambda x: x,
}
2020-12-27 16:54:06 +00:00
@dataclasses.dataclass
class Context:
user: t.Optional[User]
prefetch: t.Optional[Prefetch]
def get_object_or_404(queryset: QuerySet, **kwargs):
try:
return queryset.get(**kwargs)
except ObjectDoesNotExist as e:
2020-12-28 07:51:27 +00:00
raise HttpError("does_not_exist", str(e), status_code=status.HTTP_404_NOT_FOUND)
def is_collection_admin(collection, user):
member = collection.members.filter(user=user).first()
return (member is not None) and (member.accessLevel == AccessLevels.ADMIN)
2020-12-28 08:00:35 +00:00
response_model_dict = {"model": HttpErrorOut}
permission_responses = {403: response_model_dict}