import urllib.parse from django.contrib.auth.decorators import login_required from django.core.paginator import Paginator from django.http import HttpResponseRedirect, Http404 from django.shortcuts import render from django.urls import reverse from bookmarks import queries from bookmarks.models import Bookmark, BookmarkForm, build_tag_string from bookmarks.services.bookmarks import create_bookmark, update_bookmark, archive_bookmark, archive_bookmarks, \ unarchive_bookmark, unarchive_bookmarks, delete_bookmarks, tag_bookmarks, untag_bookmarks _default_page_size = 30 @login_required def index(request): query_string = request.GET.get('q') query_set = queries.query_bookmarks(request.user, query_string) tags = queries.query_bookmark_tags(request.user, query_string) base_url = reverse('bookmarks:index') context = get_bookmark_view_context(request, query_set, tags, base_url) return render(request, 'bookmarks/index.html', context) @login_required def archived(request): query_string = request.GET.get('q') query_set = queries.query_archived_bookmarks(request.user, query_string) tags = queries.query_archived_bookmark_tags(request.user, query_string) base_url = reverse('bookmarks:archived') context = get_bookmark_view_context(request, query_set, tags, base_url) return render(request, 'bookmarks/archive.html', context) def get_bookmark_view_context(request, query_set, tags, base_url): page = request.GET.get('page') query_string = request.GET.get('q') paginator = Paginator(query_set, _default_page_size) bookmarks = paginator.get_page(page) return_url = generate_return_url(base_url, page, query_string) link_target = request.user.profile.bookmark_link_target if request.GET.get('tag'): mod = request.GET.copy() mod.pop('tag') request.GET = mod return { 'bookmarks': bookmarks, 'tags': tags, 'query': query_string if query_string else '', 'empty': paginator.count == 0, 'return_url': return_url, 'link_target': link_target, } def generate_return_url(base_url, page, query_string): url_query = {} if query_string is not None: url_query['q'] = query_string if page is not None: url_query['page'] = page url_params = urllib.parse.urlencode(url_query) return_url = base_url if url_params == '' else base_url + '?' + url_params return urllib.parse.quote_plus(return_url) def convert_tag_string(tag_string: str): # Tag strings coming from inputs are space-separated, however services.bookmarks functions expect comma-separated # strings return tag_string.replace(' ', ',') @login_required def new(request): initial_url = request.GET.get('url') initial_auto_close = 'auto_close' in request.GET if request.method == 'POST': form = BookmarkForm(request.POST) auto_close = form.data['auto_close'] if form.is_valid(): current_user = request.user tag_string = convert_tag_string(form.data['tag_string']) create_bookmark(form.save(commit=False), tag_string, current_user) if auto_close: return HttpResponseRedirect(reverse('bookmarks:close')) else: return HttpResponseRedirect(reverse('bookmarks:index')) else: form = BookmarkForm() if initial_url: form.initial['url'] = initial_url if initial_auto_close: form.initial['auto_close'] = 'true' context = { 'form': form, 'auto_close': initial_auto_close, 'return_url': reverse('bookmarks:index') } return render(request, 'bookmarks/new.html', context) @login_required def edit(request, bookmark_id: int): try: bookmark = Bookmark.objects.get(pk=bookmark_id, owner=request.user) except Bookmark.DoesNotExist: raise Http404('Bookmark does not exist') if request.method == 'POST': form = BookmarkForm(request.POST, instance=bookmark) return_url = form.data['return_url'] if form.is_valid(): tag_string = convert_tag_string(form.data['tag_string']) update_bookmark(form.save(commit=False), tag_string, request.user) return HttpResponseRedirect(return_url) else: return_url = request.GET.get('return_url') form = BookmarkForm(instance=bookmark) return_url = return_url if return_url else reverse('bookmarks:index') form.initial['tag_string'] = build_tag_string(bookmark.tag_names, ' ') form.initial['return_url'] = return_url context = { 'form': form, 'bookmark_id': bookmark_id, 'return_url': return_url } return render(request, 'bookmarks/edit.html', context) @login_required def remove(request, bookmark_id: int): try: bookmark = Bookmark.objects.get(pk=bookmark_id, owner=request.user) except Bookmark.DoesNotExist: raise Http404('Bookmark does not exist') bookmark.delete() return_url = request.GET.get('return_url') return_url = return_url if return_url else reverse('bookmarks:index') return HttpResponseRedirect(return_url) @login_required def archive(request, bookmark_id: int): try: bookmark = Bookmark.objects.get(pk=bookmark_id, owner=request.user) except Bookmark.DoesNotExist: raise Http404('Bookmark does not exist') archive_bookmark(bookmark) return_url = request.GET.get('return_url') return_url = return_url if return_url else reverse('bookmarks:index') return HttpResponseRedirect(return_url) @login_required def unarchive(request, bookmark_id: int): try: bookmark = Bookmark.objects.get(pk=bookmark_id, owner=request.user) except Bookmark.DoesNotExist: raise Http404('Bookmark does not exist') unarchive_bookmark(bookmark) return_url = request.GET.get('return_url') return_url = return_url if return_url else reverse('bookmarks:archived') return HttpResponseRedirect(return_url) @login_required def bulk_edit(request): bookmark_ids = request.POST.getlist('bookmark_id') # Determine action if 'bulk_archive' in request.POST: archive_bookmarks(bookmark_ids, request.user) if 'bulk_unarchive' in request.POST: unarchive_bookmarks(bookmark_ids, request.user) if 'bulk_delete' in request.POST: delete_bookmarks(bookmark_ids, request.user) if 'bulk_tag' in request.POST: tag_string = convert_tag_string(request.POST['bulk_tag_string']) tag_bookmarks(bookmark_ids, tag_string, request.user) if 'bulk_untag' in request.POST: tag_string = convert_tag_string(request.POST['bulk_tag_string']) untag_bookmarks(bookmark_ids, tag_string, request.user) return_url = request.GET.get('return_url') return_url = return_url if return_url else reverse('bookmarks:index') return HttpResponseRedirect(return_url) @login_required def close(request): return render(request, 'bookmarks/close.html')