You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
102 lines
3.3 KiB
Python
102 lines
3.3 KiB
Python
from channels.generic.websocket import WebsocketConsumer
|
|
from django.shortcuts import get_object_or_404
|
|
from customercore.models import *
|
|
import json
|
|
from django.template.loader import render_to_string
|
|
from asgiref.sync import async_to_sync
|
|
from channels.generic.websocket import WebsocketConsumer
|
|
from django.shortcuts import get_object_or_404
|
|
from customercore.models import *
|
|
import json
|
|
from django.template.loader import render_to_string
|
|
from asgiref.sync import async_to_sync
|
|
|
|
|
|
class TicketRoomConsumer(WebsocketConsumer):
|
|
def connect(self):
|
|
self.user = self.scope['user']
|
|
self.ticket_id = self.scope['url_route']['kwargs']['ticket_id']
|
|
self.ticket = get_object_or_404(Ticket, id=self.ticket_id)
|
|
self.ticket_number = self.ticket.ticket_number
|
|
async_to_sync(self.channel_layer.group_add)(
|
|
self.ticket_number, self.channel_name
|
|
)
|
|
self.accept()
|
|
|
|
def disconnect(self, close_code):
|
|
async_to_sync(self.channel_layer.group_discard)(
|
|
self.ticket_number, self.channel_name
|
|
)
|
|
|
|
def receive(self, text_data):
|
|
text_data_json = json.loads(text_data)
|
|
event_type = text_data_json.get('event_type')
|
|
|
|
if event_type == 'typing':
|
|
event = {
|
|
'type': 'typing_handler',
|
|
'user': self.scope['user']
|
|
}
|
|
async_to_sync(self.channel_layer.group_send)(
|
|
self.ticket_number, event
|
|
)
|
|
elif event_type == 'stop_typing':
|
|
event = {
|
|
'type': 'stop_typing_handler',
|
|
}
|
|
async_to_sync(self.channel_layer.group_send)(
|
|
self.ticket_number, event
|
|
)
|
|
else:
|
|
body = text_data_json['description']
|
|
file_paths = text_data_json['filePath']
|
|
ticketupdate = TicketUpdate.objects.create(
|
|
added_by=self.user,
|
|
description=body,
|
|
ticket=self.ticket,
|
|
date_added=datetime.now()
|
|
)
|
|
for file_path in file_paths:
|
|
ticket_attachment = TicketAttachment(
|
|
ticket_update=ticketupdate,
|
|
file_path=file_path
|
|
)
|
|
ticket_attachment.save()
|
|
event = {
|
|
'type': 'update_handler',
|
|
'update_id': ticketupdate.id
|
|
}
|
|
async_to_sync(self.channel_layer.group_send)(
|
|
self.ticket_number, event
|
|
)
|
|
|
|
def update_handler(self, event):
|
|
update_id = event['update_id']
|
|
update = TicketUpdate.objects.get(id=update_id)
|
|
context = {
|
|
'update': update,
|
|
'user': self.user
|
|
}
|
|
|
|
|
|
html = render_to_string("details_templates/new-ticket-message.html", context=context)
|
|
self.send(text_data=json.dumps({
|
|
'event_type': 'update',
|
|
'html': html
|
|
}))
|
|
|
|
def typing_handler(self, event):
|
|
context = {
|
|
'user': event['user']
|
|
}
|
|
html = render_to_string("details_templates/typing-message.html", context=context)
|
|
self.send(text_data=json.dumps({
|
|
'event_type': 'typing',
|
|
'html': html
|
|
}))
|
|
|
|
def stop_typing_handler(self, event):
|
|
self.send(text_data=json.dumps({
|
|
'event_type': 'stop_typing'
|
|
}))
|