Adding send message in api chat

Zambom
1 parent c1b04946
Showing 1 changed file with 89 additions and 1 deletions Show diff stats
api/views.py
@@ -2,6 +2,14 @@ import requests, json
 from django.shortcuts import get_object_or_404, reverse
 from django.contrib.auth import authenticate
 from django.views.decorators.csrf import csrf_exempt
+from django.utils.translation import ugettext as _
+from django.template.loader import render_to_string
+import textwrap
+from datetime import datetime
+from django.utils import formats
+from django.utils.html import strip_tags
+
+from channels import Group
 from rest_framework import viewsets
 from rest_framework.response import Response
@@ -13,7 +21,7 @@ from django.db.models import Q
 from security.models import Security
 from chat.serializers import ChatSerializer
-from chat.models import TalkMessages
+from chat.models import TalkMessages, Conversation, ChatVisualizations
 from subjects.serializers import SubjectSerializer
 from subjects.models import Subject
@@ -221,4 +229,84 @@ class ChatViewset(viewsets.ModelViewSet):
 			response = json.dumps(info)
+		return HttpResponse(response)
+
+	@csrf_exempt
+	@list_route(methods = ['POST'], permissions_classes = [IsAuthenticated])
+	def send_message(self, request):
+		username = request.data['email']
+		user_two = request.data['user_two']
+		subject = request.data['subject']
+		msg_text = request.data['text']
+		create_date = request.data['create_date']
+
+		info = {}
+
+		if not user_two == "" and not username == "":
+			user = User.objects.get(email = username)
+			user_to = User.objects.get(email = user_two)
+
+			talks = Conversation.objects.filter((Q(user_one__email = username) & Q(user_two__email = user_two)) | (Q(user_two__email = username) & Q(user_one__email = user_two)))
+
+			if talks.count() > 0:
+				talk = talks[0]
+			else:
+				talk = Conversation()
+				talk.user_one = user
+				talk.user_two = user_to
+
+				talk.save()
+
+			subject = Subject.objects.get(slug = subject)
+
+			message = TalkMessages()
+			message.text = "<p>" + msg_text + "</p>"
+			message.user = user
+			message.talk = talk
+
+			message.save()
+
+			if not message.pk is None:
+				simple_notify = textwrap.shorten(strip_tags(message.text), width = 30, placeholder = "...")
+				
+				notification = {
+					"type": "chat",
+					"subtype": "subject",
+					"space": subject.slug,
+					"user_icon": message.user.image_url,
+					"notify_title": str(message.user),
+					"simple_notify": simple_notify,
+					"view_url": reverse("chat:view_message", args = (message.id, ), kwargs = {}),
+					"complete": render_to_string("chat/_message.html", {"talk_msg": message}, request),
+					"container": "chat-" + str(message.user.id),
+					"last_date": _("Last message in %s")%(formats.date_format(message.create_date, "SHORT_DATETIME_FORMAT"))
+				}
+
+				notification = json.dumps(notification)
+
+				Group("user-%s" % user_to.id).send({'text': notification})
+
+				ChatVisualizations.objects.create(viewed = False, message = message, user = user_to)
+
+				info["message"] = _("Message sent successfully!")
+				info["success"] = True
+				info["number"] = 1
+			else:
+				info["message"] = _("Error while sending message!")
+				info["success"] = False
+				info["number"] = 0
+		else:
+			info["message"] = _("No information received!")
+			info["success"] = False
+			info["number"] = 0
+
+		info["data"] = {}
+		info["data"]["messages"] = []
+
+		info["type"] = ""
+		info["title"] = _("Amadeus")
+		info['extra'] = 0
+
+		response = json.dumps(info)
+
 		return HttpResponse(response)
 \ No newline at end of file
	@@ -2,6 +2,14 @@ import requests, json		@@ -2,6 +2,14 @@ import requests, json
2	from django.shortcuts import get_object_or_404, reverse	2	from django.shortcuts import get_object_or_404, reverse
3	from django.contrib.auth import authenticate	3	from django.contrib.auth import authenticate
4	from django.views.decorators.csrf import csrf_exempt	4	from django.views.decorators.csrf import csrf_exempt
		5	+from django.utils.translation import ugettext as _
		6	+from django.template.loader import render_to_string
		7	+import textwrap
		8	+from datetime import datetime
		9	+from django.utils import formats
		10	+from django.utils.html import strip_tags
		11	+
		12	+from channels import Group
5		13
6	from rest_framework import viewsets	14	from rest_framework import viewsets
7	from rest_framework.response import Response	15	from rest_framework.response import Response
	@@ -13,7 +21,7 @@ from django.db.models import Q		@@ -13,7 +21,7 @@ from django.db.models import Q
13	from security.models import Security	21	from security.models import Security
14		22
15	from chat.serializers import ChatSerializer	23	from chat.serializers import ChatSerializer
16	-from chat.models import TalkMessages	24	+from chat.models import TalkMessages, Conversation, ChatVisualizations
17		25
18	from subjects.serializers import SubjectSerializer	26	from subjects.serializers import SubjectSerializer
19	from subjects.models import Subject	27	from subjects.models import Subject
	@@ -221,4 +229,84 @@ class ChatViewset(viewsets.ModelViewSet):		@@ -221,4 +229,84 @@ class ChatViewset(viewsets.ModelViewSet):
221		229
222	response = json.dumps(info)	230	response = json.dumps(info)
223		231
		232	+ return HttpResponse(response)
		233	+
		234	+ @csrf_exempt
		235	+ @list_route(methods = ['POST'], permissions_classes = [IsAuthenticated])
		236	+ def send_message(self, request):
		237	+ username = request.data['email']
		238	+ user_two = request.data['user_two']
		239	+ subject = request.data['subject']
		240	+ msg_text = request.data['text']
		241	+ create_date = request.data['create_date']
		242	+
		243	+ info = {}
		244	+
		245	+ if not user_two == "" and not username == "":
		246	+ user = User.objects.get(email = username)
		247	+ user_to = User.objects.get(email = user_two)
		248	+
		249	+ talks = Conversation.objects.filter((Q(user_one__email = username) & Q(user_two__email = user_two)) \| (Q(user_two__email = username) & Q(user_one__email = user_two)))
		250	+
		251	+ if talks.count() > 0:
		252	+ talk = talks[0]
		253	+ else:
		254	+ talk = Conversation()
		255	+ talk.user_one = user
		256	+ talk.user_two = user_to
		257	+
		258	+ talk.save()
		259	+
		260	+ subject = Subject.objects.get(slug = subject)
		261	+
		262	+ message = TalkMessages()
		263	+ message.text = "<p>" + msg_text + "</p>"
		264	+ message.user = user
		265	+ message.talk = talk
		266	+
		267	+ message.save()
		268	+
		269	+ if not message.pk is None:
		270	+ simple_notify = textwrap.shorten(strip_tags(message.text), width = 30, placeholder = "...")
		271	+
		272	+ notification = {
		273	+ "type": "chat",
		274	+ "subtype": "subject",
		275	+ "space": subject.slug,
		276	+ "user_icon": message.user.image_url,
		277	+ "notify_title": str(message.user),
		278	+ "simple_notify": simple_notify,
		279	+ "view_url": reverse("chat:view_message", args = (message.id, ), kwargs = {}),
		280	+ "complete": render_to_string("chat/_message.html", {"talk_msg": message}, request),
		281	+ "container": "chat-" + str(message.user.id),
		282	+ "last_date": _("Last message in %s")%(formats.date_format(message.create_date, "SHORT_DATETIME_FORMAT"))
		283	+ }
		284	+
		285	+ notification = json.dumps(notification)
		286	+
		287	+ Group("user-%s" % user_to.id).send({'text': notification})
		288	+
		289	+ ChatVisualizations.objects.create(viewed = False, message = message, user = user_to)
		290	+
		291	+ info["message"] = _("Message sent successfully!")
		292	+ info["success"] = True
		293	+ info["number"] = 1
		294	+ else:
		295	+ info["message"] = _("Error while sending message!")
		296	+ info["success"] = False
		297	+ info["number"] = 0
		298	+ else:
		299	+ info["message"] = _("No information received!")
		300	+ info["success"] = False
		301	+ info["number"] = 0
		302	+
		303	+ info["data"] = {}
		304	+ info["data"]["messages"] = []
		305	+
		306	+ info["type"] = ""
		307	+ info["title"] = _("Amadeus")
		308	+ info['extra'] = 0
		309	+
		310	+ response = json.dumps(info)
		311	+
224	return HttpResponse(response)	312	return HttpResponse(response)
225	\ No newline at end of file	313	\ No newline at end of file