55 lines
1.7 KiB
Python
55 lines
1.7 KiB
Python
|
# app/app_template/consumers.py
|
||
|
|
from channels.generic.websocket import JsonWebsocketConsumer
|
||
|
|
import app.app_template.actions as actions
|
||
|
|
|
||
|
|
|
||
|
|
class ExampleConsumer(JsonWebsocketConsumer):
|
||
|
|
|
||
|
|
def connect(self):
|
||
|
|
"""Event when client connects"""
|
||
|
|
# Accept the connection
|
||
|
|
self.accept()
|
||
|
|
# Make session task list
|
||
|
|
if "tasks" not in self.scope["session"]:
|
||
|
|
self.scope["session"]["tasks"] = []
|
||
|
|
self.scope["session"].save()
|
||
|
|
|
||
|
|
|
||
|
|
def disconnect(self, close_code):
|
||
|
|
"""Event when client disconnects"""
|
||
|
|
pass
|
||
|
|
|
||
|
|
def receive_json(self, data_received):
|
||
|
|
"""
|
||
|
|
Event when data is received
|
||
|
|
All information will arrive in 2 variables:
|
||
|
|
"action", with the action to be taken
|
||
|
|
"data" with the information
|
||
|
|
"""
|
||
|
|
# Get the data
|
||
|
|
data = data_received["data"]
|
||
|
|
# Depending on the action we will do one task or another.
|
||
|
|
match data_received["action"]:
|
||
|
|
case "Change page":
|
||
|
|
actions.send_page(self, data["page"])
|
||
|
|
case "Signup":
|
||
|
|
actions.action_signup(self, data)
|
||
|
|
case "Login":
|
||
|
|
actions.action_login(self, data)
|
||
|
|
case "Logout":
|
||
|
|
actions.action_logout(self)
|
||
|
|
case "Add lap":
|
||
|
|
actions.add_lap(self)
|
||
|
|
case "Add task":
|
||
|
|
actions.add_task(self, data)
|
||
|
|
|
||
|
|
|
||
|
|
def send_html(self, event):
|
||
|
|
"""Event: Send html to client"""
|
||
|
|
data = {
|
||
|
|
"selector": event["selector"],
|
||
|
|
"html": event["html"],
|
||
|
|
"append": "append" in event and event["append"],
|
||
|
|
"url": event["url"] if "url" in event else "",
|
||
|
|
}
|
||
|
|
self.send_json(data)
|