116 lines
3.4 KiB
Python
116 lines
3.4 KiB
Python
from .forms import LoginForm, SignupForm
|
|
from asgiref.sync import async_to_sync
|
|
from django.template.loader import render_to_string
|
|
from django.urls import reverse
|
|
from channels.auth import login, logout
|
|
from django.contrib.auth.models import User
|
|
from django.contrib.auth import authenticate
|
|
from datetime import datetime
|
|
|
|
|
|
def send_page(self, page):
|
|
"""Render HTML and send page to client"""
|
|
|
|
# Prepare context data for page
|
|
context = {}
|
|
match page:
|
|
case "login":
|
|
context = {"form": LoginForm()}
|
|
case "signup":
|
|
context = {"form": SignupForm()}
|
|
|
|
# Add user to context if logged in
|
|
if "user" in self.scope:
|
|
context.update({ "user": self.scope["user"]})
|
|
context.update({"active_nav": page})
|
|
|
|
# Render HTML nav and send to client
|
|
self.send_html({
|
|
"selector": "#nav",
|
|
"html": render_to_string("components/_nav.html", context),
|
|
})
|
|
|
|
# Render HTML page and send to client
|
|
self.send_html({
|
|
"selector": "#main",
|
|
"html": render_to_string(f"pages/{page}.html", context),
|
|
"url": reverse(page),
|
|
})
|
|
|
|
# Hidrate page
|
|
match page:
|
|
case "home":
|
|
update_TODO(self)
|
|
|
|
|
|
def action_signup(self, data):
|
|
"""Sign up user"""
|
|
form = SignupForm(data)
|
|
user_exist = User.objects.filter(email=data["email"]).exists()
|
|
if form.is_valid() and data["password"] == data["password_confirm"] and not user_exist:
|
|
# Create user
|
|
user = User.objects.create_user(data["username"], data["email"], data["password"])
|
|
user.is_active = True
|
|
user.save()
|
|
# Login user
|
|
send_page(self, "login")
|
|
else:
|
|
# Send form errors
|
|
self.send_html({
|
|
"selector": "#main",
|
|
"html": render_to_string("pages/signup.html", {"form": form, "user_exist": user_exist, "passwords_do_not_match": data["password"] != data["password_confirm"]}),
|
|
"append": False,
|
|
"url": reverse("signup")
|
|
})
|
|
|
|
|
|
def action_login(self, data):
|
|
"""Log in user"""
|
|
form = LoginForm(data)
|
|
user = authenticate(username=data["email"], password=data["password"])
|
|
if form.is_valid() and user:
|
|
async_to_sync(login)(self.scope, user)
|
|
self.scope["session"].save()
|
|
send_page(self, "profile")
|
|
else:
|
|
self.send_html({
|
|
"selector": "#main",
|
|
"html": render_to_string("pages/login.html", {"form": form, "user_does_not_exist": user is None}),
|
|
"append": False,
|
|
"url": reverse("login")
|
|
})
|
|
|
|
|
|
def action_logout(self):
|
|
"""Log out user"""
|
|
async_to_sync(logout)(self.scope)
|
|
self.scope["session"].save()
|
|
send_page(self, "login")
|
|
|
|
|
|
def add_lap(self):
|
|
"""Add lap to Home page"""
|
|
# Send current time to client
|
|
self.send_html({
|
|
"selector": "#laps",
|
|
"html": render_to_string("components/_lap.html", {"time": datetime.now()}),
|
|
"append": True,
|
|
})
|
|
|
|
|
|
def add_task(self, data):
|
|
"""Add task from TODO section"""
|
|
# Add task to list
|
|
self.scope["session"]["tasks"].append(data["task"])
|
|
self.scope["session"].save()
|
|
# Update task list
|
|
update_TODO(self)
|
|
|
|
|
|
def update_TODO(self):
|
|
"""Update TODO list"""
|
|
self.send_html({
|
|
"selector": "#todo",
|
|
"html": render_to_string("components/_tasks.html", {"tasks": self.scope["session"]["tasks"]}),
|
|
"append": False,
|
|
}) |