django-multipage-template-for-html-over-the-wire/app/app_template/actions.py

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":
            pass


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_task(self, data):
    """Update TODO list"""
    self.send_html({
        "selector": "#todo-list",
        "html": render_to_string("components/_task.html", {"task": data["newTask"]}),
        "append": True,
    })