diff options
Diffstat (limited to 'chat_api/serializers.py')
| -rw-r--r-- | chat_api/serializers.py | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/chat_api/serializers.py b/chat_api/serializers.py index 50d84e0..585bb36 100644 --- a/chat_api/serializers.py +++ b/chat_api/serializers.py | |||
| @@ -3,6 +3,15 @@ from rest_framework import serializers | |||
| 3 | from .models import * | 3 | from .models import * |
| 4 | 4 | ||
| 5 | class GroupSerializer(serializers.ModelSerializer): | 5 | class GroupSerializer(serializers.ModelSerializer): |
| 6 | is_mine = serializers.SerializerMethodField('_is_mine') | ||
| 7 | |||
| 8 | def _is_mine(self, obj): | ||
| 9 | request = self.context.get('request') | ||
| 10 | if request: | ||
| 11 | return request.user == obj.owner | ||
| 12 | return False | ||
| 13 | |||
| 6 | class Meta: | 14 | class Meta: |
| 7 | model = Group | 15 | model = Group |
| 8 | fields = '__all__' | 16 | fields = '__all__' |
| 17 | extra_fields = ('is_mine', ) | ||
