demo-HTML-over-WebSockets-i.../apps/back/consumers.py
Andros Fenollosa 7800581731 Add demo
2021-03-13 11:07:46 +01:00

41 lines
1.2 KiB
Python

import json
from channels.generic.websocket import WebsocketConsumer
from django.template.loader import render_to_string
from apps.back.models import Post, Comment
class BlogConsumer(WebsocketConsumer):
def connect(self):
self.accept()
def disconnect(self, close_code):
pass
def receive(self, text_data):
text_data_json = json.loads(text_data)
selector = text_data_json["selector"]
template = text_data_json["template"]
data = text_data_json["data"]
# Database
if template == "partials/blog/all_articles.html":
pag = data['pag'] if 'pag' in data else 1
amount = 3
start = pag - 1
end = start + amount
data["posts"] = Post.objects.all()[start:end]
data['pag'] = pag
if template == "partials/blog/single.html":
data["post"] = Post.objects.get(pk=data['id'])
data["comments"] = Comment.objects.filter(post__id=data['id']).all()
# Send message to WebSocket
self.send(
text_data=json.dumps(
{
"selector": selector,
"html": render_to_string(template, data)
}
)
)