forked from 00dani/lemoncurry
86 lines
2 KiB
Python
86 lines
2 KiB
Python
import html
|
|
import json
|
|
from accept_types import get_best_match
|
|
from django.conf import settings
|
|
from django.http import HttpResponse, JsonResponse
|
|
from django.http import HttpResponseForbidden, HttpResponseBadRequest
|
|
from django.utils.html import strip_tags
|
|
from os.path import join
|
|
from typing import Any, Dict, Optional
|
|
from urllib.parse import urlencode, urljoin, urlparse
|
|
|
|
from .templatetags.markdown import markdown
|
|
|
|
|
|
class PackageJson:
|
|
data: Optional[Dict[str, Any]]
|
|
|
|
def __init__(self) -> None:
|
|
self.data = None
|
|
|
|
def load(self) -> Dict[str, Any]:
|
|
if self.data is None:
|
|
with open(join(settings.BASE_DIR, "package.json")) as f:
|
|
self.data = json.load(f)
|
|
assert self.data is not None
|
|
return self.data
|
|
|
|
|
|
PACKAGE = PackageJson()
|
|
|
|
|
|
def friendly_url(url):
|
|
if "//" not in url:
|
|
url = "//" + url
|
|
(scheme, netloc, path, params, q, fragment) = urlparse(url)
|
|
if path == "/":
|
|
return netloc
|
|
return "{}\u200B{}".format(netloc, path)
|
|
|
|
|
|
def load_package_json() -> Dict[str, Any]:
|
|
return PACKAGE.load()
|
|
|
|
|
|
def origin(request):
|
|
return "{0}://{1}".format(request.scheme, request.site.domain)
|
|
|
|
|
|
def absolute_url(request, url):
|
|
return urljoin(origin(request), url)
|
|
|
|
|
|
def uri(request):
|
|
return origin(request) + request.path
|
|
|
|
|
|
def form_encoded_response(content):
|
|
return HttpResponse(
|
|
urlencode(content), content_type="application/x-www-form-urlencoded"
|
|
)
|
|
|
|
|
|
REPS = {
|
|
"application/x-www-form-urlencoded": form_encoded_response,
|
|
"application/json": JsonResponse,
|
|
}
|
|
|
|
|
|
def choose_type(request, content, reps=REPS):
|
|
accept = request.META.get("HTTP_ACCEPT", "*/*")
|
|
type = get_best_match(accept, reps.keys())
|
|
if type:
|
|
return reps[type](content)
|
|
return HttpResponse(status=406)
|
|
|
|
|
|
def bad_req(message):
|
|
return HttpResponseBadRequest(message, content_type="text/plain")
|
|
|
|
|
|
def forbid(message):
|
|
return HttpResponseForbidden(message, content_type="text/plain")
|
|
|
|
|
|
def to_plain(md):
|
|
return html.unescape(strip_tags(markdown(md)))
|