41 lines
1.2 KiB
Python
41 lines
1.2 KiB
Python
from django.db import transaction
|
|
from rest_framework import serializers
|
|
|
|
from core.apps.accounts.serializers import UserSerializer
|
|
from core.apps.tasks.models.comment import Comment
|
|
|
|
|
|
class CommentSerializer(serializers.ModelSerializer):
|
|
created_by = UserSerializer(read_only=True)
|
|
|
|
class Meta:
|
|
model = Comment
|
|
fields = [
|
|
'id', 'message', 'file', 'type', 'created_at', 'created_by'
|
|
]
|
|
|
|
def get_created_by(self, obj):
|
|
request = self.context.get('request')
|
|
return {
|
|
"id": obj.created_by.id,
|
|
"first_name": obj.created_by.first_name,
|
|
"last_name": obj.created_by.last_name,
|
|
"avatar": request.build_absolute_uri(obj.created_by.avatar.url) if obj.created_by.avatar else None
|
|
}
|
|
|
|
|
|
class CommentCreateSerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = Comment
|
|
fields = [
|
|
'id', 'message', 'file', 'type', 'task'
|
|
]
|
|
|
|
def create(self, validated_data):
|
|
with transaction.atomic():
|
|
comment = Comment.objects.create(
|
|
created_by=self.context['request'].user,
|
|
**validated_data
|
|
)
|
|
return comment
|