views.py

   1 from django.http import HttpResponse, Http404
   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.contrib.auth import authenticate, login, logout
   8 from django.contrib.auth.forms import AuthenticationForm
   9 import datetime
  10
  11
  12 def add_sandwich(request):
  13         if Sandwich.objects.count() > 5:
  14                 sandwiches = Sandwich.objects.order_by('-date_made')[:5]
  15         else:
  16                 sandwiches = Sandwich.objects.order_by('-date_made')
  17         if request.user.is_authenticated():
  18                 if request.method == 'POST': # If the form has been submitted...
  19                         form = SandwichForm(request.POST, request.FILES) # A form bound to the POST data
  20                         if form.is_valid(): # All validation rules pass
  21                                 newsandwich = form.save(commit=False)
  22                                 newsandwich.user = request.user
  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(initial={'user': request.user}) # An unbound form
  28                 return render_to_response('sandwich.html', {'sform': form, 'sandwiches': sandwiches, 'user': request.user,})
  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, 'user': request.user,})
  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, 'user': request.user,})
  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, 'user': request.user,})
  89
  90 def current_home(request):
  91         temp = Sandwich.objects.order_by('-date_made')[0]
  92         curr_month = temp.date_made.month
  93         curr_year = temp.date_made.year
  94         try:
  95                 ms = Sandwich.objects.filter(date_made__month=curr_month, date_made__year=curr_year)
  96                 if Sandwich.objects.count() > 5:
  97                         sandwiches = Sandwich.objects.order_by('-date_made')[:5]
  98                 else:
  99                         sandwiches = Sandwich.objects.order_by('-date_made')
 100         except Sandwich.DoesNotExist:
 101                 raise Http404
 102         return render_to_response('allsandwiches.html', {'allsandwiches': ms, 'sandwiches': sandwiches, 'user': request.user,})
 103
 104
 105 def specific_sandwich(request, slug):
 106         try:
 107                 s = Sandwich.objects.get(slug=slug)
 108                 if Sandwich.objects.count() > 5:
 109                         sandwiches = Sandwich.objects.order_by('-date_made')[:5]
 110                 else:
 111                         sandwiches = Sandwich.objects.order_by('-date_made')
 112         except Sandwich.DoesNotExist:
 113                 raise Http404
 114         return render_to_response('onesandwich.html', {'s': s, 'sandwiches': sandwiches, 'user': request.user,})
 115
 116 def logout_view(request):
 117         if Sandwich.objects.count() > 5:
 118                 sandwiches = Sandwich.objects.order_by('-date_made')[:5]
 119         else:
 120                 sandwiches = Sandwich.objects.order_by('-date_made')
 121         if request.user.is_authenticated():
 122                 logout(request)
 123                 thankshtml = "<p class=\"formthanks\">You have been logged out.</p>"
 124                 return HttpResponse(thankshtml)
 125         else:
 126                 thankshtml = "<p class=\"formthanks\">You are not logged in.</p>"
 127                 return HttpResponse(thankshtml)
 128
 129
 130 def login_view(request):
 131         if Sandwich.objects.count() > 5:
 132                 sandwiches = Sandwich.objects.order_by('-date_made')[:5]
 133         else:
 134                 sandwiches = Sandwich.objects.order_by('-date_made')
 135         try:
 136                 username = request.POST['username']
 137                 password = request.POST['password']
 138                 user = authenticate(username=username, password=password)
 139                 if user is not None:
 140                         if user.is_active:
 141                                 login(request, user)
 142                                 thankshtml = "<p class=\"formthanks\">You have been logged in</p>"
 143                                 return HttpResponse(thankshtml)
 144                         else:
 145                                 thankshtml = "<p class=\"formthanks\">Disabled account. Sorry.</p>"
 146                                 return HttpResponse(thankshtml)
 147                 else:
 148                         thankshtml = "<p class=\"formthanks\">Invalid login!</p>"
 149                         return HttpResponse(thankshtml)
 150         except KeyError:
 151                 aform = AuthenticationForm()
 152                 return render_to_response('login.html', {'aform': aform, 'sandwiches': sandwiches, 'user': request.user,})
 153
 154
 155 def create_user(request):
 156         if Sandwich.objects.count() > 5:
 157                 sandwiches = Sandwich.objects.order_by('-date_made')[:5]
 158         else:
 159                 sandwiches = Sandwich.objects.order_by('-date_made')
 160         if request.user.is_authenticated():
 161                 thankshtml = "<p class=\"formthanks\">You are already logged in!</p>"
 162                 return HttpResponse(thankshtml)
 163         elif request.method == 'POST': # If the form has been submitted...
 164                 form = NewUserForm(request.POST) # A form bound to the POST data
 165                 if form.is_valid(): # All validation rules pass
 166                         username = form.cleaned_data['username']
 167                         first_name = form.cleaned_data['first_name']
 168                         last_name = form.cleaned_data['last_name']
 169                         password = form.cleaned_data['password']
 170                         cpassword = form.cleaned_data['confirm_password']
 171                         email = form.cleaned_data['email']
 172                         if password == cpassword:
 173                                 user = User.objects.create_user(username, email, password)
 174                                 user.save()
 175                                 user.first_name = first_name
 176                                 user.last_name = last_name
 177                                 user.save()
 178                                 thankshtml = "<p class=\"formthanks\">Thanks! You are now a new user!</p>"
 179                                 return HttpResponse(thankshtml) # Redirect after POST
 180                         else:
 181                                 thankshtml = "<p class=\"formthanks\">Your passwords don't match!</p>"
 182                                 return HttpResponse(thankshtml) # Redirect after POST
 183         else:
 184                 form = NewUserForm() # An unbound form
 185                 return render_to_response('newuser.html', {'cform': form, 'sandwiches': sandwiches, 'user': request.user,})