lemoncurry/entries/from_url.py

from urllib.parse import urlparse

from django.contrib.sites.models import Site
from django.http import HttpResponse
from django.urls import resolve, Resolver404
from micropub import error

from .models import Entry


def from_url(url: str) -> Entry:
    domain = Site.objects.get_current().domain
    if not url:
        raise error.bad_req('url parameter required')
    if '//' not in url:
        url = '//' + url
    parts = urlparse(url, scheme='https')
    if parts.scheme not in ('http', 'https') or parts.netloc != domain:
        raise error.bad_req('url does not point to this site')

    try:
        match = resolve(parts.path)
    except Resolver404:
        raise error.bad_req('url does not point to a valid page on this site')

    if match.view_name != 'entries:entry':
        raise error.bad_req('url does not point to an entry on this site')

    try:
        entry = Entry.objects.get(pk=match.kwargs['id'])
    except Entry.DoesNotExist:
        raise error.bad_req('url does not point to an existing entry')

    return entry