| from xml.dom.minidom import parseString |
| |
| from django.core import mail |
| from django.template import Context, Template |
| from django.http import HttpResponse, HttpResponseRedirect, HttpResponseNotFound |
| from django.contrib.auth.decorators import login_required, permission_required |
| from django.forms.forms import Form |
| from django.forms import fields |
| from django.shortcuts import render_to_response |
| from django.utils.decorators import method_decorator |
| |
| def get_view(request): |
| "A simple view that expects a GET request, and returns a rendered template" |
| t = Template('This is a test. {{ var }} is the value.', name='GET Template') |
| c = Context({'var': request.GET.get('var', 42)}) |
| |
| return HttpResponse(t.render(c)) |
| |
| def post_view(request): |
| """A view that expects a POST, and returns a different template depending |
| on whether any POST data is available |
| """ |
| if request.method == 'POST': |
| if request.POST: |
| t = Template('Data received: {{ data }} is the value.', name='POST Template') |
| c = Context({'data': request.POST['value']}) |
| else: |
| t = Template('Viewing POST page.', name='Empty POST Template') |
| c = Context() |
| else: |
| t = Template('Viewing GET page.', name='Empty GET Template') |
| c = Context() |
| |
| return HttpResponse(t.render(c)) |
| |
| def view_with_header(request): |
| "A view that has a custom header" |
| response = HttpResponse() |
| response['X-DJANGO-TEST'] = 'Slartibartfast' |
| return response |
| |
| def raw_post_view(request): |
| """A view which expects raw XML to be posted and returns content extracted |
| from the XML""" |
| if request.method == 'POST': |
| root = parseString(request.raw_post_data) |
| first_book = root.firstChild.firstChild |
| title, author = [n.firstChild.nodeValue for n in first_book.childNodes] |
| t = Template("{{ title }} - {{ author }}", name="Book template") |
| c = Context({"title": title, "author": author}) |
| else: |
| t = Template("GET request.", name="Book GET template") |
| c = Context() |
| |
| return HttpResponse(t.render(c)) |
| |
| def redirect_view(request): |
| "A view that redirects all requests to the GET view" |
| if request.GET: |
| from urllib import urlencode |
| query = '?' + urlencode(request.GET, True) |
| else: |
| query = '' |
| return HttpResponseRedirect('/test_client/get_view/' + query) |
| |
| def view_with_secure(request): |
| "A view that indicates if the request was secure" |
| response = HttpResponse() |
| response.test_was_secure_request = request.is_secure() |
| return response |
| |
| def double_redirect_view(request): |
| "A view that redirects all requests to a redirection view" |
| return HttpResponseRedirect('/test_client/permanent_redirect_view/') |
| |
| def bad_view(request): |
| "A view that returns a 404 with some error content" |
| return HttpResponseNotFound('Not found!. This page contains some MAGIC content') |
| |
| TestChoices = ( |
| ('a', 'First Choice'), |
| ('b', 'Second Choice'), |
| ('c', 'Third Choice'), |
| ('d', 'Fourth Choice'), |
| ('e', 'Fifth Choice') |
| ) |
| |
| class TestForm(Form): |
| text = fields.CharField() |
| email = fields.EmailField() |
| value = fields.IntegerField() |
| single = fields.ChoiceField(choices=TestChoices) |
| multi = fields.MultipleChoiceField(choices=TestChoices) |
| |
| def form_view(request): |
| "A view that tests a simple form" |
| if request.method == 'POST': |
| form = TestForm(request.POST) |
| if form.is_valid(): |
| t = Template('Valid POST data.', name='Valid POST Template') |
| c = Context() |
| else: |
| t = Template('Invalid POST data. {{ form.errors }}', name='Invalid POST Template') |
| c = Context({'form': form}) |
| else: |
| form = TestForm(request.GET) |
| t = Template('Viewing base form. {{ form }}.', name='Form GET Template') |
| c = Context({'form': form}) |
| |
| return HttpResponse(t.render(c)) |
| |
| def form_view_with_template(request): |
| "A view that tests a simple form" |
| if request.method == 'POST': |
| form = TestForm(request.POST) |
| if form.is_valid(): |
| message = 'POST data OK' |
| else: |
| message = 'POST data has errors' |
| else: |
| form = TestForm() |
| message = 'GET form page' |
| return render_to_response('form_view.html', |
| { |
| 'form': form, |
| 'message': message |
| } |
| ) |
| |
| def login_protected_view(request): |
| "A simple view that is login protected." |
| t = Template('This is a login protected test. Username is {{ user.username }}.', name='Login Template') |
| c = Context({'user': request.user}) |
| |
| return HttpResponse(t.render(c)) |
| login_protected_view = login_required(login_protected_view) |
| |
| def login_protected_view_changed_redirect(request): |
| "A simple view that is login protected with a custom redirect field set" |
| t = Template('This is a login protected test. Username is {{ user.username }}.', name='Login Template') |
| c = Context({'user': request.user}) |
| |
| return HttpResponse(t.render(c)) |
| login_protected_view_changed_redirect = login_required(redirect_field_name="redirect_to")(login_protected_view_changed_redirect) |
| |
| def permission_protected_view(request): |
| "A simple view that is permission protected." |
| t = Template('This is a permission protected test. ' |
| 'Username is {{ user.username }}. ' |
| 'Permissions are {{ user.get_all_permissions }}.' , |
| name='Permissions Template') |
| c = Context({'user': request.user}) |
| return HttpResponse(t.render(c)) |
| permission_protected_view = permission_required('modeltests.test_perm')(permission_protected_view) |
| |
| class _ViewManager(object): |
| @method_decorator(login_required) |
| def login_protected_view(self, request): |
| t = Template('This is a login protected test using a method. ' |
| 'Username is {{ user.username }}.', |
| name='Login Method Template') |
| c = Context({'user': request.user}) |
| return HttpResponse(t.render(c)) |
| |
| @method_decorator(permission_required('modeltests.test_perm')) |
| def permission_protected_view(self, request): |
| t = Template('This is a permission protected test using a method. ' |
| 'Username is {{ user.username }}. ' |
| 'Permissions are {{ user.get_all_permissions }}.' , |
| name='Permissions Template') |
| c = Context({'user': request.user}) |
| return HttpResponse(t.render(c)) |
| |
| _view_manager = _ViewManager() |
| login_protected_method_view = _view_manager.login_protected_view |
| permission_protected_method_view = _view_manager.permission_protected_view |
| |
| def session_view(request): |
| "A view that modifies the session" |
| request.session['tobacconist'] = 'hovercraft' |
| |
| t = Template('This is a view that modifies the session.', |
| name='Session Modifying View Template') |
| c = Context() |
| return HttpResponse(t.render(c)) |
| |
| def broken_view(request): |
| """A view which just raises an exception, simulating a broken view.""" |
| raise KeyError("Oops! Looks like you wrote some bad code.") |
| |
| def mail_sending_view(request): |
| mail.EmailMessage( |
| "Test message", |
| "This is a test email", |
| "from@example.com", |
| ['first@example.com', 'second@example.com']).send() |
| return HttpResponse("Mail sent") |
| |
| def mass_mail_sending_view(request): |
| m1 = mail.EmailMessage( |
| 'First Test message', |
| 'This is the first test email', |
| 'from@example.com', |
| ['first@example.com', 'second@example.com']) |
| m2 = mail.EmailMessage( |
| 'Second Test message', |
| 'This is the second test email', |
| 'from@example.com', |
| ['second@example.com', 'third@example.com']) |
| |
| c = mail.get_connection() |
| c.send_messages([m1,m2]) |
| |
| return HttpResponse("Mail sent") |
