Some adjusts in chat api

Zambom
1 parent 6ddcd431
Showing 2 changed files with 15 additions and 2 deletions Show diff stats
chat/models.py
chat/serializers.py
 import os
 import time
+from os import path
 from django.db import models
 from django.core import validators
 from django.core.exceptions import ValidationError
 from django.utils.translation import ugettext_lazy as _
+from datetime import timezone
+
 from subjects.models import Subject
 from users.models import User
@@ -33,6 +36,17 @@ class TalkMessages(models.Model):
 	subject = models.ForeignKey(Subject, verbose_name = _('Subject'), related_name = 'message_subject', null = True)
 	create_date = models.DateTimeField(_('Create Date'), auto_now_add = True)
+	def get_timestamp(self):
+		return str(self.create_date.replace(tzinfo = timezone.utc).timestamp())
+
+	@property
+	def image_url(self):
+		if self.image and hasattr(self.image, 'url'):
+			if path.exists(self.image.path):
+				return self.image.url
+		
+		return ""
+
 class ChatVisualizations(models.Model):
 	viewed = models.BooleanField(_('Viewed'), default = False)
 	message = models.ForeignKey(TalkMessages, verbose_name = _('Message'), related_name = 'chat_visualization_message', null = True)
@@ -8,8 +8,7 @@ from users.serializers import UserSerializer
 class ChatSerializer(serializers.ModelSerializer):
 	user = UserSerializer()
 	subject = SubjectSerializer()
-	image = serializers.CharField(required = False, allow_blank = True, max_length = 255)
 	class Meta:
 		model = TalkMessages
-		exclude = ["talk"]
 \ No newline at end of file
+		fields = ('text', 'user', 'subject', 'image_url', 'create_date', )