lemoncurry/home/views.py

from annoying.decorators import render_to
from django.core.paginator import Paginator
from django.http import HttpResponse
from django.shortcuts import get_object_or_404, redirect
from django.urls import reverse
from users.models import User
from urllib.parse import urljoin

from entries import kinds
from lemoncurry import breadcrumbs, utils

breadcrumbs.add('home:index', 'home')


@render_to('home/index.html')
def index(request, page):
    query = User.objects.prefetch_related('entries', 'profiles', 'keys')
    user = get_object_or_404(query, pk=1)
    paginator = Paginator(user.entries.filter(kind__in=kinds.on_home), 10)

    # If we explicitly got /page/1 in the URL then redirect to the version with
    # no page suffix.
    if page == '1':
        return redirect('home:index', permanent=True)
    entries = paginator.page(page or 1)

    return {
        'user': user,
        'entries': entries,
        'atom': 'entries:atom',
        'rss': 'entries:rss',
        'meta': user.as_meta(request),
    }


def robots(request):
    base = utils.origin(request)
    lines = (
        'User-agent: *',
        'Sitemap: {0}'.format(urljoin(base, reverse('sitemap')))
    )
    return HttpResponse("\n".join(lines) + "\n", content_type='text/plain')