views.py

   1 from django.http import HttpResponse
   2 from django.contrib.auth.models import User
   3 from forms import SandwichForm, IngredientForm, NewUserForm
   4 from django.shortcuts import render_to_response
   5 from django.core.files.uploadedfile import SimpleUploadedFile
   6 from models import Sandwich, Ingredient
   7 from django.http import Http404
   8 from django.contrib.auth import authenticate, login
   9 from django.contrib.auth.forms import AuthenticationForm
  10 import datetime
  11
  12
  13 def add_sandwich(request):
  14         if Sandwich.objects.count() > 5:
  15                 sandwiches = Sandwich.objects.order_by('date_made')[:5]
  16         else:
  17                 sandwiches = Sandwich.objects.order_by('date_made')
  18         if request.user.is_authenticated():
  19                 if request.method == 'POST': # If the form has been submitted...
  20                         form = SandwichForm(request.POST, request.FILES) # A form bound to the POST data
  21                         if form.is_valid(): # All validation rules pass
  22                                 newsandwich = form.save()
  23                                 newsandwich.save()
  24                                 thankshtml = "<p class=\"formthanks\">Thanks! Your sandwich has been added!</p>"
  25                                 return HttpResponse(thankshtml) # Redirect after POST
  26                 else:
  27                         form = SandwichForm() # An unbound form
  28                 return render_to_response('sandwich.html', {'sform': form, 'sandwiches': sandwiches,})
  29         else:
  30                 thankshtml = "<p class=\"formthanks\">You are not logged in.</p>"
  31                 return HttpResponse(thankshtml) # Redirect after POST
  32
  33 def add_ingredient(request):
  34         if request.user.is_authenticated():
  35                 if Sandwich.objects.count() > 5:
  36                         sandwiches = Sandwich.objects.order_by('date_made')[:5]
  37                 else:
  38                         sandwiches = Sandwich.objects.order_by('date_made')
  39                 if request.method == 'POST': # If the form has been submitted...
  40                         form = IngredientForm(request.POST) # A form bound to the POST data
  41                         if form.is_valid(): # All validation rules pass
  42                                 newsandwich = form.save()
  43                                 newsandwich.save()
  44                                 thankshtml = "<p class=\"formthanks\">Thanks! Your ingredient has been saved!</p>"
  45                                 return HttpResponse(thankshtml) # Redirect after POST
  46                 else:
  47                         form = IngredientForm() # An unbound form
  48
  49                 return render_to_response('ingredient.html', {'iform': form, 'sandwiches': sandwiches})
  50         else:
  51                 thankshtml = "<p class=\"formthanks\">You are not logged in.</p>"
  52                 return HttpResponse(thankshtml) # Redirect after POST
  53
  54 def all_sandwich(request):
  55         try:
  56                 allsandwiches = Sandwich.objects.all()
  57                 if Sandwich.objects.count() > 5:
  58                         sandwiches = Sandwich.objects.order_by('date_made')[:5]
  59                 else:
  60                         sandwiches = Sandwich.objects.order_by('date_made')
  61         except Sandwich.DoesNotExist:
  62                 raise Http404
  63         return render_to_response('allsandwiches.html', {'allsandwiches': allsandwiches, 'sandwiches': sandwiches})
  64
  65
  66 def baseview(request):
  67         try:
  68                 if Sandwich.objects.count() > 5:
  69                         sandwiches = Sandwich.objects.order_by('date_made')[:5]
  70                         allsandwiches = Sandwich.objects.all()
  71                 else:
  72                         sandwiches = Sandwich.objects.order_by('date_made')
  73                         allsandwiches = Sandwich.objects.all()
  74         except Sandwich.DoesNotExist:
  75                 raise Http404
  76         return render_to_response('base.html', {'sandwiches': sandwiches, 'all': allsandwiches,})
  77
  78
  79 def sandwich_month(request, year, month):
  80         try:
  81                 ms = Sandwich.objects.filter(date_made__month=month, date_made__year=year)
  82                 if Sandwich.objects.count() > 5:
  83                         sandwiches = Sandwich.objects.order_by('date_made')[:5]
  84                 else:
  85                         sandwiches = Sandwich.objects.order_by('date_made')
  86         except Sandwich.DoesNotExist:
  87                 raise Http404
  88         return render_to_response('allsandwiches.html', {'allsandwiches': ms, 'sandwiches': sandwiches,})
  89
  90
  91 def specific_sandwich(request, slug):
  92         try:
  93                 s = Sandwich.objects.get(slug=slug)
  94                 if Sandwich.objects.count() > 5:
  95                         sandwiches = Sandwich.objects.order_by('date_made')[:5]
  96                 else:
  97                         sandwiches = Sandwich.objects.order_by('date_made')
  98         except Sandwich.DoesNotExist:
  99                 raise Http404
 100         return render_to_response('onesandwich.html', {'s': s, 'sandwiches': sandwiches,})
 101
 102
 103 def login_view(request):
 104         if Sandwich.objects.count() > 5:
 105                 sandwiches = Sandwich.objects.order_by('date_made')[:5]
 106         else:
 107                 sandwiches = Sandwich.objects.order_by('date_made')
 108         try:
 109                 username = request.POST['username']
 110                 password = request.POST['password']
 111                 user = authenticate(username=username, password=password)
 112                 if user is not None:
 113                         if user.is_active:
 114                                 login(request, user)
 115                                 thankshtml = "<p class=\"formthanks\">You have been logged in</p>"
 116                                 return HttpResponse(thankshtml)
 117                         else:
 118                                 thankshtml = "<p class=\"formthanks\">Disabled account. Sorry.</p>"
 119                                 return HttpResponse(thankshtml)
 120                 else:
 121                         thankshtml = "<p class=\"formthanks\">Invalid login!</p>"
 122                         return HttpResponse(thankshtml)
 123         except KeyError:
 124                 aform = AuthenticationForm()
 125                 return render_to_response('login.html', {'aform': aform, 'sandwiches': sandwiches,})
 126
 127
 128 def create_user(request):
 129         if Sandwich.objects.count() > 5:
 130                 sandwiches = Sandwich.objects.order_by('date_made')[:5]
 131         else:
 132                 sandwiches = Sandwich.objects.order_by('date_made')
 133         if request.user.is_authenticated():
 134                 thankshtml = "<p class=\"formthanks\">You are already logged in!</p>"
 135                 return HttpResponse(thankshtml)
 136         elif request.method == 'POST': # If the form has been submitted...
 137                 form = NewUserForm(request.POST) # A form bound to the POST data
 138                 if form.is_valid(): # All validation rules pass
 139                         username = form.cleaned_data['username']
 140                         first_name = form.cleaned_data['first_name']
 141                         last_name = form.cleaned_data['last_name']
 142                         password = form.cleaned_data['password']
 143                         cpassword = form.cleaned_data['confirm_password']
 144                         email = form.cleaned_data['email']
 145                         if password == cpassword:
 146                                 user = User.objects.create_user(username, email, password)
 147                                 user.save()
 148                                 user.first_name = first_name
 149                                 user.last_name = last_name
 150                                 user.save()
 151                                 thankshtml = "<p class=\"formthanks\">Thanks! You are now a new user!</p>"
 152                                 return HttpResponse(thankshtml) # Redirect after POST
 153                         else:
 154                                 thankshtml = "<p class=\"formthanks\">Your passwords don't match!</p>"
 155                                 return HttpResponse(thankshtml) # Redirect after POST
 156         else:
 157                 form = NewUserForm() # An unbound form
 158                 return render_to_response('newuser.html', {'cform': form, 'sandwiches': sandwiches,})