Default view is now latest sandwich's month's sandwiches. Fixed other minor things.
[~kgodey/maayanwich.git] / views.py
index 2b92417..2a175af 100644 (file)
--- a/views.py
+++ b/views.py
-from django.http import HttpResponse
+from django.http import HttpResponse, Http404
 from django.contrib.auth.models import User
 from forms import SandwichForm, IngredientForm, NewUserForm
 from django.shortcuts import render_to_response
 from django.core.files.uploadedfile import SimpleUploadedFile
 from models import Sandwich, Ingredient
-from django.http import Http404
-from django.contrib.auth import authenticate, login
+from django.contrib.auth import authenticate, login, logout
 from django.contrib.auth.forms import AuthenticationForm
 import datetime
 
 
 def add_sandwich(request):
        if Sandwich.objects.count() > 5:
-               sandwiches = Sandwich.objects.order_by('date_made')[:5]
+               sandwiches = Sandwich.objects.order_by('-date_made')[:5]
        else:
-               sandwiches = Sandwich.objects.order_by('date_made')
-       if request.method == 'POST': # If the form has been submitted...
-               form = SandwichForm(request.POST, request.FILES) # A form bound to the POST data
-               if form.is_valid(): # All validation rules pass
-                       newsandwich = form.save()
-                       newsandwich.save()
-                       thankshtml = "<p class=\"formthanks\">Thanks! Your sandwich has been added!</p>"
-                       return HttpResponse(thankshtml) # Redirect after POST
+               sandwiches = Sandwich.objects.order_by('-date_made')
+       if request.user.is_authenticated():
+               if request.method == 'POST': # If the form has been submitted...
+                       form = SandwichForm(request.POST, request.FILES) # A form bound to the POST data
+                       if form.is_valid(): # All validation rules pass
+                               newsandwich = form.save()
+                               newsandwich.save()
+                               thankshtml = "<p class=\"formthanks\">Thanks! Your sandwich has been added!</p>"
+                               return HttpResponse(thankshtml) # Redirect after POST
+               else:
+                       form = SandwichForm() # An unbound form
+               return render_to_response('sandwich.html', {'sform': form, 'sandwiches': sandwiches, 'user': request.user,})
        else:
-               form = SandwichForm() # An unbound form
-               
-       return render_to_response('sandwich.html', {'sform': form, 'sandwiches': sandwiches,})
-
+               thankshtml = "<p class=\"formthanks\">You are not logged in.</p>"
+               return HttpResponse(thankshtml) # Redirect after POST
 
 def add_ingredient(request):
-       if Sandwich.objects.count() > 5:
-               sandwiches = Sandwich.objects.order_by('date_made')[:5]
-       else:
-               sandwiches = Sandwich.objects.order_by('date_made')
-       if request.method == 'POST': # If the form has been submitted...
-               form = IngredientForm(request.POST) # A form bound to the POST data
-               if form.is_valid(): # All validation rules pass
-                       newsandwich = form.save()
-                       newsandwich.save()
-                       thankshtml = "<p class=\"formthanks\">Thanks! Your ingredient has been saved!</p>"
-                       return HttpResponse(thankshtml) # Redirect after POST
-       else:
-               form = IngredientForm() # An unbound form
+       if request.user.is_authenticated():
+               if Sandwich.objects.count() > 5:
+                       sandwiches = Sandwich.objects.order_by('-date_made')[:5]
+               else:
+                       sandwiches = Sandwich.objects.order_by('-date_made')
+               if request.method == 'POST': # If the form has been submitted...
+                       form = IngredientForm(request.POST) # A form bound to the POST data
+                       if form.is_valid(): # All validation rules pass
+                               newsandwich = form.save()
+                               newsandwich.save()
+                               thankshtml = "<p class=\"formthanks\">Thanks! Your ingredient has been saved!</p>"
+                               return HttpResponse(thankshtml) # Redirect after POST
+               else:
+                       form = IngredientForm() # An unbound form
 
-       return render_to_response('ingredient.html', {'iform': form, 'sandwiches': sandwiches})
-       
+               return render_to_response('ingredient.html', {'iform': form, 'sandwiches': sandwiches, 'user': request.user,})
+       else:
+               thankshtml = "<p class=\"formthanks\">You are not logged in.</p>"
+               return HttpResponse(thankshtml) # Redirect after POST
 
 def all_sandwich(request):
        try:
                allsandwiches = Sandwich.objects.all()
                if Sandwich.objects.count() > 5:
-                       sandwiches = Sandwich.objects.order_by('date_made')[:5]
+                       sandwiches = Sandwich.objects.order_by('-date_made')[:5]
                else:
-                       sandwiches = Sandwich.objects.order_by('date_made')
+                       sandwiches = Sandwich.objects.order_by('-date_made')
        except Sandwich.DoesNotExist:
                raise Http404
-       return render_to_response('allsandwiches.html', {'allsandwiches': allsandwiches, 'sandwiches': sandwiches})
+       return render_to_response('allsandwiches.html', {'allsandwiches': allsandwiches, 'sandwiches': sandwiches, 'user': request.user,})
+
 
-       
 def baseview(request):
        try:
                if Sandwich.objects.count() > 5:
-                       sandwiches = Sandwich.objects.order_by('date_made')[:5]
+                       sandwiches = Sandwich.objects.order_by('-date_made')[:5]
                        allsandwiches = Sandwich.objects.all()
                else:
-                       sandwiches = Sandwich.objects.order_by('date_made')
+                       sandwiches = Sandwich.objects.order_by('-date_made')
                        allsandwiches = Sandwich.objects.all()
        except Sandwich.DoesNotExist:
                raise Http404
        return render_to_response('base.html', {'sandwiches': sandwiches, 'all': allsandwiches,})
-       
-       
+
+
 def sandwich_month(request, year, month):
        try:
-               sandwiches = Sandwich.objects.filter(date_made__month=month, date_made__year=year)
+               ms = Sandwich.objects.filter(date_made__month=month, date_made__year=year)
+               if Sandwich.objects.count() > 5:
+                       sandwiches = Sandwich.objects.order_by('-date_made')[:5]
+               else:
+                       sandwiches = Sandwich.objects.order_by('-date_made')
        except Sandwich.DoesNotExist:
                raise Http404
-       return render_to_response('allsandwiches.html', {'sandwiches': sandwiches,})
-       
+       return render_to_response('allsandwiches.html', {'allsandwiches': ms, 'sandwiches': sandwiches, 'user': request.user,})
        
+def current_home(request):
+       temp = Sandwich.objects.order_by('-date_made')[0]
+       curr_month = temp.date_made.month
+       curr_year = temp.date_made.year
+       try:
+               ms = Sandwich.objects.filter(date_made__month=curr_month, date_made__year=curr_year)
+               if Sandwich.objects.count() > 5:
+                       sandwiches = Sandwich.objects.order_by('-date_made')[:5]
+               else:
+                       sandwiches = Sandwich.objects.order_by('-date_made')
+       except Sandwich.DoesNotExist:
+               raise Http404
+       return render_to_response('allsandwiches.html', {'allsandwiches': ms, 'sandwiches': sandwiches, 'user': request.user,})
+
+
 def specific_sandwich(request, slug):
        try:
                s = Sandwich.objects.get(slug=slug)
                if Sandwich.objects.count() > 5:
-                       sandwiches = Sandwich.objects.order_by('date_made')[:5]
+                       sandwiches = Sandwich.objects.order_by('-date_made')[:5]
                else:
-                       sandwiches = Sandwich.objects.order_by('date_made')
+                       sandwiches = Sandwich.objects.order_by('-date_made')
        except Sandwich.DoesNotExist:
                raise Http404
-       return render_to_response('onesandwich.html', {'s': s, 'sandwiches': sandwiches,})
+       return render_to_response('onesandwich.html', {'s': s, 'sandwiches': sandwiches, 'user': request.user,})
+
+def logout_view(request):
+       if Sandwich.objects.count() > 5:
+               sandwiches = Sandwich.objects.order_by('-date_made')[:5]
+       else:
+               sandwiches = Sandwich.objects.order_by('-date_made')
+       if request.user.is_authenticated():
+               logout(request)
+               thankshtml = "<p class=\"formthanks\">You have been logged out.</p>"
+               return HttpResponse(thankshtml)
+       else:
+               thankshtml = "<p class=\"formthanks\">You are not logged in.</p>"
+               return HttpResponse(thankshtml)
 
 
 def login_view(request):
        if Sandwich.objects.count() > 5:
-               sandwiches = Sandwich.objects.order_by('date_made')[:5]
+               sandwiches = Sandwich.objects.order_by('-date_made')[:5]
        else:
-               sandwiches = Sandwich.objects.order_by('date_made')
+               sandwiches = Sandwich.objects.order_by('-date_made')
        try:
                username = request.POST['username']
                password = request.POST['password']
@@ -113,13 +148,14 @@ def login_view(request):
                        return HttpResponse(thankshtml)
        except KeyError:
                aform = AuthenticationForm()
-               return render_to_response('login.html', {'aform': aform, 'sandwiches': sandwiches,})
-               
+               return render_to_response('login.html', {'aform': aform, 'sandwiches': sandwiches, 'user': request.user,})
+
+
 def create_user(request):
        if Sandwich.objects.count() > 5:
-               sandwiches = Sandwich.objects.order_by('date_made')[:5]
+               sandwiches = Sandwich.objects.order_by('-date_made')[:5]
        else:
-               sandwiches = Sandwich.objects.order_by('date_made')
+               sandwiches = Sandwich.objects.order_by('-date_made')
        if request.user.is_authenticated():
                thankshtml = "<p class=\"formthanks\">You are already logged in!</p>"
                return HttpResponse(thankshtml)
@@ -145,4 +181,4 @@ def create_user(request):
                                return HttpResponse(thankshtml) # Redirect after POST   
        else:
                form = NewUserForm() # An unbound form
-               return render_to_response('newuser.html', {'cform': form, 'sandwiches': sandwiches,})
\ No newline at end of file
+               return render_to_response('newuser.html', {'cform': form, 'sandwiches': sandwiches, 'user': request.user,})
\ No newline at end of file