27 lines
826 B
Python
27 lines
826 B
Python
|
from django.shortcuts import get_object_or_404
|
||
|
|
from hatchway import Schema, api_view
|
||
|
|
|
||
|
|
from activities.models import Post, PostInteraction
|
||
|
|
from api import schemas
|
||
|
|
from api.decorators import identity_required
|
||
|
|
|
||
|
|
|
||
|
|
class PostVoteSchema(Schema):
|
||
|
|
choices: list[int]
|
||
|
|
|
||
|
|
|
||
|
|
@identity_required
|
||
|
|
@api_view.get
|
||
|
|
def get_poll(request, id: str) -> schemas.Poll:
|
||
|
|
post = get_object_or_404(Post, pk=id, type=Post.Types.question)
|
||
|
|
return schemas.Poll.from_post(post, identity=request.identity)
|
||
|
|
|
||
|
|
|
||
|
|
@identity_required
|
||
|
|
@api_view.post
|
||
|
|
def vote_poll(request, id: str, details: PostVoteSchema) -> schemas.Poll:
|
||
|
|
post = get_object_or_404(Post, pk=id, type=Post.Types.question)
|
||
|
|
PostInteraction.create_votes(post, request.identity, details.choices)
|
||
|
|
post.refresh_from_db()
|
||
|
|
return schemas.Poll.from_post(post, identity=request.identity)
|