core/views.py

   1 from django.shortcuts import render, redirect
   2 from django.contrib.auth import login, authenticate
   3 from .forms import UnifiedRegisterForm, SketchLoginForm
   4
   5 def login_register_view(request):
   6     # Prepare standard empty forms for GET requests
   7     login_form = SketchLoginForm()
   8     register_form = UnifiedRegisterForm()
   9
  10     # Track which section should be visible on reload/error
  11     active_form = 'login'
  12
  13     if request.method == 'POST':
  14         # Scenario 1: User submitted the REGISTER form
  15         if 'register_submit' in request.POST:
  16             active_form = 'register'
  17             register_form = UnifiedRegisterForm(request.POST)
  18             if register_form.is_valid():
  19                 user = register_form.save()
  20                 login(request, user) # Auto-login after registration
  21                 return redirect('homepage') # Define this later
  22             # else will reload page with errors displayed
  23
  24         # Scenario 2: User submitted the LOGIN form
  25         elif 'login_submit' in request.POST:
  26             active_form = 'login'
  27             login_form = SketchLoginForm(request, data=request.POST)
  28             if login_form.is_valid():
  29                 login(request, login_form.get_user())
  30                 return redirect('homepage') # Define this later
  31             # else reloads page displaying authentication errors
  32
  33     return render(request, 'core/index.html', {
  34         'login_form': login_form,
  35         'register_form': register_form,
  36         'active_form': active_form, # Tells template which tab is active
  37     })
  38
  39 def homepage_view(request):
  40     return render(request, 'core/home.html', {})