X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/e390e737951ea78e783070b9729a877945daaa42..4831dedf044348077af2f8c64c40a78ba1c8f8d5:/catalogue/views.py diff --git a/catalogue/views.py b/catalogue/views.py index dc2e7b107..ed0436e64 100644 --- a/catalogue/views.py +++ b/catalogue/views.py @@ -6,38 +6,49 @@ from django.core.urlresolvers import reverse from django.db.models import Q from django.contrib.auth.decorators import login_required from django.utils.datastructures import SortedDict -from django.views.decorators.http import require_GET, require_POST - +from django.views.decorators.http import require_POST +from django.contrib import auth +from django.contrib.auth.forms import UserCreationForm, AuthenticationForm +from django.utils import simplejson +from django.utils.functional import Promise +from django.utils.encoding import force_unicode + +from newtagging.views import tagged_object_list from catalogue import models from catalogue import forms from catalogue.utils import split_tags -def catalogue_redirect(request, tags=''): - if len(request.GET['q']) > 0: +class LazyEncoder(simplejson.JSONEncoder): + def default(self, obj): + if isinstance(obj, Promise): + return force_unicode(obj) + return obj + + +def search(request): + query = request.GET.get('q', '') + tags = request.GET.get('tags', '') + if tags == '': + tags = [] + + try: + tag_list = models.Tag.get_tag_list(tags) + tag = models.Tag.objects.get(name=query) + except models.Tag.DoesNotExist: + print "%r, %r" % (query, tags) try: - tag = models.Tag.objects.get(name=request.GET['q']) - if len(tags): - tags += '/' - tags = tags + tag.slug - except models.Tag.DoesNotExist: - book = get_object_or_404(models.Book, title=request.GET['q']) + book = models.Book.objects.get(title=query) return HttpResponseRedirect(book.get_absolute_url()) - if len(tags) > 0: - return HttpResponseRedirect(reverse('catalogue.views.tagged_book_list', kwargs=dict(tags=tags))) + except models.Book.DoesNotExist: + return HttpResponseRedirect(reverse('catalogue.views.main_page')) else: - return HttpResponseRedirect(reverse('catalogue.views.main_page')) - -# -# @require_GET -# def tag_search(request): -# search_form = forms.SearchForm(request.GET) -# if search_form.is_valid(): -# query = search_form.cleaned_data['q'] -# tags = search_form.cleaned_data['tags'] -# - - + tag_list.append(tag) + return HttpResponseRedirect(reverse('catalogue.views.tagged_book_list', + kwargs={'tags': '/'.join(tag.slug for tag in tag_list)} + )) + + def tags_starting_with(request): try: prefix = request.GET['q'] @@ -59,10 +70,7 @@ def tags_starting_with(request): return HttpResponse('') -def main_page(request): - if 'q' in request.GET: - return catalogue_redirect(request) - +def main_page(request): if request.user.is_authenticated(): extra_where = '(NOT catalogue_tag.category = "set" OR catalogue_tag.user_id = %d)' % request.user.id else: @@ -76,9 +84,6 @@ def main_page(request): def book_list(request): - if 'q' in request.GET: - return catalogue_redirect(request) - books = models.Book.objects.all() form = forms.SearchForm() @@ -91,46 +96,69 @@ def book_list(request): def tagged_book_list(request, tags=''): - if 'q' in request.GET: - return catalogue_redirect(request, tags) - - choices_split = tags.split('/') - tags = [] - for tag in choices_split: - tag = get_object_or_404(models.Tag, slug=tag) - if tag.category == 'set' and (not request.user.is_authenticated() or request.user != tag.user): - raise Http404 - tags.append(tag) + try: + tags = models.Tag.get_tag_list(tags) + except models.Tag.DoesNotExist: + raise Http404 - books = models.Book.objects.with_all(tags) - if request.user.is_authenticated(): extra_where = '(NOT catalogue_tag.category = "set" OR catalogue_tag.user_id = %d)' % request.user.id else: extra_where = 'NOT catalogue_tag.category = "set"' related_tags = models.Tag.objects.related_for_model(tags, models.Book, counts=True, extra={'where': [extra_where]}) categories = split_tags(related_tags) - - form = forms.SearchForm() - - return render_to_response('catalogue/tagged_book_list.html', dict( + + return tagged_object_list( + request, + tag_model=models.Tag, + queryset_or_model=models.Book, tags=tags, - form=form, - books=books, - categories=categories, - ), context_instance=RequestContext(request)) + template_name='catalogue/tagged_book_list.html', + extra_context = {'categories': categories }, + ) def book_detail(request, slug): book = get_object_or_404(models.Book, slug=slug) tags = list(book.tags.filter(~Q(category='set'))) categories = split_tags(tags) - search_form = forms.SearchForm() return render_to_response('catalogue/book_detail.html', locals(), context_instance=RequestContext(request)) +def logout_then_redirect(request): + auth.logout(request) + return HttpResponseRedirect(request.GET.get('next', '/')) + + +@require_POST +def register(request): + registration_form = UserCreationForm(request.POST, prefix='registration') + if registration_form.is_valid(): + user = registration_form.save() + user = auth.authenticate( + username=registration_form.cleaned_data['username'], + password=registration_form.cleaned_data['password1'] + ) + auth.login(request, user) + response_data = {'success': True, 'errors': {}} + else: + response_data = {'success': False, 'errors': registration_form.errors} + return HttpResponse(LazyEncoder(ensure_ascii=False).encode(response_data)) + + +@require_POST +def login(request): + form = AuthenticationForm(data=request.POST, prefix='login') + if form.is_valid(): + auth.login(request, form.get_user()) + response_data = {'success': True, 'errors': {}} + else: + response_data = {'success': False, 'errors': form.errors} + return HttpResponse(LazyEncoder(ensure_ascii=False).encode(response_data)) + + @login_required def book_sets(request, slug): book = get_object_or_404(models.Book, slug=slug)